Browse Source

multi batch

tags/v1.1.0
wjm 3 years ago
parent
commit
a8c720f8e8
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/graph/build/memory/block_mem_assigner.cc

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

@@ -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;


Loading…
Cancel
Save