Skip to content

Commit c59362f

Browse files
committed
usm_ary -> ary_base in to_dlpack_versioned_capsule
1 parent df168d2 commit c59362f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

dpctl/tensor/_dlpack.pyx

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -498,8 +498,8 @@ cpdef to_dlpack_versioned_capsule(usm_ndarray usm_ary, bint copied):
498498
dlmv_tensor.version.major = DLPACK_MAJOR_VERSION
499499
dlmv_tensor.version.minor = DLPACK_MINOR_VERSION
500500

501-
dlmv_tensor.manager_ctx = <void*>usm_ary
502-
cpython.Py_INCREF(usm_ary)
501+
dlmv_tensor.manager_ctx = <void*>ary_base
502+
cpython.Py_INCREF(ary_base)
503503
dlmv_tensor.deleter = _managed_tensor_versioned_deleter
504504

505505
return cpython.PyCapsule_New(dlmv_tensor, 'dltensor_versioned', _pycapsule_versioned_deleter)

0 commit comments

Comments
 (0)