From a8c720f8e8537018fd0cd78e929547b05e387022 Mon Sep 17 00:00:00 2001 From: wjm Date: Sat, 14 Nov 2020 17:46:08 +0800 Subject: [PATCH] multi batch --- ge/graph/build/memory/block_mem_assigner.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ge/graph/build/memory/block_mem_assigner.cc b/ge/graph/build/memory/block_mem_assigner.cc index f89a9c13..cdca7fb7 100755 --- a/ge/graph/build/memory/block_mem_assigner.cc +++ b/ge/graph/build/memory/block_mem_assigner.cc @@ -883,7 +883,7 @@ MemoryBlock *BlockMemAssigner::ApplyMemory(size_t block_size, size_t real_size, std::string batch_label; if (reusable_block->IsSameLabel(batch_label)) { std::string op_label; - (void)ge::AttrUtils::GerStr(node_op_desc, Attr_NAME_BATCH_LABEL, op_label); + (void)ge::AttrUtils::GetStr(node_op_desc, ATTR_NAME_BATCH_LABEL, op_label); if (batch_label != op_label) { GELOGI("label diff, op name %s", node_op_desc->GetName().c_str()); continue;