Renamed and test CheckedRelaxedAtomicAccessor

crtrott · Oct 2, 2024 · 542d7ae · 542d7ae
1 parent 817843d
commit 542d7ae
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/core/src/View/MDSpan/Kokkos_MDSpan_Accessor.hpp b/core/src/View/MDSpan/Kokkos_MDSpan_Accessor.hpp
@@ -396,7 +396,7 @@ using CheckedReferenceCountedAccessor =
 
 template <class ElementType, class MemorySpace,
           class MemoryScope = desul::MemoryScopeDevice>
-using checked_atomic_accessor_relaxed =
+using CheckedRelaxedAtomicAccessor =
     SpaceAwareAccessor<MemorySpace, AtomicAccessorRelaxed<ElementType>>;
 
 template <class ElementType, class MemorySpace,

diff --git a/core/unit_test/view/TestBasicView.hpp b/core/unit_test/view/TestBasicView.hpp
@@ -232,9 +232,14 @@ void test_atomic_accessor() {
           T, typename ExecutionSpace::memory_space>;
   using view_type =
       Kokkos::BasicView<T, extents_type, layout_type, accessor_type>;
+  using um_accessor_type = Kokkos::Impl::CheckedRelaxedAtomicAccessor<
+      T, typename ExecutionSpace::memory_space>;
+  using um_view_type =
+      Kokkos::BasicView<T, extents_type, layout_type, um_accessor_type>;
 
   extents_type extents{};
   auto view = view_type("test_view", extents);
+  um_view_type um_view(view);
 
   EXPECT_TRUE(view.data_handle().has_record());
   EXPECT_NE(view.data_handle().get(), nullptr);
@@ -243,6 +248,8 @@ void test_atomic_accessor() {
       make_spanning_mdrange_policy_from_extents<ExecutionSpace>(extents);
 
   Kokkos::parallel_for(mdrange_policy, MDRangeTestFunctor<view_type>{view});
+  Kokkos::parallel_for(mdrange_policy,
+                       MDRangeTestFunctor<um_view_type>{um_view});
 }
 
 TEST(TEST_CATEGORY, basic_view_atomic_accessor) {