multi batch

5 years ago · a8c720f8e8
parent 8e67c970b8
commit a8c720f8e8
1 changed files with 1 additions and 1 deletions
--- 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;