Browse Source

!626 gpu dynamic memory pool can't reuse allReduce in multi-stream

Merge pull request !626 from limingqi107/master
tags/v0.2.0-alpha
mindspore-ci-bot Gitee 5 years ago
parent
commit
c646faed1e
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      mindspore/ccsrc/device/gpu/gpu_kernel_runtime.cc

+ 3
- 0
mindspore/ccsrc/device/gpu/gpu_kernel_runtime.cc View File

@@ -322,6 +322,9 @@ void GPUKernelRuntime::FreeKernelDynamicRes(const mindspore::AnfNodePtr &kernel,
MS_EXCEPTION_IF_NULL(mem_reuse_util_ptr);
auto cnode = kernel->cast<CNodePtr>();
MS_EXCEPTION_IF_NULL(cnode);
if (AnfAlgo::GetCNodeName(kernel) == kAllReduceOpName) {
return;
}
// Free the input of kernel by reference count.
for (size_t i = 0; i < AnfAlgo::GetInputTensorNum(kernel); ++i) {
auto kernel_ref_count_ptr = mem_reuse_util_ptr->GetKernelInputRef(cnode, i);


Loading…
Cancel
Save