Touch ups to Array

python/kvikio/kvikio/_lib/arr.pxd

@@ -28,6 +28,9 @@ cdef class Array:
     cpdef Py_ssize_t _nbytes(self)
 
 
+cpdef Array asarray(obj)
+
+
 cdef pair[uintptr_t, size_t] parse_buffer_argument(
     buf, size, bint accept_host_buffer
 ) except *

python/kvikio/kvikio/_lib/arr.pyi

@@ -1,4 +1,4 @@
-# Copyright (c) 2020-2021, NVIDIA CORPORATION. All rights reserved.
+# Copyright (c) 2020-2024, NVIDIA CORPORATION. All rights reserved.
 # See file LICENSE for terms.
 
 from typing import Generic, Tuple, TypeVar

python/kvikio/kvikio/_lib/arr.pyx

@@ -6,11 +6,20 @@
 
 from cpython.array cimport array, newarrayobject
 from cpython.buffer cimport PyBuffer_IsContiguous
-from cpython.memoryview cimport PyMemoryView_FromObject, PyMemoryView_GET_BUFFER
+from cpython.memoryview cimport (
+    PyMemoryView_FromObject,
+    PyMemoryView_GET_BUFFER,
+)
 from cpython.object cimport PyObject
 from cpython.ref cimport Py_INCREF
 from cpython.tuple cimport PyTuple_New, PyTuple_SET_ITEM
-from cython cimport auto_pickle, boundscheck, initializedcheck, nonecheck, wraparound
+from cython cimport (
+    auto_pickle,
+    boundscheck,
+    initializedcheck,
+    nonecheck,
+    wraparound,
+)
 from libc.stdint cimport uintptr_t
 from libc.string cimport memcpy
 
@@ -289,9 +298,10 @@ cdef inline Py_ssize_t _nbytes(Py_ssize_t itemsize,
         nbytes *= shape_mv[i]
     return nbytes
 
-cpdef asarray(obj):
+
+cpdef Array asarray(obj):
     if isinstance(obj, Array):
-        return obj
+        return <Array>obj
     else:
         return Array(obj)