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

!21652 fix bugs: DestroyHccl must be called before FreeDeviceMemory 

Merge pull request !21652 from jjfeing/r1.4
This commit is contained in:
i-robot 2021-08-11 11:59:51 +00:00 committed by Gitee
parent fb2b4f0617 8192c0034a
commit 6efa88c486
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -249,6 +249,8 @@ void AscendKernelRuntime::ReleaseDeviceRes() {
MS_EXCEPTION_IF_NULL(context_ptr); MS_EXCEPTION_IF_NULL(context_ptr);
uint32_t device_id = context_ptr->get_param<uint32_t>(MS_CTX_DEVICE_ID); uint32_t device_id = context_ptr->get_param<uint32_t>(MS_CTX_DEVICE_ID);
// DestroyHccl must be called before FreeDeviceMemory
(void)DestroyHccl();
if (mem_manager_ != nullptr) { if (mem_manager_ != nullptr) {
mem_manager_->FreeDeviceMemory(); mem_manager_->FreeDeviceMemory();
} }
@ -258,7 +260,6 @@ void AscendKernelRuntime::ReleaseDeviceRes() {
MS_LOG(EXCEPTION) << "Reg SetTaskFailCallback failed, error: " << rt_ret; MS_LOG(EXCEPTION) << "Reg SetTaskFailCallback failed, error: " << rt_ret;
} }
(void)DestroyHccl();
(void)ResetDevice(device_id); (void)ResetDevice(device_id);
(void)ProfilingManager::GetInstance().StopProfiling(); (void)ProfilingManager::GetInstance().StopProfiling();
current_graph_ = nullptr; current_graph_ = nullptr;