cuijinrui
/
mindspore
forked from OSSInnovation/mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!302 fix workspace memory reuse 

Merge pull request !302 from kisnwang/fix-workspace-malloc
This commit is contained in:
mindspore-ci-bot 2020-04-14 17:28:51 +08:00 committed by Gitee
parent e364936bd3 e53376092f
commit 7a6fdaf132
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/device/kernel_runtime.cc
View File

@ -466,7 +466,7 @@ void KernelRuntime::AssignWorkSpaceMem(int flag, const AnfNodePtr &node) {
MS_EXCEPTION_IF_NULL(kernel_mod);
size_t index = 0;
for (auto &size : kernel_mod->GetWorkspaceSizeList()) {
auto ptr = mem_manager_->MallocWorkSpaceMem(node, flag, index, size);
auto ptr = mem_manager_->MallocWorkSpaceMem(node, index, flag, size);
AnfAlgo::SetWorkspaceAddr(CreateDeviceAddress(ptr, size, "", kTypeUnknown), index, node.get());
index++;
}