change the logic of cuda forward and backward

A-006 · A-006 · commit 0b88bc8d4b94 · 2024-11-06T14:44:03.000+08:00
diff --git a/source/module_base/module_fft/fft_cuda.cpp b/source/module_base/module_fft/fft_cuda.cpp
@@ -77,22 +77,25 @@ void FFT_CUDA<float>::fft3D_forward(std::complex<float>* in, std::complex<float>
                              CUFFT_FORWARD));
 }
 template <>
-void FFT_CUDA<double>::fft3D_backward(std::complex<double>* in, std::complex<double>* out) const
+void FFT_CUDA<double>::fft3D_forward(std::complex<double>* in, std::complex<double>* out) const
 {
     CHECK_CUFFT(cufftExecZ2Z(this->z_handle, reinterpret_cast<cufftDoubleComplex*>(in),
                              reinterpret_cast<cufftDoubleComplex*>(out), CUFFT_FORWARD));
 }
 template <>
 void FFT_CUDA<float>::fft3D_backward(std::complex<float>* in, std::complex<float>* out) const
 {
-    CHECK_CUFFT(cufftExecC2C(this->c_handle, reinterpret_cast<cufftComplex*>(in), reinterpret_cast<cufftComplex*>(out),
-                             CUFFT_INVERSE));
+    CHECK_CUFFT(cufftExecC2C(this->c_handle, reinterpret_cast<cufftComplex*>(in), 
+                             reinterpret_cast<cufftComplex*>(out),CUFFT_INVERSE));
 }
+
 template <>
-void FFT_CUDA<double>::fft3D_forward(std::complex<double>* in, std::complex<double>* out) const
+void FFT_CUDA<double>::fft3D_backward(std::complex<double>* in, std::complex<double>* out) const
 {
     CHECK_CUFFT(cufftExecZ2Z(this->z_handle, reinterpret_cast<cufftDoubleComplex*>(in),
                              reinterpret_cast<cufftDoubleComplex*>(out), CUFFT_INVERSE));
 }
+
+
 template FFT_CUDA<float>::FFT_CUDA();
 template FFT_CUDA<double>::FFT_CUDA();
diff --git a/source/module_base/module_fft/fft_rcom.cpp b/source/module_base/module_fft/fft_rcom.cpp
@@ -77,7 +77,7 @@ void FFT_RCOM<float>::fft3D_forward(std::complex<float>* in, std::complex<float>
                               reinterpret_cast<hipfftComplex*>(out), HIPFFT_FORWARD));
 }
 template <>
-void FFT_RCOM<double>::fft3D_backward(std::complex<double>* in, std::complex<double>* out) const
+void FFT_RCOM<double>::fft3D_forward(std::complex<double>* in, std::complex<double>* out) const
 {
     CHECK_CUFFT(hipfftExecZ2Z(this->z_handle, reinterpret_cast<hipfftDoubleComplex*>(in),
                               reinterpret_cast<hipfftDoubleComplex*>(out), HIPFFT_FORWARD));
@@ -89,10 +89,12 @@ void FFT_RCOM<float>::fft3D_backward(std::complex<float>* in, std::complex<float
                               reinterpret_cast<hipfftComplex*>(out), HIPFFT_BACKWARD));
 }
 template <>
-void FFT_RCOM<double>::fft3D_forward(std::complex<double>* in, std::complex<double>* out) const
+void FFT_RCOM<double>::fft3D_backward(std::complex<double>* in, std::complex<double>* out) const
 {
     CHECK_CUFFT(hipfftExecZ2Z(this->z_handle, reinterpret_cast<hipfftDoubleComplex*>(in),
                               reinterpret_cast<hipfftDoubleComplex*>(out), HIPFFT_BACKWARD));
 }
+
+
 template FFT_RCOM<float>::FFT_RCOM();
 template FFT_RCOM<double>::FFT_RCOM();

Original file line number	Diff line number	Diff line change
`@@ -77,22 +77,25 @@ void FFT_CUDA<float>::fft3D_forward(std::complex<float>* in, std::complex<float>`
`77`	`77`	`CUFFT_FORWARD));`
`78`	`78`	`}`
`79`	`79`	`template <>`
`80`		`-void FFT_CUDA<double>::fft3D_backward(std::complex<double>* in, std::complex<double>* out) const`
	`80`	`+void FFT_CUDA<double>::fft3D_forward(std::complex<double>* in, std::complex<double>* out) const`
