Add carray_int64_t_to_tuple function

Author: mdboom

cuda_core/cuda/core/experimental/_memoryview.pyx

@@ -12,6 +12,7 @@ from typing import Any, Optional
 import numpy
 
 from cuda.core.experimental._utils.cuda_utils import handle_return, driver
+from cuda.core.experimental._utils cimport cuda_utils
 
 
 # TODO(leofang): support NumPy structured dtypes
@@ -213,13 +214,9 @@ cdef StridedMemoryView view_as_dlpack(obj, stream_ptr, view=None):
     buf.ptr = <intptr_t>(dl_tensor.data)
 
     # Construct shape and strides tuples using the Python/C API for speed
-    buf.shape = cpython.PyTuple_New(dl_tensor.ndim)
-    for i in range(dl_tensor.ndim):
-        cpython.PyTuple_SET_ITEM(buf.shape, i, cpython.PyLong_FromLong(dl_tensor.shape[i]))
+    buf.shape = cuda_utils.carray_int64_t_to_tuple(dl_tensor.shape, dl_tensor.ndim)
     if dl_tensor.strides:
-        buf.strides = cpython.PyTuple_New(dl_tensor.ndim)
-        for i in range(dl_tensor.ndim):
-            cpython.PyTuple_SET_ITEM(buf.strides, i, cpython.PyLong_FromLong(dl_tensor.strides[i]))
+        buf.strides = cuda_utils.carray_int64_t_to_tuple(dl_tensor.strides, dl_tensor.ndim)
     else:
         # C-order
         buf.strides = None

cuda_core/cuda/core/experimental/_utils/cuda_utils.pxd

@@ -2,7 +2,19 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
+
+cimport cpython
+cimport libc.stdint
+
+
 cpdef int _check_driver_error(error) except?-1
 cpdef int _check_runtime_error(error) except?-1
 cpdef int _check_nvrtc_error(error) except?-1
 cpdef check_or_create_options(type cls, options, str options_description=*, bint keep_none=*)
+
+
+cdef inline tuple carray_int64_t_to_tuple(libc.stdint.int64_t *ptr, int length):
+    result = cpython.PyTuple_New(length)
+    for i in range(length):
+        cpython.PyTuple_SET_ITEM(result, i, cpython.PyLong_FromLongLong(ptr[i]))
+    return result