Fix: force error of DFTU in GPU/DCU version

Author: dyzheng

source/module_hamilt_pw/hamilt_pwdft/kernels/cuda/force_op.cu

@@ -405,7 +405,7 @@ __global__ void cal_force_onsite(int wg_nc,
                                   int nkb,
                                   const int* atom_nh,
                                   const int* atom_na,
-                                  int tpiba,
+                                  const FPTYPE tpiba,
                                   const FPTYPE* d_wg,
                                   const thrust::complex<FPTYPE>* vu,
                                   const int* orbital_corr,
@@ -477,7 +477,7 @@ __global__ void cal_force_onsite_np1(int wg_nc,
                                   int nkb,
                                   const int* atom_nh,
                                   const int* atom_na,
-                                  int tpiba,
+                                  const FPTYPE tpiba,
                                   const FPTYPE* d_wg,
                                   const thrust::complex<FPTYPE>* vu,
                                   const int* orbital_corr,
@@ -543,7 +543,7 @@ __global__ void cal_force_onsite(int wg_nc,
                                  int nkb,
                                  const int* atom_nh,
                                  const int* atom_na,
-                                 int tpiba,
+                                 const FPTYPE tpiba,
                                  const FPTYPE* d_wg,
                                  const FPTYPE* lambda,
                                  const thrust::complex<FPTYPE>* becp,
@@ -603,7 +603,7 @@ __global__ void cal_force_onsite_np1(int wg_nc,
                                  int nkb,
                                  const int* atom_nh,
                                  const int* atom_na,
-                                 int tpiba,
+                                 const FPTYPE tpiba,
                                  const FPTYPE* d_wg,
                                  const FPTYPE* lambda,
                                  const thrust::complex<FPTYPE>* becp,

source/module_hamilt_pw/hamilt_pwdft/kernels/rocm/force_op.hip.cu

@@ -421,7 +421,7 @@ __global__ void cal_force_onsite(int wg_nc,
                                   int nkb,
                                   const int* atom_nh,
                                   const int* atom_na,
-                                  int tpiba,
+                                  const FPTYPE tpiba,
                                   const FPTYPE* d_wg,
                                   const thrust::complex<FPTYPE>* vu,
                                   const int* orbital_corr,
@@ -488,7 +488,7 @@ __global__ void cal_force_onsite(int wg_nc,
                                  int nkb,
                                  const int* atom_nh,
                                  const int* atom_na,
-                                 int tpiba,
+                                 const FPTYPE tpiba,
                                  const FPTYPE* d_wg,
                                  const FPTYPE* lambda,
                                  const thrust::complex<FPTYPE>* becp,