Browse Source

fix summarynodes in new control_flow

tags/v1.1.0
Margaret_wangrui 5 years ago
parent
commit
6c5e58c5fc
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      mindspore/ccsrc/backend/session/ascend_session.cc

+ 1
- 0
mindspore/ccsrc/backend/session/ascend_session.cc View File

@@ -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


Loading…
Cancel
Save