clear memory after run op in CPU

4 years ago · aa7b49fe32
parent ab5d77f943
commit aa7b49fe32
1 changed files with 1 additions and 0 deletions
--- 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";
 }