From 6c5e58c5fc0c2cf38d34bf8f7c1217c8e5c5a346 Mon Sep 17 00:00:00 2001
From: Margaret_wangrui <wangrui178@huawei.com>
Date: Wed, 23 Sep 2020 20:06:05 +0800
Subject: [PATCH] fix summarynodes in new control_flow

---
 mindspore/ccsrc/backend/session/ascend_session.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/mindspore/ccsrc/backend/session/ascend_session.cc b/mindspore/ccsrc/backend/session/ascend_session.cc
index 2a662fbe511..a5baddd5178 100644
--- 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