Browse Source

!1429 buffer pool skip continuous memory

From: @selfws
Reviewed-by: @tangqunzhang,@xchu42,@wqtshg
Signed-off-by: @wqtshg
tags/v1.3.0
mindspore-ci-bot Gitee 3 years ago
parent
commit
aeff798209
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      ge/graph/build/memory/block_mem_assigner.cc

+ 2
- 1
ge/graph/build/memory/block_mem_assigner.cc View File

@@ -1662,7 +1662,8 @@ Status BlockMemAssigner::AssignOutputMemoryWithReuse(const NodePtr &node, vector
for (auto iter = stream_workspace_blocks_.begin(); iter != stream_workspace_blocks_.end(); for (auto iter = stream_workspace_blocks_.begin(); iter != stream_workspace_blocks_.end();
++iter) { ReleaseMemorys(iter->second[stream_id], reusable_blocks_[iter->first][stream_id]); ++iter) { ReleaseMemorys(iter->second[stream_id], reusable_blocks_[iter->first][stream_id]);
iter->second[stream_id].clear();}); iter->second[stream_id].clear();});
if (IsContinuousOutput(node)) {
bool need_apply_continuous_memory = IsContinuousOutput(node) && (!is_buffer_pool_mem_supported);
if (need_apply_continuous_memory) {
return ApplyContinuousMemory(node, ranges, is_op_reuse_mem_); return ApplyContinuousMemory(node, ranges, is_op_reuse_mem_);
} }
for (uint32_t i = 0; i < static_cast<uint32_t>(op_desc->GetOutputsSize()); i++) { for (uint32_t i = 0; i < static_cast<uint32_t>(op_desc->GetOutputsSize()); i++) {


Loading…
Cancel
Save