[src] stop at first successful gpu select (fix bug from #3750) (#3776)

kaldi-asr · Dec 18, 2019 · ba92f60 · ba92f60
1 parent b5bcf2f
commit ba92f60
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/cudamatrix/cu-device.cc b/src/cudamatrix/cu-device.cc
@@ -472,7 +472,7 @@ bool CuDevice::SelectGpuIdAuto() {
     success = SelectGpuId(dev_id);
 
     max_id++;
-  } while (success && (max_id < free_mem_ratio.size()));
+  } while (!success && (max_id < free_mem_ratio.size()));
 
   if (e != cudaSuccess) {
     KALDI_WARN << "Failed to (automatically) select any device";