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

merge lmpfftsettings_kokkos.h into lmpfftsettings.h

Browse Source
This commit is contained in:
Axel Kohlmeyer
2024-01-31 12:54:03 -05:00
parent 23ed9a0980
commit e319555f64
3 changed files with 29 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
src/lmpfftsettings.h
View File

@ -16,7 +16,7 @@
#ifndef LMP_FFT_SETTINGS_H
#define LMP_FFT_SETTINGS_H
// if user set FFTW, it means FFTW3
// if a user sets FFTW, it means FFTW3
#ifdef FFT_FFTW
#ifndef FFT_FFTW3
@ -24,6 +24,14 @@
#endif
#endif
#ifdef LMP_KOKKOS
#ifdef FFT_KOKKOS_FFTW
#ifndef FFT_KOKKOS_FFTW3
#define FFT_KOKKOS_FFTW3
#endif
#endif
#endif
// set strings for library info output
#if defined(FFT_FFTW3)
@ -38,6 +46,20 @@
#define LMP_FFT_LIB "KISS FFT"
#endif
#ifdef LMP_KOKKOS
#if defined(FFT_KOKKOS_FFTW3)
#define LMP_FFT_KOKKOS_LIB "FFTW3"
#elif defined(FFT_KOKKOS_MKL)
#define LMP_FFT_KOKKOS_LIB "MKL FFT"
#elif defined(FFT_KOKKOS_CUFFT)
#define LMP_FFT_KOKKOS_LIB "cuFFT"
#elif defined(FFT_KOKKOS_HIPFFT)
#define LMP_FFT_KOKKOS_LIB "hipFFT"
#else
#define LMP_FFT_KOKKOS_LIB "KISS FFT"
#endif
#endif
#ifdef FFT_SINGLE
typedef float FFT_SCALAR;
#define FFT_PRECISION 1