6#include "cuda_runtime.h"
9template <
typename FPTYPE>
47 std::complex<FPTYPE>* out)
const override;
56 std::complex<FPTYPE>* out)
const override;
virtual int int nz_in
Definition fft_base.h:21
virtual int ny_in
Definition fft_base.h:20
void setupFFT() override
Setup the fft plan and data as pure virtual function.
void fft3D_backward(std::complex< FPTYPE > *in, std::complex< FPTYPE > *out) const override
Backward FFT in 3D.
void clear() override
Clear the fft data as pure virtual function.
cufftHandle z_handle
Definition fft_cuda.h:59
void fft3D_forward(std::complex< FPTYPE > *in, std::complex< FPTYPE > *out) const override
Forward FFT in 3D.
std::complex< FPTYPE > * get_auxr_3d_data() const override
Get the real space data.
void cleanFFT() override
Clean the fft plan as pure virtual function.
FFT_CUDA()
Definition fft_cuda.h:13
~FFT_CUDA()
Definition fft_cuda.h:14
cufftHandle c_handle
Definition fft_cuda.h:58
void initfft(int nx_in, int ny_in, int nz_in) override
Initialize the fft parameters.
Definition fft_cuda.cpp:9
std::complex< double > * z_auxr_3d
Definition fft_cuda.h:62
std::complex< float > * c_auxr_3d
Definition fft_cuda.h:61