Add deduction guides for BVH and BruteForce taking in indexable getter

arborx · Oct 14, 2024 · 1648842 · 1648842
1 parent 04453d7
commit 1648842
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/src/spatial/ArborX_BruteForce.hpp b/src/spatial/ArborX_BruteForce.hpp
@@ -136,6 +136,17 @@ KOKKOS_FUNCTION
         typename Details::AccessValues<Values, PrimitivesTag>::memory_space,
         typename Details::AccessValues<Values, PrimitivesTag>::value_type>;
 
+template <typename ExecutionSpace, typename Values, typename IndexableGetter>
+#if KOKKOS_VERSION >= 40400
+KOKKOS_DEDUCTION_GUIDE
+#else
+KOKKOS_FUNCTION
+#endif
+    BruteForce(ExecutionSpace, Values, IndexableGetter) -> BruteForce<
+        typename Details::AccessValues<Values, PrimitivesTag>::memory_space,
+        typename Details::AccessValues<Values, PrimitivesTag>::value_type,
+        IndexableGetter>;
+
 template <typename MemorySpace, typename Value, typename IndexableGetter,
           typename BoundingVolume>
 template <typename ExecutionSpace, typename UserValues>

diff --git a/src/spatial/ArborX_LinearBVH.hpp b/src/spatial/ArborX_LinearBVH.hpp
@@ -178,6 +178,18 @@ KOKKOS_FUNCTION
         typename Details::AccessValues<Values, PrimitivesTag>::memory_space,
         typename Details::AccessValues<Values, PrimitivesTag>::value_type>;
 
+template <typename ExecutionSpace, typename Values, typename IndexableGetter>
+#if KOKKOS_VERSION >= 40400
+KOKKOS_DEDUCTION_GUIDE
+#else
+KOKKOS_FUNCTION
+#endif
+    BoundingVolumeHierarchy(ExecutionSpace, Values, IndexableGetter)
+        -> BoundingVolumeHierarchy<
+            typename Details::AccessValues<Values, PrimitivesTag>::memory_space,
+            typename Details::AccessValues<Values, PrimitivesTag>::value_type,
+            IndexableGetter>;
+
 template <typename MemorySpace, typename Value,
           typename IndexableGetter = Details::DefaultIndexableGetter,
           typename BoundingVolume = Box<