Skip to content

Commit 417e7d9

Browse files
committed
fix ort_session
1 parent a441ffc commit 417e7d9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

onnx_diagnostic/ort_session.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -241,7 +241,7 @@ def _ortvalues_to_numpy_tensor(
241241
el_type = ortvalues[i].element_type()
242242
if el_type < onnx.TensorProto.BFLOAT16:
243243
try:
244-
a = np.from_dlpack(ortvalues[i].to_dlpack())
244+
a = np.from_dlpack(ortvalues[i])
245245
except RuntimeError as e:
246246
assert "ORT only supports contiguous tensor for now." in str(e), (
247247
f"As it says, non-contiguous OrtValue are not supported "

0 commit comments

Comments
 (0)