guilhem_lavaux/cosmotool
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix 1d support

Browse Source
This commit is contained in:
Guilhem Lavaux 2021-04-22 08:42:24 +02:00
parent 3e013139f2
commit 8068ebe3ae
2 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/fourier/fft/fftw_calls.hpp
View File

@ -134,6 +134,11 @@ namespace CosmoTool {
unsigned flags) { \ unsigned flags) { \
return prefix##_plan_dft_2d(Nx, Ny, in, out, sign, flags); \ return prefix##_plan_dft_2d(Nx, Ny, in, out, sign, flags); \
} \ } \
static plan_type plan_dft_1d( \
int Nx, complex_type *in, complex_type *out, int sign, \
unsigned flags) { \
return prefix##_plan_dft_1d(Nx, in, out, sign, flags); \
} \
static void destroy_plan(plan_type plan) { prefix##_destroy_plan(plan); } \ static void destroy_plan(plan_type plan) { prefix##_destroy_plan(plan); } \
} }

7
src/fourier/fft/fftw_calls_mpi.hpp
View File

@ -85,7 +85,7 @@ namespace CosmoTool {
return prefix##_mpi_plan_dft_r2c_2d(Nx, Ny, in, out, comm, flags); \ return prefix##_mpi_plan_dft_r2c_2d(Nx, Ny, in, out, comm, flags); \
} \ } \
\ \
static plan_type plan_dft_r2c_1d( \ static plan_type plan_dft_c2r_1d( \
int n, complex_type *in, real_type *out, MPI_Comm, unsigned flags) { \ int n, complex_type *in, real_type *out, MPI_Comm, unsigned flags) { \
return prefix##_plan_dft_c2r_1d(n, in, out, flags); \ return prefix##_plan_dft_c2r_1d(n, in, out, flags); \
} \ } \
@ -126,6 +126,11 @@ namespace CosmoTool {
int sign, unsigned flags) { \ int sign, unsigned flags) { \
return prefix##_mpi_plan_dft_2d(Nx, Ny, in, out, comm, sign, flags); \ return prefix##_mpi_plan_dft_2d(Nx, Ny, in, out, comm, sign, flags); \
} \ } \
static plan_type plan_dft_1d( \
int Nx, complex_type *in, complex_type *out, MPI_Comm comm, int sign, \
unsigned flags) { \
return prefix##_plan_dft_1d(Nx, in, out, sign, flags); \
} \
static void destroy_plan(plan_type plan) { prefix##_destroy_plan(plan); } \ static void destroy_plan(plan_type plan) { prefix##_destroy_plan(plan); } \
} }