Add a CTAD to ChunkSpan from a Chunk

CExA-project · Dec 13, 2024 · 3621b75 · 3621b75
1 parent 0f8473c
commit 3621b75
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 0 deletions.
diff --git a/include/ddc/chunk_span.hpp b/include/ddc/chunk_span.hpp
@@ -387,6 +387,22 @@ ChunkSpan(KokkosView const& view, DiscreteDomain<DDims...> domain)
                 detail::kokkos_to_mdspan_layout_t<typename KokkosView::array_layout>,
                 typename KokkosView::memory_space>;
 
+template <class ElementType, class SupportType, class Allocator>
+ChunkSpan(Chunk<ElementType, SupportType, Allocator>& other)
+        -> ChunkSpan<
+                ElementType,
+                SupportType,
+                Kokkos::layout_right,
+                typename Allocator::memory_space>;
+
+template <class ElementType, class SupportType, class Allocator>
+ChunkSpan(Chunk<ElementType, SupportType, Allocator> const& other)
+        -> ChunkSpan<
+                const ElementType,
+                SupportType,
+                Kokkos::layout_right,
+                typename Allocator::memory_space>;
+
 template <
         class ElementType,
         class SupportType,

diff --git a/tests/chunk_span.cpp b/tests/chunk_span.cpp
@@ -3,6 +3,7 @@
 // SPDX-License-Identifier: MIT
 
 #include <type_traits>
+#include <utility>
 
 #include <ddc/ddc.hpp>
 
@@ -33,3 +34,14 @@ TEST(ChunkSpan1DTest, ConstructionFromChunk)
     EXPECT_TRUE((std::is_constructible_v<ChunkSpanX<const double>, ChunkX<double>&>));
     EXPECT_TRUE((std::is_constructible_v<ChunkSpanX<const double>, ChunkX<double> const&>));
 }
+
+TEST(ChunkSpan1DTest, CTAD)
+{
+    ChunkX<int> chk("chk", DDomX(DElemX(0), DVectX(0)));
+    EXPECT_TRUE((std::is_same_v<
+                 decltype(ddc::ChunkSpan(chk)),
+                 ddc::ChunkSpan<int, DDomX, Kokkos::layout_right, Kokkos::HostSpace>>));
+    EXPECT_TRUE((std::is_same_v<
+                 decltype(ddc::ChunkSpan(std::as_const(chk))),
+                 ddc::ChunkSpan<const int, DDomX, Kokkos::layout_right, Kokkos::HostSpace>>));
+}