huawei
/
mindspore2022
0
0
Fork 53
Code Issues Pull Requests 29 Projects Releases Wiki Activity

fix SummaryNodes in new control_flow

This commit is contained in:
Margaret_wangrui 2020-09-28 11:11:48 +08:00
parent 3638886ec6
commit 4df609c94e
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mindspore/ccsrc/backend/session/ascend_session.cc
View File

@ -179,6 +179,7 @@ GraphId AscendSession::CompileGraph(NotNull<FuncGraphPtr> func_graph) {
debugger_->PreExecute(root_graph); debugger_->PreExecute(root_graph);
} }
#endif #endif
SetSummaryNodes(root_graph.get());
// alloc mem // alloc mem
MemoryAlloc(root_graph.get()); MemoryAlloc(root_graph.get());
// generate and load task into device // generate and load task into device