Bundle tiled TensorMap options and type retained views.

Centralize the tiled descriptor arguments in an options object, keep dtype-like inputs on the public path while using raw driver values internally, and declare StridedMemoryView in a pxd so retained views stay typed without extra helper indirection.

Made-with: Cursor

Author: cpcloud

cuda_core/cuda/core/_memoryview.pxd

@@ -0,0 +1,24 @@
+from libc.stdint cimport intptr_t
+
+from cuda.core._dlpack cimport DLTensor
+from cuda.core._layout cimport _StridedLayout
+
+
+cdef class StridedMemoryView:
+    cdef readonly:
+        intptr_t ptr
+        int device_id
+        bint is_device_accessible
+        bint readonly
+        object exporting_obj
+
+    cdef:
+        object metadata
+        DLTensor* dl_tensor
+        _StridedLayout _layout
+        object _buffer
+        object _dtype
+
+    cdef inline _StridedLayout get_layout(self)
+    cdef inline object get_buffer(self)
+    cdef inline object get_dtype(self)

cuda_core/cuda/core/_memoryview.pyx

@@ -107,35 +107,6 @@ cdef class StridedMemoryView:
         it will be the Buffer instance passed to the method.
 
     """
-    cdef readonly:
-        intptr_t ptr
-        int device_id
-        bint is_device_accessible
-        bint readonly
-        object exporting_obj
-
-    cdef:
-        # If using dlpack, this is a strong reference to the result of
-        # obj.__dlpack__() so we can lazily create shape and strides from
-        # it later.  If using CAI, this is a reference to the source
-        # `__cuda_array_interface__` object.
-        object metadata
-
-        # The tensor object if has obj has __dlpack__, otherwise must be NULL
-        DLTensor *dl_tensor
-
-        # Memoized properties
-        # Either lazily inferred from dl_tensor/metadata,
-        # or explicitly provided if created with from_buffer().
-        _StridedLayout _layout
-        # Either exporting_obj if it is a Buffer, otherwise a Buffer instance
-        # with owner set to the exporting object.
-        object _buffer
-        # Either lazily inferred from dl_tensor/metadata,
-        # or explicitly provided if created with from_buffer().
-        # In the latter case, it can be None.
-        object _dtype
-
     def __init__(self, obj: object = None, stream_ptr: int | None = None) -> None:
         cdef str clsname = self.__class__.__name__
         if obj is not None:
@@ -318,8 +289,9 @@ cdef class StridedMemoryView:
 
     def as_tensor_map(
         self,
-        box_dim,
+        box_dim=None,
         *,
+        options=None,
         element_strides=None,
         data_type=None,
         interleave=None,
@@ -330,11 +302,15 @@ cdef class StridedMemoryView:
         """Create a tiled :obj:`TensorMapDescriptor` from this view.
 
         This is the public entry point for creating tiled tensor map
-        descriptors in ``cuda.core``.
+        descriptors in ``cuda.core``. Pass either ``box_dim`` and the
+        individual keyword arguments directly, or provide bundled tiled
+        options via ``options=``.
         """
         from cuda.core._tensor_map import TensorMapDescriptor
 
         kwargs = {}
+        if options is not None:
+            kwargs["options"] = options
         if element_strides is not None:
             kwargs["element_strides"] = element_strides
         if data_type is not None:

cuda_core/cuda/core/_tensor_map.pxd

@@ -4,14 +4,15 @@
 
 from cuda.bindings cimport cydriver
 from libc.stdint cimport intptr_t
+from cuda.core._memoryview cimport StridedMemoryView
 
 
 cdef class TensorMapDescriptor:
     cdef cydriver.CUtensorMap _tensor_map
     cdef int _device_id
     cdef intptr_t _context
     cdef object _source_ref
-    cdef object _view_ref
+    cdef StridedMemoryView _view_ref
     cdef object _repr_info
 
     cdef int _check_context_compat(self) except -1