Browse Source

!9548 fix cpu pynative mode mixed precision error

From: @chujinjin
Reviewed-by: @jjfeing,@kisnwang
Signed-off-by: @kisnwang
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
a554913704
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

@@ -164,6 +164,7 @@ void CPUSession::SetOutputFlags(const VectorRef &base_ref, std::vector<tensor::T
} else if (utils::isa<tensor::TensorPtr>(base_ref[i])) { } else if (utils::isa<tensor::TensorPtr>(base_ref[i])) {
auto tensor_ptr = utils::cast<std::shared_ptr<tensor::Tensor>>(base_ref[i]); auto tensor_ptr = utils::cast<std::shared_ptr<tensor::Tensor>>(base_ref[i]);
tensor_ptr->SetNeedWait(false); tensor_ptr->SetNeedWait(false);
tensor_ptr->data_sync(false);
outputs_tensors->push_back(tensor_ptr); outputs_tensors->push_back(tensor_ptr);
} }
} }


Loading…
Cancel
Save