p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix bug of invalid DEVICE_ID env for context

This commit is contained in:
caifubi 2021-03-09 17:27:58 +08:00
parent fe3b26a637
commit 837ee12c02
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/ccsrc/pipeline/jit/pipeline.cc
View File

@ -1071,8 +1071,7 @@ void InitHccl() {
runtime_instance->PreInit();
(void)context::OpenTsd(ms_context);
if (!runtime_instance->Init()) {
MS_LOG(ERROR) << "Kernel runtime init error.";
return;
MS_LOG(EXCEPTION) << "Runtime init failed.";
}
} else {
(void)context::OpenTsd(ms_context);

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

@ -47,8 +47,13 @@ MsContext::MsContext(const std::string &policy, const std::string &target) {
set_param<bool>(MS_CTX_ENABLE_REDUCE_PRECISION, true);
auto env_device = common::GetEnv("DEVICE_ID");
if (!env_device.empty()) {
uint32_t device_id = UlongToUint(std::stoul(env_device.c_str()));
set_param<uint32_t>(MS_CTX_DEVICE_ID, device_id);
try {
uint32_t device_id = UlongToUint(std::stoul(env_device));
set_param<uint32_t>(MS_CTX_DEVICE_ID, device_id);
} catch (std::invalid_argument &e) {
MS_LOG(WARNING) << "Invalid DEVICE_ID env:" << env_device << ". Please set DEVICE_ID to 0-7";
set_param<uint32_t>(MS_CTX_DEVICE_ID, 0);
}
} else {
set_param<uint32_t>(MS_CTX_DEVICE_ID, 0);
}