!6802 fix summarynodes in new control_flow

Merge pull request !6802 from Margaret_wangrui/fix_summarynodes_in_new_control_flow
4 years ago · eeb707a7db
parent 7f390467e9 6c5e58c5fc
commit eeb707a7db
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
@ -177,6 +177,7 @@ GraphId AscendSession::CompileGraph(NotNull<FuncGraphPtr> func_graph) {
  if (debugger_) {
    debugger_->PreExecute(root_graph);
  }
+  SetSummaryNodes(root_graph.get());
  // alloc mem
  MemoryAlloc(root_graph.get());
  // generate and load task into device