Merge pull request #65 from Rombur/hip_support

Add support for HIP
kokkos · May 14, 2024 · 6c6a25a · 6c6a25a
2 parents 967d3de + 0b28a56
commit 6c6a25a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/flcl-cxx.hpp b/src/flcl-cxx.hpp
@@ -58,6 +58,8 @@ typedef struct _flcl_nd_array_t {
 namespace flcl {
 #ifdef KOKKOS_ENABLE_CUDA
 using HostMemorySpace = Kokkos::CudaUVMSpace;
+#elif defined(KOKKOS_ENABLE_HIP)
+using HostMemorySpace = Kokkos::HIPManagedSpace;
 #else
 using HostMemorySpace = Kokkos::HostSpace;
 #endif
@@ -75,6 +77,8 @@ namespace flcl {
 
   #ifdef KOKKOS_ENABLE_CUDA
     using HostMemorySpace = Kokkos::CudaUVMSpace;
+  #elif defined(KOKKOS_ENABLE_HIP)
+    using HostMemorySpace = Kokkos::HIPManagedSpace;
   #else
     using HostMemorySpace = Kokkos::HostSpace;
   #endif