From 02c8052e897c2dbc31f5e9d41d7557ca6698420b Mon Sep 17 00:00:00 2001 From: chujinjin Date: Sat, 5 Dec 2020 14:06:15 +0800 Subject: [PATCH] fix cpu pynative mixed precision error --- mindspore/ccsrc/backend/session/cpu_session.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/mindspore/ccsrc/backend/session/cpu_session.cc b/mindspore/ccsrc/backend/session/cpu_session.cc index 5fda81d88e..f2287c0d28 100644 --- a/mindspore/ccsrc/backend/session/cpu_session.cc +++ b/mindspore/ccsrc/backend/session/cpu_session.cc @@ -163,6 +163,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); } }