fix cpu pynative mixed precision error

4 years ago · 02c8052e89
parent 3b7acc57c9
commit 02c8052e89
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
@ -163,6 +163,7 @@ void CPUSession::SetOutputFlags(const VectorRef &base_ref, std::vector<tensor::T
    } else if (utils::isa<tensor::TensorPtr>(base_ref[i])) {
      auto tensor_ptr = utils::cast<std::shared_ptr<tensor::Tensor>>(base_ref[i]);
      tensor_ptr->SetNeedWait(false);
+      tensor_ptr->data_sync(false);
      outputs_tensors->push_back(tensor_ptr);
    }
  }