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.
max_version
__dlpack__
1 parent c54a90a commit 9afae01Copy full SHA for 9afae01
dpctl/tensor/_usmarray.pyx
@@ -1127,8 +1127,14 @@ cdef class usm_ndarray:
1127
stream.submit_barrier(dependent_events=[ev])
1128
return _caps
1129
else:
1130
+ if not isinstance(max_version, tuple) or len(max_version) != 2:
1131
+ raise TypeError(
1132
+ "`__dlpack__` expects `max_version` to be a "
1133
+ "2-tuple of integers `(major, minor)`, instead "
1134
+ f"got {type(max_version)}"
1135
+ )
1136
dpctl_dlpack_version = get_build_dlpack_version()
- if max_version >= dpctl_dlpack_version or max_version[0] == dpctl_dlpack_version[0]:
1137
+ if max_version[0] >= dpctl_dlpack_version[0]:
1138
# DLManagedTensorVersioned path
1139
# TODO: add logic for targeting a device
1140
if dl_device is not None:
0 commit comments