From 8068ebe3ae87537deb47c00746a645136f049c62 Mon Sep 17 00:00:00 2001 From: Guilhem Lavaux Date: Thu, 22 Apr 2021 08:42:24 +0200 Subject: [PATCH] Fix 1d support --- src/fourier/fft/fftw_calls.hpp | 5 +++++ src/fourier/fft/fftw_calls_mpi.hpp | 7 ++++++- 2 files changed, 11 insertions(+), 1 deletion(-) diff --git a/src/fourier/fft/fftw_calls.hpp b/src/fourier/fft/fftw_calls.hpp index 5ffefcc..67a714a 100644 --- a/src/fourier/fft/fftw_calls.hpp +++ b/src/fourier/fft/fftw_calls.hpp @@ -134,6 +134,11 @@ namespace CosmoTool { unsigned 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); } \ } diff --git a/src/fourier/fft/fftw_calls_mpi.hpp b/src/fourier/fft/fftw_calls_mpi.hpp index 160a604..aa31233 100644 --- a/src/fourier/fft/fftw_calls_mpi.hpp +++ b/src/fourier/fft/fftw_calls_mpi.hpp @@ -85,7 +85,7 @@ namespace CosmoTool { 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) { \ return prefix##_plan_dft_c2r_1d(n, in, out, flags); \ } \ @@ -126,6 +126,11 @@ namespace CosmoTool { int sign, unsigned 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); } \ }