update the template file

Author: A-006

source/module_basis/module_pw/module_fft/fft_cpu.cpp

@@ -3,8 +3,8 @@
 namespace ModulePW
 {
 
-template <>
-void FFT_CPU<double>::initfft(int nx_in, 
+template <typename FPTYPE>
+void FFT_CPU<FPTYPE>::initfft(int nx_in, 
                               int ny_in, 
                               int nz_in, 
                               int lixy_in, 
@@ -457,8 +457,4 @@ template <> std::complex<double>*
 FFT_CPU<double>::get_auxr_data()   const {return z_auxr;}
 template <> std::complex<double>* 
 FFT_CPU<double>::get_auxg_data()   const {return z_auxg;}
-
-
-template FFT_CPU<double>::FFT_CPU();
-template FFT_CPU<double>::~FFT_CPU();
 }

source/module_basis/module_pw/module_fft/fft_cpu.h

@@ -33,7 +33,7 @@ class FFT_CPU : public FFT_BASE<FPTYPE>
      * @param gamma_only_in  whether only gamma point is used.
      * @param xprime_in  whether xprime is used.
      */
-    __attribute__((weak)) 
+
     void initfft(int nx_in, 
                  int ny_in, 
                  int nz_in, 
@@ -169,5 +169,7 @@ class FFT_CPU : public FFT_BASE<FPTYPE>
          */
         int fft_mode = 0; 
 };
+template FFT_CPU<float>::FFT_CPU();
+template FFT_CPU<double>::FFT_CPU();
 }
 #endif // FFT_CPU_H

source/module_basis/module_pw/module_fft/fft_cpu_float.cpp

@@ -3,42 +3,6 @@
 namespace ModulePW
 {
 template <>
-void FFT_CPU<float>::initfft(int nx_in, 
-                              int ny_in, 
-                              int nz_in, 
-                              int lixy_in, 
-                              int rixy_in, 
-                              int ns_in, 
-                              int nplane_in, 
-				              int nproc_in, 
-                              bool gamma_only_in, 
-                              bool xprime_in)
-{
-    this->gamma_only = gamma_only_in;
-    this->xprime = xprime_in;
-    this->fftnx = this->nx = nx_in;
-    this->fftny = this->ny = ny_in;
-    if (this->gamma_only)
-    {
-        if (xprime) {
-            this->fftnx = int(this->nx / 2) + 1;
-        } else {
-            this->fftny = int(this->ny / 2) + 1;
-        }
-    }
-    this->nz = nz_in;
-    this->ns = ns_in;
-    this->lixy = lixy_in;
-    this->rixy = rixy_in;
-    this->nplane = nplane_in;
-    this->nproc = nproc_in;
-    this->nxy = this->nx * this->ny;
-    this->fftnxy = this->fftnx * this->fftny;
-    const int nrxx = this->nxy * this->nplane;
-    const int nsz = this->nz * this->ns;
-    this->maxgrids = (nsz > nrxx) ? nsz : nrxx;
-}
-template <>
 void FFT_CPU<float>::setupFFT()
 {
     unsigned int flag = FFTW_ESTIMATE;
@@ -466,6 +430,4 @@ template <> std::complex<float>*
 FFT_CPU<float>::get_auxr_data()   const {return c_auxr;}
 template <> std::complex<float>* 
 FFT_CPU<float>::get_auxg_data()   const {return c_auxg;}
-template FFT_CPU<float>::FFT_CPU();
-template FFT_CPU<float>::~FFT_CPU();
 }

source/module_basis/module_pw/module_fft/fft_cuda.cpp

@@ -105,7 +105,4 @@ template <> std::complex<float>*
 FFT_CUDA<float>::get_auxr_3d_data()  const {return this->c_auxr_3d;}
 template <> std::complex<double>* 
 FFT_CUDA<double>::get_auxr_3d_data() const {return this->z_auxr_3d;}
-
-template FFT_CUDA<float>::FFT_CUDA();
-template FFT_CUDA<double>::FFT_CUDA();
 }// namespace ModulePW

source/module_basis/module_pw/module_fft/fft_cuda.h

@@ -62,5 +62,7 @@ class FFT_CUDA : public FFT_BASE<FPTYPE>
         std::complex<double>* z_auxr_3d = nullptr; // fft space
 
 };
+template FFT_CUDA<float>::FFT_CUDA();
+template FFT_CUDA<double>::FFT_CUDA();
 } // namespace ModulePW
 #endif

source/module_basis/module_pw/module_fft/fft_rcom.cpp

@@ -103,7 +103,4 @@ template <> std::complex<float>*
 FFT_RCOM<float>::get_auxr_3d_data()  const {return this->c_auxr_3d;}
 template <> std::complex<double>* 
 FFT_RCOM<double>::get_auxr_3d_data() const {return this->z_auxr_3d;}
-
-template FFT_RCOM<float>::FFT_RCOM();
-template FFT_RCOM<double>::FFT_RCOM();
 }// namespace ModulePW

source/module_basis/module_pw/module_fft/fft_rcom.h

@@ -57,5 +57,7 @@ class FFT_ROCM : public FFT_BASE<FPTYPE>
         mutable std::complex<double>* z_auxr_3d = nullptr; // fft space
 
 };
+template FFT_RCOM<float>::FFT_RCOM();
+template FFT_RCOM<double>::FFT_RCOM();
 }// namespace ModulePW
 #endif

source/module_basis/module_pw/test/Makefile

@@ -27,7 +27,7 @@ GTEST_DIR = /home/qianrui/gnucompile/g_gtest
 HONG = -D__NORMAL
 INCLUDES = -I. -I../../../ -I../../../module_base/module_container
 LIBS = 
-OPTS = -Ofast -march=native -std=c++11 -m64  ${INCLUDES}
+OPTS = -Ofast -march=native -std=c++11 -m64  ${INCLUDES} -w -g
 OBJ_DIR = obj
 LIBNAME = libpw.a
 GEN     = OFF