!9548 fix cpu pynative mode mixed precision error
From: @chujinjin Reviewed-by: @jjfeing,@kisnwang Signed-off-by: @kisnwang
This commit is contained in:
commit
a554913704
|
@ -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);
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue