Lines Matching refs:gpu_indices
761 std::vector<int> gpu_indices = CSVToVector<int>(options.gpu_index); in CreateSiftGPUExtractor() local
762 CHECK_EQ(gpu_indices.size(), 1) << "SiftGPU can only run on one GPU"; in CreateSiftGPUExtractor()
770 if (!options.darkness_adaptivity && gpu_indices[0] < 0) { in CreateSiftGPUExtractor()
771 gpu_indices[0] = 0; in CreateSiftGPUExtractor()
774 if (gpu_indices[0] >= 0) { in CreateSiftGPUExtractor()
776 sift_gpu_args.push_back(std::to_string(gpu_indices[0])); in CreateSiftGPUExtractor()
783 if (gpu_indices[0] >= 0) { in CreateSiftGPUExtractor()
837 sift_gpu->gpu_index = gpu_indices[0]; in CreateSiftGPUExtractor()
838 if (sift_extraction_mutexes.count(gpu_indices[0]) == 0) { in CreateSiftGPUExtractor()
840 gpu_indices[0], std::unique_ptr<std::mutex>(new std::mutex())); in CreateSiftGPUExtractor()
1094 const std::vector<int> gpu_indices = in CreateSiftGPUMatcher() local
1096 CHECK_EQ(gpu_indices.size(), 1) << "SiftGPU can only run on one GPU"; in CreateSiftGPUMatcher()
1104 if (gpu_indices[0] >= 0) { in CreateSiftGPUMatcher()
1106 gpu_indices[0]); in CreateSiftGPUMatcher()
1139 sift_match_gpu->gpu_index = gpu_indices[0]; in CreateSiftGPUMatcher()
1140 if (sift_matching_mutexes.count(gpu_indices[0]) == 0) { in CreateSiftGPUMatcher()
1142 gpu_indices[0], std::unique_ptr<std::mutex>(new std::mutex())); in CreateSiftGPUMatcher()