fix summarynodes in new control_flow

2020-09-23 20:06:05 +08:00 · 2020-09-23 20:06:05 +08:00 · 6c5e58c5fc
parent 8d20d1b8e6
commit 6c5e58c5fc
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