vilc
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!31628 fix codecheck and runtime num thread default 

Merge pull request !31628 from fangzehua/fix_code_0321
This commit is contained in:
i-robot 2022-03-22 11:14:46 +00:00 committed by Gitee
parent 4b9e0708b8 f2dfd01d4e
commit 6a85204890
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/core/utils/ms_context.cc
View File

@ -103,7 +103,9 @@ MsContext::MsContext(const std::string &policy, const std::string &target) {
set_param<bool>(MS_CTX_ENABLE_RECOVERY, false);
uint32_t kDefaultRuntimeNumThreads = 30;
set_param<uint32_t>(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<uint32_t>(MS_CTX_RUNTIME_NUM_THREADS, runtime_num_threads_default);
backend_policy_ = policy_map_[policy];
}