forked from mindspore-Ecosystem/mindspore
!13800 clear memory after run op in CPU
From: @simson_wu Reviewed-by: @chujinjin,@zhoufeng54 Signed-off-by: @chujinjin
This commit is contained in:
commit
ad4dedb1db
|
@ -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…
Reference in New Issue