From 9b604d66e042fcfed2ba8c117fd9ea9cbbcda5ab Mon Sep 17 00:00:00 2001
From: Andrey Prokopenko <prokopenkoav@ornl.gov>
Date: Sat, 5 Oct 2024 10:21:36 -0400
Subject: [PATCH] Modify conditions on memory spaces in Distributor

---
 src/distributed/detail/ArborX_Distributor.hpp | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/src/distributed/detail/ArborX_Distributor.hpp b/src/distributed/detail/ArborX_Distributor.hpp
index f8e675109..a7637f020 100644
--- 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,