Fix the omp_num_threads bug

23e0d11c · lichao18 · 8217088f · 23e0d11c
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

mace/core/runtime/cpu/cpu_runtime.cc mace/core/runtime/cpu/cpu_runtime.cc +1 -1

未找到文件。
--- a/mace/core/runtime/cpu/cpu_runtime.cc
+++ b/mace/core/runtime/cpu/cpu_runtime.cc
@@ -223,7 +223,7 @@ MaceStatus CPURuntime::SetOpenMPThreadsAndAffinityPolicy(
      }
      ++cores_to_use;
    }
-    num_threads_hint = cores_to_use;
+    num_threads_hint = std::min(num_threads_hint, cores_to_use);
  } else {
    cores_to_use = num_threads_hint;
  }