!996 move end_task after assignadd on iterator_loop case

From: @wangxiaotian22
Reviewed-by: @xchu42,@wqtshg
Signed-off-by: @ji_chen
pull/996/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit f41cbfee3b

@ -673,7 +673,7 @@ Status TaskGenerator::AutoFindBpOpIndex(const ComputeGraphPtr &graph, ProfilingP
}
}
if (graph->GetNeedIteration()) {
if (op_desc->GetName() == NODE_NAME_NET_OUTPUT + '_' + NODE_NAME_STREAM_SWITCH + "_StreamActive") {
if (op_desc->GetName() == NODE_NAME_FLOWCTRL_LOOP_ASSIGNADD) {
profiling_point.end_index.insert(current_idx);
GELOGI("Iter end name %s, idx %u, from Node_Output_IteratorCtrl_StreamSwitch_StreamActive",
op_desc->GetName().c_str(), current_idx);

Loading…
Cancel
Save