From 59ef2d0fcdc10127a50c5bb6ed8f69be36b151ed Mon Sep 17 00:00:00 2001 From: wangzhengjun Date: Thu, 1 Apr 2021 15:54:20 +0800 Subject: [PATCH] buffer pool skip continuous memory --- ge/graph/build/memory/block_mem_assigner.cc | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ge/graph/build/memory/block_mem_assigner.cc b/ge/graph/build/memory/block_mem_assigner.cc index ad5ed1a2..9825d1ed 100755 --- a/ge/graph/build/memory/block_mem_assigner.cc +++ b/ge/graph/build/memory/block_mem_assigner.cc @@ -1662,7 +1662,8 @@ Status BlockMemAssigner::AssignOutputMemoryWithReuse(const NodePtr &node, vector 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->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_); } for (uint32_t i = 0; i < static_cast(op_desc->GetOutputsSize()); i++) {