Modify conditions on memory spaces in Distributor

arborx · Oct 6, 2024 · 9b604d6 · 9b604d6
1 parent 2e7ef7b
commit 9b604d6
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/src/distributed/detail/ArborX_Distributor.hpp b/src/distributed/detail/ArborX_Distributor.hpp
@@ -13,6 +13,7 @@
 
 #include <ArborX_Config.hpp>
 
+#include <kokkos_ext/ArborX_KokkosExtAccessibilityTraits.hpp>
 #include <kokkos_ext/ArborX_KokkosExtMinMaxReduce.hpp>
 #include <kokkos_ext/ArborX_KokkosExtViewHelpers.hpp>
 #include <misc/ArborX_Exception.hpp>
@@ -279,6 +280,15 @@ class Distributor
     Kokkos::Profiling::ScopedRegion guard(
         "ArborX::Distributor::doPostsAndWaits");
 
+    static_assert(
+        KokkosExt::is_accessible_from<typename ExportView::memory_space,
+                                      ExecutionSpace>::value);
+    static_assert(
+        KokkosExt::is_accessible_from<typename ImportView::memory_space,
+                                      ExecutionSpace>::value);
+    static_assert(
+        KokkosExt::is_accessible_from<typename decltype(_permute)::memory_space,
+                                      ExecutionSpace>::value);
     static_assert(ExportView::rank == 1 &&
                   (std::is_same_v<typename ExportView::array_layout,
                                   Kokkos::LayoutLeft> ||
@@ -290,12 +300,6 @@ class Distributor
                    std::is_same_v<typename ImportView::array_layout,
                                   Kokkos::LayoutRight>));
 
-    using MemorySpace = typename ExportView::memory_space;
-    static_assert(
-        std::is_same_v<MemorySpace, typename ImportView::memory_space>);
-    static_assert(
-        std::is_same_v<MemorySpace, typename decltype(_permute)::memory_space>);
-
     using ValueType = typename ImportView::value_type;
     static_assert(
         std::is_same<ValueType,