We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 74aa500 commit a441ffcCopy full SHA for a441ffc
onnx_diagnostic/ort_session.py
@@ -439,6 +439,8 @@ def run_dlpack(
439
new_feeds = {}
440
for k, v in feeds.items():
441
assert hasattr(v, "__dlpack__"), f"class {type(v)} should be serialized"
442
+ if not v.is_contiguous():
443
+ v = v.contiguous()
444
new_feeds[k] = ORTC.OrtValue.from_dlpack(v.__dlpack__(), v.dtype == torch.bool)
445
if self.nvtx:
446
self.torch.cuda.nvtx.range_push("run_with_ort_values")
0 commit comments