Browse Source

!1025 Fix concat memory bug

From: @tangqunzhang
Reviewed-by: @xchu42,@ji_chen
Signed-off-by: @ji_chen
tags/v1.2.0
mindspore-ci-bot Gitee 3 years ago
parent
commit
ec108c527f
3 changed files with 4 additions and 4 deletions
  1. +2
    -2
      ge/graph/build/memory/graph_mem_assigner.cc
  2. +1
    -1
      metadef
  3. +1
    -1
      parser

+ 2
- 2
ge/graph/build/memory/graph_mem_assigner.cc View File

@@ -528,7 +528,7 @@ Status GraphMemoryAssigner::AssignContinuousInputMemory(const ge::NodePtr &node,


GELOGI("[IMAS]Continuous input : Set %s name[%s] optype[%s] output[%d] offset to [%zu] stream_id[%ld] memtype[%ld] " GELOGI("[IMAS]Continuous input : Set %s name[%s] optype[%s] output[%d] offset to [%zu] stream_id[%ld] memtype[%ld] "
"size[%zu] realsize[%ld] nopadding[%d].", node->GetOwnerComputeGraph()->GetName().c_str(), "size[%zu] realsize[%ld] nopadding[%d].", node->GetOwnerComputeGraph()->GetName().c_str(),
node->GetType().c_str(), peer_op_desc->GetName().c_str(),peer_out_data_anchor->GetIdx(),
peer_op_desc->GetName().c_str(), node->GetType().c_str(), peer_out_data_anchor->GetIdx(),
output_list.at(peer_out_data_anchor->GetIdx()), peer_op_desc->GetStreamId(), memory_type, output_list.at(peer_out_data_anchor->GetIdx()), peer_op_desc->GetStreamId(), memory_type,
is_continuous_input_allocated ? 0UL : align_size, real_size, is_nopadding); is_continuous_input_allocated ? 0UL : align_size, real_size, is_nopadding);
} }
@@ -618,7 +618,7 @@ Status GraphMemoryAssigner::AssignContinuousOutputMemory(const ge::NodePtr &node
} }
GELOGI("[IMAS]Continuous output : Set %s name[%s] optype[%s] output[%d] offset to [%zu] stream_id[%ld] memtype[%ld]" GELOGI("[IMAS]Continuous output : Set %s name[%s] optype[%s] output[%d] offset to [%zu] stream_id[%ld] memtype[%ld]"
" size[%zu] realsize[%ld] nopadding[%d].", node->GetOwnerComputeGraph()->GetName().c_str(), " size[%zu] realsize[%ld] nopadding[%d].", node->GetOwnerComputeGraph()->GetName().c_str(),
node->GetType().c_str(), out_op_desc->GetName().c_str(), out_data_anchor->GetIdx(),
out_op_desc->GetName().c_str(), node->GetType().c_str(), out_data_anchor->GetIdx(),
output_list[out_data_anchor->GetIdx()], out_op_desc->GetStreamId(), memory_type, 0UL, output_list[out_data_anchor->GetIdx()], out_op_desc->GetStreamId(), memory_type, 0UL,
is_nopadding ? nopadding_size : tensor_desc_size, is_nopadding); is_nopadding ? nopadding_size : tensor_desc_size, is_nopadding);
} }


+ 1
- 1
metadef

@@ -1 +1 @@
Subproject commit 8ab60be2870b80b1ec952bb21c7f05ae2a624984
Subproject commit c8e05214e35ec13050aeeae1c4ab2b795a0e2987

+ 1
- 1
parser

@@ -1 +1 @@
Subproject commit 98f17f4a2a37f283797858eabefa9dba1d06a66b
Subproject commit 6989d1fcc704b19e7b0c62660ca35d8839d201c0

Loading…
Cancel
Save