add std::dims

NVIDIA · Nov 25, 2024 · da1ffe0 · da1ffe0
1 parent 83d180f
commit da1ffe0
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/docs/libcudacxx/standard_api.rst b/docs/libcudacxx/standard_api.rst
@@ -101,5 +101,7 @@ Feature availability:
 
 -  C++23 ``<mdspan>`` is available in C++17.
 
-   -  mdspan is feature complete in C++17 onwards.
-   -  mdspan on msvc is only supported in C++20 and onwards.
+   -  ``mdspan`` is feature complete in C++17 onwards.
+   -  ``mdspan`` on msvc is only supported in C++20 and onwards.
+
+-  C++26 ``std::dims`` is available in C++17.
diff --git a/docs/libcudacxx/standard_api/container_library/mdspan.rst b/docs/libcudacxx/standard_api/container_library/mdspan.rst
@@ -7,6 +7,7 @@ Extensions
 ----------
 
 -  All features of ``<mdspan>`` are made available in C++17 onwards
+-  C++26 ``std::dims`` is made available in C++17 onwards
 
 Restrictions
 ------------

diff --git a/libcudacxx/include/cuda/std/__mdspan/extents.h b/libcudacxx/include/cuda/std/__mdspan/extents.h
@@ -525,6 +525,9 @@ struct __make_dextents<_IndexType, 0, _CUDA_VSTD::extents<_IndexType, _ExtentsPa
 template <class _IndexType, size_t _Rank>
 using dextents = typename __detail::__make_dextents<_IndexType, _Rank>::type;
 
+template <size_t _Rank, class _IndexType = size_t>
+using dims = dextents<_IndexType, _Rank>;
+
 #  if defined(__MDSPAN_USE_CLASS_TEMPLATE_ARGUMENT_DEDUCTION)
 template <class... _IndexTypes>
 _CCCL_HOST_DEVICE extents(_IndexTypes...)
-Original file line number
+Diff line change
@@ Expand Up / @@ -7,6 +7,7 @@ Extensions @@
     ----------
     -  All features of ``<mdspan>`` are made available in C++17 onwards
+    -  C++26 ``std::dims`` is made available in C++17 onwards
     Restrictions
     ------------
@@ Expand Down @@