From 4faadcfec6bbb21e4a99792b444a0c676e05f1dd Mon Sep 17 00:00:00 2001
From: tanghuikang <tanghuikang@huawei.com>
Date: Thu, 25 Feb 2021 21:51:20 +0800
Subject: [PATCH] Clear memory in singlel op graph after pre build

---
 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 54aad0acd6..118ef9998f 100644
--- a/mindspore/ccsrc/backend/session/ascend_session.cc
+++ b/mindspore/ccsrc/backend/session/ascend_session.cc
@@ -760,6 +760,7 @@ void AscendSession::BuildOpsInGraph(const GraphId &graph_id, const std::map<AnfN
   BuildKernel(kernels);
   // Record single op graphs in run_op_graphs_ so that these graphs can be reused in BuildOpImpl
   for (const auto &single_op_graph : single_op_graphs) {
+    RunOpMemoryClear(single_op_graph.first.get());
     for (const auto &graph_info : single_op_graph.second) {
       run_op_graphs_[graph_info] = single_op_graph.first;
       MS_LOG(DEBUG) << "Pre build op finished, graph info: " << graph_info;