File tree Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Original file line number Diff line number Diff line change @@ -160,8 +160,10 @@ cdef void _pycapsule_versioned_deleter(object dlt_capsule) noexcept:
160
160
161
161
cdef void _managed_tensor_versioned_deleter(DLManagedTensorVersioned * dlmv_tensor) noexcept with gil:
162
162
if dlmv_tensor is not NULL :
163
+ # we only delete shape, because we make single allocation to
164
+ # acommodate both shape and strides if strides are needed
163
165
stdlib.free(dlmv_tensor.dl_tensor.shape)
164
- cpython.Py_DECREF(< usm_ndarray > dlmv_tensor.manager_ctx)
166
+ cpython.Py_DECREF(< object > dlmv_tensor.manager_ctx)
165
167
dlmv_tensor.manager_ctx = NULL
166
168
stdlib.free(dlmv_tensor)
167
169
@@ -543,14 +545,15 @@ cdef class _DLManagedTensorVersionedOwner:
543
545
Helper class managing the lifetime of the DLManagedTensorVersioned
544
546
struct transferred from a 'dlpack_versioned' capsule.
545
547
"""
546
- cdef DLManagedTensorVersioned * dlmv_tensor
548
+ cdef DLManagedTensorVersioned * dlmv_tensor
547
549
548
550
def __cinit__ (self ):
549
551
self .dlmv_tensor = NULL
550
552
551
553
def __dealloc__ (self ):
552
554
if self .dlmv_tensor:
553
555
self .dlmv_tensor.deleter(self .dlmv_tensor)
556
+ self .dlmv_tensor = NULL
554
557
555
558
@staticmethod
556
559
cdef _DLManagedTensorVersionedOwner _create(DLManagedTensorVersioned * dlmv_tensor_src):
You can’t perform that action at this time.
0 commit comments