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

!7193 fix hccl dynamic kernel finalize bug 

Merge pull request !7193 from caifubi/dynamic_shape
This commit is contained in:
mindspore-ci-bot 2020-10-12 19:45:10 +08:00 committed by Gitee
parent 81d9015ddb fa6293d905
commit 74b989d425
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
mindspore/ccsrc/runtime/device/ascend/executor/hccl_dynamic_kernel.cc
View File

@ -138,6 +138,11 @@ bool HcclExecutorManager::Initialize() {
if (initialized_) {
return true;
}
auto context = MsContext::GetInstance();
MS_EXCEPTION_IF_NULL(context);
if (!context->get_param<bool>(MS_CTX_ENABLE_HCCL)) {
return true;
}
initialized_ = true;
MS_LOG(INFO) << "Start Initialize Hccl DynamicKernel";
handle_ = dlopen(kHcomGraphAdaptorPath, RTLD_NOW | RTLD_GLOBAL);
@ -165,9 +170,12 @@ bool HcclExecutorManager::Initialize() {
}
bool HcclExecutorManager::Finalize() {
if (!initialized_) {
return true;
}
auto HcomExecutorFinalize = (HcclResult(*)())dlsym(handle_, "HcomExcutorFinalize");
if (HcomExecutorFinalize == nullptr) {
MS_LOG(ERROR) << "Faile to dlsym HcomExecutorFinalize";
MS_LOG(ERROR) << "Fail to dlsym HcomExecutorFinalize";
return false;
}
HcclResult hccl_ret = HcomExecutorFinalize();
@ -179,7 +187,7 @@ bool HcclExecutorManager::Finalize() {
MS_LOG(ERROR) << "Failed to close hcom handle";
return false;
}
MS_LOG(INFO) << "Hccl DynamicKernel Finalize failed";
MS_LOG(INFO) << "Hccl DynamicKernel Finalize success";
return true;
}
} // namespace ascend