|
|
|
@ -881,14 +881,14 @@ MemoryBlock *BlockMemAssigner::ApplyMemory(size_t block_size, size_t real_size,
|
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
std::string batch_label;
|
|
|
|
|
if (reusable_block->IsSameLabel(batch_label)) {
|
|
|
|
|
std::string op_label;
|
|
|
|
|
(void)ge::AttrUtils::GetStr(node_op_desc, ATTR_NAME_BATCH_LABEL, op_label);
|
|
|
|
|
if (batch_label != op_label) {
|
|
|
|
|
if (reusable_block->IsSameLabel(batch_label)) {
|
|
|
|
|
std::string 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;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// A node can reuse blocks of the same stream and preorder streams
|
|
|
|
|
if (CanReuseBySize(reusable_block_counts_, *reusable_block, block_size, real_size, continuous)) {
|
|
|
|
|