feat: add size property to StridedMemoryView (NVIDIA#1363)

cpcloud · web-flow · commit 4f910cb5f7ea · 2025-12-12T20:58:26.000Z
diff --git a/cuda_core/cuda/core/experimental/_memoryview.pyx b/cuda_core/cuda/core/experimental/_memoryview.pyx
@@ -296,6 +296,10 @@ cdef class StridedMemoryView:
         """
         return self.get_layout()
 
+    @property
+    def size(self) -> int:
+        return self.get_layout().get_volume()
+
     @property
     def shape(self) -> tuple[int, ...]:
         """
diff --git a/cuda_core/tests/test_utils.py b/cuda_core/tests/test_utils.py
@@ -92,6 +92,7 @@ def _check_view(self, view, in_arr):
         assert isinstance(view, StridedMemoryView)
         assert view.ptr == in_arr.ctypes.data
         assert view.shape == in_arr.shape
+        assert view.size == in_arr.size
         strides_in_counts = convert_strides_to_counts(in_arr.strides, in_arr.dtype.itemsize)
         if in_arr.flags.c_contiguous:
             assert view.strides is None
@@ -172,6 +173,7 @@ def _check_view(self, view, in_arr, dev):
         assert isinstance(view, StridedMemoryView)
         assert view.ptr == gpu_array_ptr(in_arr)
         assert view.shape == in_arr.shape
+        assert view.size == in_arr.size
         strides_in_counts = convert_strides_to_counts(in_arr.strides, in_arr.dtype.itemsize)
         if in_arr.flags["C_CONTIGUOUS"]:
             assert view.strides in (None, strides_in_counts)
@@ -204,6 +206,7 @@ def _check_view(self, view, in_arr, dev):
         assert isinstance(view, StridedMemoryView)
         assert view.ptr == gpu_array_ptr(in_arr)
         assert view.shape == in_arr.shape
+        assert view.size == in_arr.size
         strides_in_counts = convert_strides_to_counts(in_arr.strides, in_arr.dtype.itemsize)
         if in_arr.flags["C_CONTIGUOUS"]:
             assert view.strides is None
