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 output_tensors; SetOutputFlags(*outputs, &output_tensors); + runtime_.RunOpClearMemory(kernel_graph.get()); MS_LOG(INFO) << "Run Op end"; }