aboutsummaryrefslogtreecommitdiff
path: root/Source/FieldSolver/SpectralSolver
diff options
context:
space:
mode:
Diffstat (limited to 'Source/FieldSolver/SpectralSolver')
-rw-r--r--Source/FieldSolver/SpectralSolver/SpectralFieldData.cpp4
-rw-r--r--Source/FieldSolver/SpectralSolver/WrapFFTW.cpp12
2 files changed, 15 insertions, 1 deletions
diff --git a/Source/FieldSolver/SpectralSolver/SpectralFieldData.cpp b/Source/FieldSolver/SpectralSolver/SpectralFieldData.cpp
index bdb631063..056c030c0 100644
--- a/Source/FieldSolver/SpectralSolver/SpectralFieldData.cpp
+++ b/Source/FieldSolver/SpectralSolver/SpectralFieldData.cpp
@@ -145,6 +145,8 @@ SpectralFieldData::ForwardTransform (const int lev,
#endif
// Loop over boxes
+ // Note: we do NOT OpenMP parallelize here, since we use OpenMP threads for
+ // the FFTs on each box!
for ( MFIter mfi(mf); mfi.isValid(); ++mfi ){
if (cost && WarpX::load_balance_costs_update_algo == LoadBalanceCostsUpdateAlgo::Timers)
{
@@ -247,6 +249,8 @@ SpectralFieldData::BackwardTransform( const int lev,
#endif
// Loop over boxes
+ // Note: we do NOT OpenMP parallelize here, since we use OpenMP threads for
+ // the iFFTs on each box!
for ( MFIter mfi(mf); mfi.isValid(); ++mfi ){
if (cost && WarpX::load_balance_costs_update_algo == LoadBalanceCostsUpdateAlgo::Timers)
{
diff --git a/Source/FieldSolver/SpectralSolver/WrapFFTW.cpp b/Source/FieldSolver/SpectralSolver/WrapFFTW.cpp
index a4dfc8b29..57a0fad04 100644
--- a/Source/FieldSolver/SpectralSolver/WrapFFTW.cpp
+++ b/Source/FieldSolver/SpectralSolver/WrapFFTW.cpp
@@ -1,4 +1,4 @@
-/* Copyright 2019-2020
+/* Copyright 2019-2021
*
* This file is part of WarpX.
*
@@ -32,6 +32,16 @@ namespace AnyFFT
{
FFTplan fft_plan;
+#if defined(AMREX_USE_OMP) && defined(WarpX_FFTW_OMP)
+# ifdef AMREX_USE_FLOAT
+ fftwf_init_threads();
+ fftwf_plan_with_nthreads(omp_get_max_threads());
+# else
+ fftw_init_threads();
+ fftw_plan_with_nthreads(omp_get_max_threads());
+# endif
+#endif
+
// Initialize fft_plan.m_plan with the vendor fft plan.
// Swap dimensions: AMReX FAB are Fortran-order but FFTW is C-order
if (dir == direction::R2C){