Browse Source

!13800 clear memory after run op in CPU

From: @simson_wu
Reviewed-by: @chujinjin,@zhoufeng54
Signed-off-by: @chujinjin
pull/13800/MERGE
mindspore-ci-bot Gitee 5 years ago
parent
commit
ad4dedb1db
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      mindspore/ccsrc/backend/session/cpu_session.cc

+ 1
- 0
mindspore/ccsrc/backend/session/cpu_session.cc View File

@@ -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";
}



Loading…
Cancel
Save