Merge pull request ornladios#3571 from anagainaru/hip-adios-view

Allowing Put/Get to receive Kokkos Views with HIP backend
vicentebolea · Mar 31, 2023 · f102c60 · f102c60
2 parents a9dce35 + 60877b3
commit f102c60
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/bindings/CXX11/adios2/cxx11/KokkosView.h b/bindings/CXX11/adios2/cxx11/KokkosView.h
@@ -39,6 +39,26 @@ struct memspace_kokkos_to_adios2<Kokkos::CudaHostPinnedSpace>
 
 #endif
 
+#if defined(KOKKOS_ENABLE_HIP) && defined(ADIOS2_HAVE_GPU_SUPPORT)
+template <>
+struct memspace_kokkos_to_adios2<Kokkos::Experimental::HIPSpace>
+{
+    static constexpr adios2::MemorySpace value = adios2::MemorySpace::GPU;
+};
+
+template <>
+struct memspace_kokkos_to_adios2<Kokkos::Experimental::HIPHostPinnedSpace>
+{
+    static constexpr adios2::MemorySpace value = adios2::MemorySpace::GPU;
+};
+
+template <>
+struct memspace_kokkos_to_adios2<Kokkos::Experimental::HIPManagedSpace>
+{
+    static constexpr adios2::MemorySpace value = adios2::MemorySpace::GPU;
+};
+#endif
+
 } // namespace detail
 
 template <class T, class... Parameters>