From 353cbc99ff4912a070b84bdf5b7098fd64eb3ef2 Mon Sep 17 00:00:00 2001
From: Zhang Qinghua <zhangqinghua3@huawei.com>
Date: Thu, 30 Apr 2020 10:38:05 +0800
Subject: [PATCH] Remove redundant process of keeping roots.

---
 mindspore/ccsrc/ir/func_graph_cloner.h | 2 +-
 mindspore/ccsrc/optimizer/optimizer.h  | 7 -------
 2 files changed, 1 insertion(+), 8 deletions(-)

diff --git a/mindspore/ccsrc/ir/func_graph_cloner.h b/mindspore/ccsrc/ir/func_graph_cloner.h
index 426cf447a3..10b4b0111e 100644
--- a/mindspore/ccsrc/ir/func_graph_cloner.h
+++ b/mindspore/ccsrc/ir/func_graph_cloner.h
@@ -59,7 +59,7 @@ class Cloner {
 
   // Map of replicate nodes and graphs
   std::unordered_map<AnfNodePtr, AnfNodePtr> *cloned_node() { return &repl_node_; }
-  std::unordered_map<FuncGraphPtr, FuncGraphPtr> cloned_func_graph() { return repl_func_graph_; }
+  std::unordered_map<FuncGraphPtr, FuncGraphPtr> &cloned_func_graph() { return repl_func_graph_; }
 
   // Scope of cloned graphs
   void set_scope(const ScopePtr &scope) { scope_ = scope; }
diff --git a/mindspore/ccsrc/optimizer/optimizer.h b/mindspore/ccsrc/optimizer/optimizer.h
index cadbde0842..1a0ddbc65f 100644
--- a/mindspore/ccsrc/optimizer/optimizer.h
+++ b/mindspore/ccsrc/optimizer/optimizer.h
@@ -185,13 +185,6 @@ class Optimizer : public std::enable_shared_from_this<Optimizer> {
         break;
       }
     }
-
-    auto keep_root = [&func_graph, this]() {
-      std::vector<FuncGraphPtr> func_graphs;
-      func_graphs.push_back(func_graph);
-      resource_->manager()->KeepRoots(func_graphs);
-    };
-    use_profile ? WITH(MsProfile::GetProfile()->Step("keep_roots")) keep_root : keep_root();
     return func_graph;
   }