提交 7a6fdaf1 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!302 fix workspace memory reuse

Merge pull request !302 from kisnwang/fix-workspace-malloc
......@@ -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++;
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册