File tree Expand file tree Collapse file tree 3 files changed +9
-4
lines changed
source/module_basis/module_pw/module_fft Expand file tree Collapse file tree 3 files changed +9
-4
lines changed Original file line number Diff line number Diff line change @@ -105,6 +105,7 @@ template <> std::complex<float>*
105105FFT_CUDA<float >::get_auxr_3d_data() const {return this ->c_auxr_3d ;}
106106template <> std::complex <double >*
107107FFT_CUDA<double >::get_auxr_3d_data() const {return this ->z_auxr_3d ;}
108+
108109template FFT_CUDA<float >::FFT_CUDA();
109110template FFT_CUDA<double >::FFT_CUDA();
110111}// namespace ModulePW
Original file line number Diff line number Diff line change @@ -43,15 +43,17 @@ class FFT_CUDA : public FFT_BASE<FPTYPE>
4343 *
4444 * This function performs the forward FFT in 3D.
4545 */
46- void fft3D_forward (std::complex <FPTYPE>* in, std::complex <FPTYPE>* out) const override ;
46+ void fft3D_forward (std::complex <FPTYPE>* in,
47+ std::complex <FPTYPE>* out) const override ;
4748 /* *
4849 * @brief Backward FFT in 3D
4950 * @param in input data, complex FPTYPE
5051 * @param out output data, complex FPTYPE
5152 *
5253 * This function performs the backward FFT in 3D.
5354 */
54- void fft3D_backward (std::complex <FPTYPE>* in, std::complex <FPTYPE>* out) const override ;
55+ void fft3D_backward (std::complex <FPTYPE>* in,
56+ std::complex <FPTYPE>* out) const override ;
5557 private:
5658 cufftHandle c_handle = {};
5759 cufftHandle z_handle = {};
Original file line number Diff line number Diff line change @@ -40,14 +40,16 @@ class FFT_ROCM : public FFT_BASE<FPTYPE>
4040 * @param in input data, complex FPTYPE
4141 * @param out output data, complex FPTYPE
4242 */
43- void fft3D_forward (std::complex <FPTYPE>* in, std::complex <FPTYPE>* out) const override ;
43+ void fft3D_forward (std::complex <FPTYPE>* in,
44+ std::complex <FPTYPE>* out) const override ;
4445
4546 /* *
4647 * @brief Backward FFT in 3D for ROCM
4748 * @param in input data, complex FPTYPE
4849 * @param out output data, complex FPTYPE
4950 */
50- void fft3D_backward (std::complex <FPTYPE>* in, std::complex <FPTYPE>* out) const override ;
51+ void fft3D_backward (std::complex <FPTYPE>* in,
52+ std::complex <FPTYPE>* out) const override ;
5153 private:
5254 hipfftHandle c_handle = {};
5355 hipfftHandle z_handle = {};
You can’t perform that action at this time.
0 commit comments