From e46c8518e11207605d06b474eb11ef6adc89fe0e Mon Sep 17 00:00:00 2001 From: wangxiaotian22 Date: Thu, 28 Jan 2021 15:14:45 +0800 Subject: [PATCH] fix profiling fushu problem --- ge/graph/build/task_generator.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ge/graph/build/task_generator.cc b/ge/graph/build/task_generator.cc index 8bd7d32e..7548b3ea 100755 --- a/ge/graph/build/task_generator.cc +++ b/ge/graph/build/task_generator.cc @@ -784,7 +784,7 @@ Status TaskGenerator::FindBpOfEnv(const ComputeGraphPtr &graph, const std::strin } 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);