tx258/lammps-gran-kokkos
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Small tweaks

Browse Source
This commit is contained in:
Stan Moore
2019-07-02 08:27:49 -06:00
parent dcf9351514
commit b1781e5129
2 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/KOKKOS/fft3d_kokkos.cpp
View File

@ -594,8 +594,13 @@ struct fft_plan_3d_kokkos<DeviceType>* FFT3dKokkos<DeviceType>::fft_3d_create_pl
// and scaling normalization
#if defined(FFT_FFTW3)
if (nthreads > 1)
#if defined(FFT_FFTW_THREADS)
if (nthreads > 1) {
fftw_init_threads();
fftw_plan_with_nthreads(nthreads);
}
#endif
plan->plan_fast_forward =
FFTW_API(plan_many_dft)(1, &nfast,plan->total1/plan->length1,
@ -704,7 +709,9 @@ void FFT3dKokkos<DeviceType>::fft_3d_destroy_plan_kokkos(struct fft_plan_3d_kokk
delete plan;
delete remapKK;
#ifdef FFT_KISSFFT
#if defined (FFT_FFTW_THREADS)
fftw_cleanup_threads();
#elif defined (FFT_KISSFFT)
delete kissfftKK;
#endif
}

4
src/KOKKOS/fft3d_kokkos.h
View File

@ -25,6 +25,10 @@
#define FFT_FFTW3
#endif
#ifdef FFT_FFTW_THREADS
#define FFT_FFTW3
#endif
#if defined(FFT_FFTW3)
#include "fftw3.h"
#if defined(FFT_SINGLE)