Add support for hipFFT backend in PPPMKokkos.

2022-04-20 12:06:38 +10:00
parent 691ba89b6f
commit 8cdbf380cf
6 changed files with 84 additions and 4 deletions
--- a/cmake/Modules/Packages/KOKKOS.cmake
+++ b/cmake/Modules/Packages/KOKKOS.cmake
@ -130,6 +130,11 @@ if(PKG_KSPACE)
      target_compile_definitions(lammps PRIVATE -DFFT_CUFFT)
      target_link_libraries(lammps PRIVATE cufft)
    endif()
+  elseif(Kokkos_ENABLE_HIP)
+    if(NOT (FFT STREQUAL "KISS"))
+      target_compile_definitions(lammps PRIVATE -DFFT_HIPFFT)
+      target_link_libraries(lammps PRIVATE hipfft)
+    endif()
  endif()
 endif()