From 4b3bee2531c4ab7392437a97bb6955a792ec1765 Mon Sep 17 00:00:00 2001
From: simson <wudengsong@huawei.com>
Date: Tue, 23 Mar 2021 17:05:26 +0800
Subject: [PATCH] clear memory after run op in CPU

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

diff --git a/mindspore/ccsrc/backend/session/cpu_session.cc b/mindspore/ccsrc/backend/session/cpu_session.cc
index d7a44dbdaf..fb3b6f171f 100644
--- a/mindspore/ccsrc/backend/session/cpu_session.cc
+++ b/mindspore/ccsrc/backend/session/cpu_session.cc
@@ -220,6 +220,7 @@ void CPUSession::RunOpImpl(const GraphInfo &graph_info, OpRunInfo *op_run_info,
 
   std::vector<tensor::TensorPtr> output_tensors;
   SetOutputFlags(*outputs, &output_tensors);
+  runtime_.RunOpClearMemory(kernel_graph.get());
   MS_LOG(INFO) << "Run Op end";
 }