Skip to content

Commit c720c0d

Browse files
committed
cuda.core: bump torch tensor-bridge upper cap to 2.14
1 parent 6de0ff9 commit c720c0d

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

‎cuda_core/cuda/core/_memoryview.pyx‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ cdef dict _torch_type_cache = {}
5151
cdef object _torch_version_ok = None
5252

5353
cdef inline bint _torch_version_check():
54-
"""Return True if 2.3 <= torch <= 2.12 (known AOTI ABI range). Memoized.
54+
"""Return True if 2.3 <= torch <= 2.14 (known AOTI ABI range). Memoized.
5555
5656
Lower bound: AOTI functions we use were introduced in PyTorch 2.3.
5757
Upper bound: the ``pyobj_to_aten_handle`` trick relies on the
@@ -72,7 +72,7 @@ cdef inline bint _torch_version_check():
7272
try:
7373
major, minor = int(torch.__version__.split(".")[0]), \
7474
int(torch.__version__.split(".")[1])
75-
_torch_version_ok = (2, 3) <= (major, minor) <= (2, 12)
75+
_torch_version_ok = (2, 3) <= (major, minor) <= (2, 14)
7676
except (ValueError, IndexError):
7777
_torch_version_ok = False
7878
return <bint>_torch_version_ok

0 commit comments

Comments
 (0)