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

!19270 fix coredump when env is null 

Merge pull request !19270 from jjfeing/master
This commit is contained in:
i-robot 2021-07-05 01:04:25 +00:00 committed by Gitee
parent bee52f52c3 bc5105776f
commit 0e585335f7
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/ccsrc/runtime/device/ascend/ascend_kernel_runtime.cc
View File

@ -82,7 +82,7 @@ std::string GetRankId() {
MS_EXCEPTION_IF_NULL(mpi_config_ptr);
if (mpi_config_ptr->enable_mpi()) {
int rank_id = GetMPIRankId();
const std::string offset = std::getenv("RANK_OFFSET");
const std::string offset = common::GetEnv("RANK_OFFSET");
if (offset.empty()) {
try {
int rank_offset = std::stoi(offset);
@ -95,10 +95,10 @@ std::string GetRankId() {
}
rank_id_str = std::to_string(rank_id);
} else {
rank_id_str = std::getenv("RANK_ID");
rank_id_str = common::GetEnv("RANK_ID");
}
#else
rank_id_str = std::getenv("RANK_ID");
rank_id_str = common::GetEnv("RANK_ID");
#endif
if (rank_id_str.empty()) {
MS_LOG(ERROR) << "Get hccl rankid failed, please set env RANK_ID";