Browse Source

move end_task after assginadd on iterator_loop case

tags/v1.2.0
wangxiaotian22 3 years ago
parent
commit
de16a4f89b
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/graph/build/task_generator.cc

+ 1
- 1
ge/graph/build/task_generator.cc View File

@@ -681,7 +681,7 @@ Status TaskGenerator::AutoFindBpOpIndex(const ComputeGraphPtr &graph, ProfilingP
} }
} }
if (graph->GetNeedIteration()) { 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); profiling_point.end_index.insert(current_idx);
GELOGI("Iter end name %s, idx %u, from Node_Output_IteratorCtrl_StreamSwitch_StreamActive", GELOGI("Iter end name %s, idx %u, from Node_Output_IteratorCtrl_StreamSwitch_StreamActive",
op_desc->GetName().c_str(), current_idx); op_desc->GetName().c_str(), current_idx);


Loading…
Cancel
Save