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