[Feature](mlu-ops): fix driver-process error. (#931)

Cambricon · Jan 22, 2024 · c4821a1 · c4821a1
1 parent 3f5b4ff
commit c4821a1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/test/mlu_op_gtest/pb_gtest/src/executor.cpp b/test/mlu_op_gtest/pb_gtest/src/executor.cpp
@@ -188,7 +188,7 @@ void Executor::setup(std::string file,
   deviceRestSpaceMalloc();
 
   // when get MLUOP_GTEST_FILL_RAM env,
-  // fill nram/sram/warm for nan or inf beore compute for each case
+  // fill nram/sram/warm for nan or inf before compute for each case
   fillRam();
 
   VLOG(4) << "Algorithm Search.";

diff --git a/test/mlu_op_gtest/pb_gtest/src/gtest/test_env.cpp b/test/mlu_op_gtest/pb_gtest/src/gtest/test_env.cpp
@@ -505,8 +505,9 @@ void TestEnvironment::setDevice() {
       device_id = sorted_device_ids[i].device_id;
       process_count = sorted_device_ids[i].process_count;
       std::string bus_id = sorted_device_ids[i].bus_id;
-      if ((process_count == 0 &&
-           (global_var.run_on_jenkins_ || setComputeMode(bus_id, '0'))) ||
+      if ((process_count == 0 && (global_var.run_on_jenkins_ ||
+                                  sorted_device_ids[i].compute_mode == '0' ||
+                                  setComputeMode(bus_id, '0'))) ||
           process_count > 0) {
         is_picked = true;
         break;