Add KOKKOS_FUNCTION to indexable_get()

aprokop · Jan 24, 2024 · 87b279b · 87b279b
1 parent 410efb6
commit 87b279b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/src/ArborX_BruteForce.hpp b/src/ArborX_BruteForce.hpp
@@ -85,7 +85,10 @@ class BruteForce
         std::forward<View>(view), std::forward<Args>(args)...);
   }
 
-  auto const &indexable_get() const { return _indexable_getter; }
+  KOKKOS_FUNCTION auto const &indexable_get() const
+  {
+    return _indexable_getter;
+  }
 
 private:
   size_type _size{0};

diff --git a/src/ArborX_LinearBVH.hpp b/src/ArborX_LinearBVH.hpp
@@ -121,7 +121,10 @@ class BoundingVolumeHierarchy
     tree_traversal(predicate);
   }
 
-  auto const &indexable_get() const { return _indexable_getter; }
+  KOKKOS_FUNCTION auto const &indexable_get() const
+  {
+    return _indexable_getter;
+  }
 
 private:
   friend struct Details::HappyTreeFriends;