Loading src/KOKKOS/fft3d_kokkos.cpp +9 −2 Original line number Diff line number Diff line Loading @@ -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, Loading Loading @@ -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 } Loading src/KOKKOS/fft3d_kokkos.h +4 −0 Original line number Diff line number Diff line Loading @@ -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) Loading Loading
src/KOKKOS/fft3d_kokkos.cpp +9 −2 Original line number Diff line number Diff line Loading @@ -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, Loading Loading @@ -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 } Loading
src/KOKKOS/fft3d_kokkos.h +4 −0 Original line number Diff line number Diff line Loading @@ -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) Loading
