Need to update define for HIP unified memory

2024-12-13 11:45:12 -08:00
parent 4c347a719e
commit f557b7b0e5
1 changed files with 1 additions and 1 deletions
--- a/src/KOKKOS/kokkos_type.h
+++ b/src/KOKKOS/kokkos_type.h
@ -159,7 +159,7 @@ template<class DeviceType>
 class KKDevice {
 public:
 #if ((defined(KOKKOS_ENABLE_CUDA) && defined(KOKKOS_ENABLE_CUDA_UVM)) || \
-     (defined(KOKKOS_ENABLE_HIP) && defined(KOKKOS_ENABLE_IMPL_HIP_UNIFIED_MEMORY)))
+     (defined(KOKKOS_ENABLE_HIP) && defined(KOKKOS_ARCH_AMD_GFX942_APU)))
  typedef Kokkos::Device<DeviceType,LMPDeviceType::memory_space> value;
 #else
  typedef Kokkos::Device<DeviceType,typename DeviceType::memory_space> value;