p85620714
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix summarynodes in new control_flow

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

@ -177,6 +177,7 @@ GraphId AscendSession::CompileGraph(NotNull<FuncGraphPtr> func_graph) {
if (debugger_) { if (debugger_) {
debugger_->PreExecute(root_graph); debugger_->PreExecute(root_graph);
} }
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