From 333dd32dc119d931072c7de9a7d47d079a2cc745 Mon Sep 17 00:00:00 2001 From: Leo Fang Date: Thu, 12 Dec 2024 20:00:58 +0000 Subject: [PATCH] defer check for versioned --- cuda_core/cuda/core/experimental/_memoryview.pyx | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/cuda_core/cuda/core/experimental/_memoryview.pyx b/cuda_core/cuda/core/experimental/_memoryview.pyx index d8eba464a..0b0331468 100644 --- a/cuda_core/cuda/core/experimental/_memoryview.pyx +++ b/cuda_core/cuda/core/experimental/_memoryview.pyx @@ -180,21 +180,21 @@ cdef StridedMemoryView view_as_dlpack(obj, stream_ptr, view=None): capsule = obj.__dlpack__( stream=stream_ptr, max_version=(DLPACK_MAJOR_VERSION, DLPACK_MINOR_VERSION)) - versioned = True except TypeError: capsule = obj.__dlpack__( stream=stream_ptr) - versioned = False cdef void* data = NULL - if versioned and cpython.PyCapsule_IsValid( + if cpython.PyCapsule_IsValid( capsule, DLPACK_VERSIONED_TENSOR_UNUSED_NAME): data = cpython.PyCapsule_GetPointer( capsule, DLPACK_VERSIONED_TENSOR_UNUSED_NAME) - elif not versioned and cpython.PyCapsule_IsValid( + versioned = True + elif cpython.PyCapsule_IsValid( capsule, DLPACK_TENSOR_UNUSED_NAME): data = cpython.PyCapsule_GetPointer( capsule, DLPACK_TENSOR_UNUSED_NAME) + versioned = False else: assert False