Skip to content

Commit

Permalink
Fix fmt
Browse files Browse the repository at this point in the history
  • Loading branch information
take-cheeze committed Dec 16, 2024
1 parent 596d5c2 commit 36f7d52
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions onnxruntime/python/onnxruntime_pybind_ortvalue.cc
Original file line number Diff line number Diff line change
Expand Up @@ -361,8 +361,7 @@ void addOrtValueMethods(pybind11::module& m) {
ORT_ENFORCE(ort_value->IsTensor(), "Only tensor type OrtValues are supported");
const onnxruntime::Tensor& tensor = ort_value->Get<Tensor>();
DLDevice device = onnxruntime::dlpack::GetDlpackDevice(*ort_value, tensor.Location().device.Id());
return py::make_tuple(static_cast<int>(device.device_type), device.device_id); }, "Returns a tuple of integers, (device, device index) (part of __dlpack__ protocol).")
;
return py::make_tuple(static_cast<int>(device.device_type), device.device_id); }, "Returns a tuple of integers, (device, device index) (part of __dlpack__ protocol).");

py::class_<std::vector<OrtValue>>(m, "OrtValueVector")
.def(py::init<>())
Expand Down

0 comments on commit 36f7d52

Please sign in to comment.