diff --git a/mindspore/ccsrc/backend/session/cpu_session.cc b/mindspore/ccsrc/backend/session/cpu_session.cc index 4c196f0a91..9b789e2c25 100644 --- a/mindspore/ccsrc/backend/session/cpu_session.cc +++ b/mindspore/ccsrc/backend/session/cpu_session.cc @@ -164,6 +164,7 @@ void CPUSession::SetOutputFlags(const VectorRef &base_ref, std::vector(base_ref[i])) { auto tensor_ptr = utils::cast>(base_ref[i]); tensor_ptr->SetNeedWait(false); + tensor_ptr->data_sync(false); outputs_tensors->push_back(tensor_ptr); } }