saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-12 17:57:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

error handling ffor eager mode's data transfer (#9261)

Browse source
This commit is contained in:
Tang, Cheng 2021-10-07 17:16:33 -07:00 committed by GitHub
parent 70cf61fa84
commit 68601fc296
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
orttraining/orttraining/eager/ort_ops.cpp
View file

@ -24,9 +24,7 @@ OrtValue reshape_copy(
auto* ort_shape_tensor = shape_tensor.GetMutable<onnxruntime::Tensor>();
CopyVectorToTensor<int64_t>(invoker, new_shape, *ort_shape_tensor);
std::vector<OrtValue> result(1);
auto status = invoker.Invoke("Reshape", {input, shape_tensor}, result, nullptr);
if (!status.IsOK())
throw std::runtime_error("ORT return failure status: " + status.ErrorMessage());
ORT_THROW_IF_ERROR(invoker.Invoke("Reshape", {input, shape_tensor}, result, nullptr));
return result[0];
}
@ -38,7 +36,7 @@ void copy(onnxruntime::ORTInvoker& invoker,
auto* dst_tensor = dst.GetMutable<onnxruntime::Tensor>();
if (!dst_tensor)
throw std::runtime_error("ORT copy: dst is not a tensor");
ort_ep.GetDataTransfer()->CopyTensor(src_tensor, *dst_tensor);
ORT_THROW_IF_ERROR(ort_ep.GetDataTransfer()->CopyTensor(src_tensor, *dst_tensor));
}
} // namespace eager