From aa7b49fe328eec3934b65bd7f089747b3ccecd65 Mon Sep 17 00:00:00 2001
From: simson <wudengsong@huawei.com>
Date: Tue, 23 Mar 2021 10:23:29 +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 fb32f8030b..76c93921b6 100644
--- a/mindspore/ccsrc/backend/session/cpu_session.cc
+++ b/mindspore/ccsrc/backend/session/cpu_session.cc
@@ -229,6 +229,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";
 }