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;