diff --git a/mindspore/core/utils/ms_context.cc b/mindspore/core/utils/ms_context.cc index e0cecdca84d..571fda0c27e 100644 --- a/mindspore/core/utils/ms_context.cc +++ b/mindspore/core/utils/ms_context.cc @@ -103,7 +103,9 @@ MsContext::MsContext(const std::string &policy, const std::string &target) { set_param(MS_CTX_ENABLE_RECOVERY, false); uint32_t kDefaultRuntimeNumThreads = 30; - set_param(MS_CTX_RUNTIME_NUM_THREADS, kDefaultRuntimeNumThreads); + uint32_t cpu_core_num = std::thread::hardware_concurrency() - 1; + uint32_t runtime_num_threads_default = std::min(cpu_core_num, kDefaultRuntimeNumThreads); + set_param(MS_CTX_RUNTIME_NUM_THREADS, runtime_num_threads_default); backend_policy_ = policy_map_[policy]; }