Browse Source

!10509 adapt to hccl interface changes

From: @nicholas_yhr
Reviewed-by: @kisnwang,@zhoufeng54
Signed-off-by: @zhoufeng54
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
1cbc50e54a
1 changed files with 4 additions and 4 deletions
  1. +4
    -4
      mindspore/ccsrc/runtime/device/ascend/executor/hccl_dynamic_kernel.cc

+ 4
- 4
mindspore/ccsrc/runtime/device/ascend/executor/hccl_dynamic_kernel.cc View File

@@ -109,7 +109,7 @@ void HcclDynamicKernel::Execute() {

auto callback = [this](HcclResult status) {
if (status != HCCL_SUCCESS) {
MS_LOG(ERROR) << "HcomExcutorInitialize failed, ret:" << status;
MS_LOG(ERROR) << "HcomExecutorInitialize failed, ret:" << status;
}
std::lock_guard<std::mutex> lock(this->hccl_mutex_);
this->cond_.notify_all();
@@ -145,7 +145,7 @@ bool HcclExecutorManager::Initialize() {
return false;
}

auto HcomExecutorInitialize = (HcclResult(*)())dlsym(handle_, "HcomExcutorInitialize");
auto HcomExecutorInitialize = (HcclResult(*)())dlsym(handle_, "HcomExecInitialize");
if (HcomExecutorInitialize == nullptr) {
MS_LOG(ERROR) << "dlsym HcomExecutorInitialize failed";
return false;
@@ -167,7 +167,7 @@ bool HcclExecutorManager::Finalize() {
if (!initialized_) {
return true;
}
auto HcomExecutorFinalize = (HcclResult(*)())dlsym(handle_, "HcomExcutorFinalize");
auto HcomExecutorFinalize = (HcclResult(*)())dlsym(handle_, "HcomExecFinalize");
if (HcomExecutorFinalize == nullptr) {
MS_LOG(ERROR) << "Fail to dlsym HcomExecutorFinalize";
return false;
@@ -185,7 +185,7 @@ bool HcclExecutorManager::Finalize() {
return true;
}

void *HcclExecutorManager::GetHcomOpertion() { return dlsym(handle_, "EnqueueHcomOpertion"); }
void *HcclExecutorManager::GetHcomOpertion() { return dlsym(handle_, "HcomExecEnqueueOperation"); }
void HcclExecutorManager::CloseHandle() {
if (dlclose(handle_) != 0) {
MS_LOG(WARNING) << "Failed to close hcom handle";


Loading…
Cancel
Save