fix SummaryNodes in new control_flow

6 years ago · 4df609c94e
parent 3638886ec6
commit 4df609c94e
1 changed files with 1 additions and 0 deletions
--- a/mindspore/ccsrc/backend/session/ascend_session.cc
+++ b/mindspore/ccsrc/backend/session/ascend_session.cc
@ -179,6 +179,7 @@ GraphId AscendSession::CompileGraph(NotNull<FuncGraphPtr> func_graph) {
    debugger_->PreExecute(root_graph);
  }
 #endif
+  SetSummaryNodes(root_graph.get());
  // alloc mem
  MemoryAlloc(root_graph.get());
  // generate and load task into device