fix complie bug and change the fftwf logic

Author: A-006

source/module_basis/module_pw/module_fft/fft_bundle.cpp

@@ -26,20 +26,10 @@ FFT_Bundle::FFT_Bundle(std::string device_in,std::string precision_in)
     assert(precision_in=="single" || precision_in=="double" || precision_in=="mixing");
     this->device = device_in;
     this->precision = precision_in;
-    if (device=="cpu")
-    {
-        fft_float = make_unique<FFT_CPU<float>>();
-        fft_double = make_unique<FFT_CPU<double>>();
-    }
-    // else if (device=="gpu")
-    // {        
-    //     #if defined(__ROCM)
-    //         fft_float = new FFT_RCOM<float>();
-    //         fft_double = new FFT_RCOM<double>();
-    //     #elif defined(__CUDA)
-    //         fft_float = new FFT_CUDA<float>();
-    //         fft_double = new FFT_CUDA<double>();
-    //     #endif
+    // if (device=="cpu")
+    // {
+    //     fft_float = make_unique<FFT_CPU<float>>();
+    //     fft_double = make_unique<FFT_CPU<double>>();
     // }
 }
 
@@ -143,18 +133,6 @@ void FFT_Bundle::clear()
     {
         fft_double->clear();
     }
-    // if (fft_float!=nullptr)
-    // {
-    //     delete fft_float;
-    //     fft_float=nullptr;
-    //     float_flag = false;
-    // }
-    // if (fft_double!=nullptr)
-    // {
-    //     delete fft_double;
-    //     fft_double=nullptr;
-    //     double_flag = false;
-    // }
 }
 // access the real space data
 template <>

source/module_basis/module_pw/module_fft/fft_cpu.cpp

@@ -84,8 +84,6 @@ void FFT_CPU<double>::setupFFT()
     default:
         break;
     }
-    // if (!this->mpifft)
-    // {
         z_auxg = (std::complex<double>*)fftw_malloc(sizeof(fftw_complex) * this->maxgrids);
         z_auxr = (std::complex<double>*)fftw_malloc(sizeof(fftw_complex) * this->maxgrids);
         d_rspace = (double*)z_auxg;
@@ -149,7 +147,6 @@ void FFT_CPU<double>::setupFFT()
                                                     1, (fftw_complex*)z_auxr, embed, this->nplane, 1, FFTW_BACKWARD, flag);
             }
         }
-    // }
     return;
 }
 

source/module_basis/module_pw/module_fft/fft_cpu_float.cpp

@@ -1,9 +1,5 @@
 #include "fft_cpu.h"
 
-// #include "fftw3f.h"
-// #if defined(__FFTW3_MPI) && defined(__MPI)
-// #include "fftw3f-mpi.h"
-// //#include "fftw3-mpi_mkl.h"
 namespace ModulePW
 {
 template <>
@@ -56,8 +52,8 @@ void FFT_CPU<float>::setupFFT()
     default:
         break;
     }
-    if (!this->mpifft)
-    {
+    // if (!this->mpifft)
+    // {
         c_auxg = (std::complex<float>*)fftwf_malloc(sizeof(fftwf_complex) * this->maxgrids); 
         c_auxr = (std::complex<float>*)fftwf_malloc(sizeof(fftwf_complex) * this->maxgrids);
         s_rspace = (float*)c_auxg;
@@ -131,16 +127,6 @@ void FFT_CPU<float>::setupFFT()
                                         (fftwf_complex*)c_auxr, embed, this->nplane, 1, FFTW_BACKWARD, flag);
             }
         }
-        }
-    #if defined(__FFTW3_MPI) && defined(__MPI)
-        else
-        {
-            // this->initplan_mpi();
-            // if (this->precision == "single") {
-            //     this->initplanf_mpi();
-            // }
-        }
-    #endif
     return;
 }
 
@@ -308,14 +294,12 @@ void FFT_CPU<float>::fftxyc2r(std::complex<float>* in, float* out) const
     }
     else
     {
-        fftwf_execute_dft(this->planfxfor1, (fftwf_complex*)in, (fftwf_complex*)in);
+        fftwf_execute_dft(this->planfxbac1, (fftwf_complex*)in, (fftwf_complex*)in);
 
         for (int i = 0; i < this->nx; ++i)
         {
-            fftwf_execute_dft(this->planfybac, (fftwf_complex*)&in[i * npy], (fftwf_complex*)&in[i * npy]);
+            fftwf_execute_dft_c2r(this->planfyc2r, (fftwf_complex*)&in[i * npy], &out[i * npy]);
         }
-
-        fftwf_execute_dft_c2r(this->planfyc2r, (fftwf_complex*)in, out);
     }
 }
 }

source/module_hamilt_general/module_xc/test/CMakeLists.txt

@@ -39,7 +39,7 @@ AddTest(
     ../../../module_base/libm/sincos.cpp
     ../../../module_base/blas_connector.cpp
     ../../../module_basis/module_pw/module_fft/fft_base.cpp
-    ../../../module_basis/module_pw/module_fft/fft_temp.cpp
+    ../../../module_basis/module_pw/module_fft/fft_bundle.cpp
     ../../../module_basis/module_pw/module_fft/fft_cpu.cpp
 )
 
@@ -77,6 +77,6 @@ AddTest(
     ../../../module_base/libm/branred.cpp
     ../../../module_base/libm/sincos.cpp
     ../../../module_basis/module_pw/module_fft/fft_base.cpp
-    ../../../module_basis/module_pw/module_fft/fft_temp.cpp
+    ../../../module_basis/module_pw/module_fft/fft_bundle.cpp
     ../../../module_basis/module_pw/module_fft/fft_cpu.cpp
 )