Commit c88b064d by Francois Gygi

updated call to set number of threads for FFTW3_THREADS

git-svn-id: http://qboxcode.org/svn/qb/trunk@1564 cba15fb0-1239-40c8-b417-11db7ca47a34
parent da3e6a67
......@@ -27,6 +27,11 @@
#if _OPENMP
#include <omp.h>
#else
// _OPENMP is not defined
#if defined(USE_FFTW3_THREADS)
#error "Need OpenMP to use FFTW3 threads"
#endif
#endif
#if USE_MPI
......@@ -1453,13 +1458,13 @@ void FourierTransform::init_lib(void)
#elif USE_FFTW3
vector<complex<double> > aux(np0_*np1_);
#if defined(USE_FFTW3MKL) && !defined(USE_FFTW3_THREADS)
fftw3_mkl.number_of_user_threads = omp_get_num_threads();
#if defined(USE_FFTW3MKL) && !defined(USE_FFTW3_THREADS) && _OPENMP
fftw3_mkl.number_of_user_threads = omp_get_max_threads();
#endif
#if USE_FFTW3_THREADS
fftw_init_threads();
fftw_plan_with_nthreads(omp_get_num_threads());
fftw_plan_with_nthreads(omp_get_max_threads());
vector<complex<double> > aux1(np0_*np1_*np2_loc_[myproc_]);
// xy
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment