!9548 fix cpu pynative mode mixed precision error

From: @chujinjin
Reviewed-by: @jjfeing,@kisnwang
Signed-off-by: @kisnwang
This commit is contained in:
mindspore-ci-bot 2020-12-05 15:53:23 +08:00 committed by Gitee
commit a554913704
1 changed files with 1 additions and 0 deletions

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])) {
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);
}
}