Fix tests for Unified Memory option being on

crtrott · Mar 5, 2024 · 94d624f · 94d624f
1 parent f433545
commit 94d624f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/core/unit_test/cuda/TestCuda_Spaces.cpp b/core/unit_test/cuda/TestCuda_Spaces.cpp
@@ -164,6 +164,10 @@ TEST(cuda, space_access) {
 #ifndef KOKKOS_ENABLE_IMPL_CUDA_UNIFIED_MEMORY
   static_assert(std::is_same<Kokkos::Impl::HostMirror<Kokkos::CudaSpace>::Space,
                              Kokkos::HostSpace>::value);
+#else
+  static_assert(std::is_same<Kokkos::Impl::HostMirror<Kokkos::CudaSpace>::Space,
+                             Kokkos::Device<Kokkos::HostSpace::execution_space,
+                                            Kokkos::CudaSpace>>::value);
 #endif
 
   static_assert(