From 2b0464ad81bffebc7a38bfd886bf31b653323c03 Mon Sep 17 00:00:00 2001 From: wangxiaotian22 Date: Fri, 29 Jan 2021 09:06:40 +0800 Subject: [PATCH] move end_task before active stream when bpfp set by env --- 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 b6a75ef4..bb5bbd52 100644 --- a/ge/graph/build/task_generator.cc +++ b/ge/graph/build/task_generator.cc @@ -773,7 +773,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);