ICODE-ADC/lammps
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Need to update define for HIP unified memory

Browse Source
This commit is contained in:
Stan Moore
2024-12-13 11:45:12 -08:00
parent 4c347a719e
commit f557b7b0e5
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/KOKKOS/kokkos_type.h
View File

@ -159,7 +159,7 @@ template<class DeviceType>
class KKDevice { class KKDevice {
public: public:
#if ((defined(KOKKOS_ENABLE_CUDA) && defined(KOKKOS_ENABLE_CUDA_UVM)) || \ #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; typedef Kokkos::Device<DeviceType,LMPDeviceType::memory_space> value;
#else #else
typedef Kokkos::Device<DeviceType,typename DeviceType::memory_space> value; typedef Kokkos::Device<DeviceType,typename DeviceType::memory_space> value;