`81`	`81`	`{`
`82`	`82`	`CHECK_CUFFT(cufftExecZ2Z(this->z_handle, reinterpret_cast<cufftDoubleComplex*>(in),`
`83`	`83`	`reinterpret_cast<cufftDoubleComplex*>(out), CUFFT_FORWARD));`
`84`	`84`	`}`
`85`	`85`	`template <>`
`86`	`86`	`void FFT_CUDA<float>::fft3D_backward(std::complex<float>* in, std::complex<float>* out) const`
`87`	`87`	`{`
`88`		`- CHECK_CUFFT(cufftExecC2C(this->c_handle, reinterpret_cast<cufftComplex>(in), reinterpret_cast<cufftComplex>(out),`
`89`		`- CUFFT_INVERSE));`
	`88`	`+ CHECK_CUFFT(cufftExecC2C(this->c_handle, reinterpret_cast<cufftComplex*>(in),`
	`89`	`+ reinterpret_cast<cufftComplex*>(out),CUFFT_INVERSE));`
`90`	`90`	`}`
	`91`	`+`
`91`	`92`	`template <>`
`92`		`-void FFT_CUDA<double>::fft3D_forward(std::complex<double>* in, std::complex<double>* out) const`
	`93`	`+void FFT_CUDA<double>::fft3D_backward(std::complex<double>* in, std::complex<double>* out) const`
`93`	`94`	`{`
`94`	`95`	`CHECK_CUFFT(cufftExecZ2Z(this->z_handle, reinterpret_cast<cufftDoubleComplex*>(in),`
`95`	`96`	`reinterpret_cast<cufftDoubleComplex*>(out), CUFFT_INVERSE));`
`96`	`97`	`}`
	`98`	`+`
	`99`	`+`
`97`	`100`	`template FFT_CUDA<float>::FFT_CUDA();`
`98`	`101`	`template FFT_CUDA<double>::FFT_CUDA();`
Original file line number	Diff line number	Diff line change
`@@ -77,7 +77,7 @@ void FFT_RCOM<float>::fft3D_forward(std::complex<float>* in, std::complex<float>`
`77`	`77`	`reinterpret_cast<hipfftComplex*>(out), HIPFFT_FORWARD));`
`78`	`78`	`}`
`79`	`79`	`template <>`
`80`		`-void FFT_RCOM<double>::fft3D_backward(std::complex<double>* in, std::complex<double>* out) const`
	`80`	`+void FFT_RCOM<double>::fft3D_forward(std::complex<double>* in, std::complex<double>* out) const`
`81`	`81`	`{`
`82`	`82`	`CHECK_CUFFT(hipfftExecZ2Z(this->z_handle, reinterpret_cast<hipfftDoubleComplex*>(in),`
`83`	`83`	`reinterpret_cast<hipfftDoubleComplex*>(out), HIPFFT_FORWARD));`
`@@ -89,10 +89,12 @@ void FFT_RCOM<float>::fft3D_backward(std::complex<float>* in, std::complex<float`
`89`	`89`	`reinterpret_cast<hipfftComplex*>(out), HIPFFT_BACKWARD));`
`90`	`90`	`}`
`91`	`91`	`template <>`
`92`		`-void FFT_RCOM<double>::fft3D_forward(std::complex<double>* in, std::complex<double>* out) const`
	`92`	`+void FFT_RCOM<double>::fft3D_backward(std::complex<double>* in, std::complex<double>* out) const`
`93`	`93`	`{`
`94`	`94`	`CHECK_CUFFT(hipfftExecZ2Z(this->z_handle, reinterpret_cast<hipfftDoubleComplex*>(in),`
`95`	`95`	`reinterpret_cast<hipfftDoubleComplex*>(out), HIPFFT_BACKWARD));`
`96`	`96`	`}`
	`97`	`+`
	`98`	`+`
`97`	`99`	`template FFT_RCOM<float>::FFT_RCOM();`
`98`	`100`	`template FFT_RCOM<double>::FFT_RCOM();`