fix wrong setmem

Author: Qianruipku

source/module_hamilt_general/operator.h

@@ -57,6 +57,7 @@ class Operator
     ///do operation : |hpsi_choosed> = V|psi_choosed>
     ///V is the target operator act on choosed psi, the consequence should be added to choosed hpsi
     /// interface type 1: pointer-only (default)
+    /// @note PW: nbasis = max_npw * npol, nbands = nband * npol, npol = npol. Strange but PAY ATTENTION!!!
     virtual void act(const int nbands,
         const int nbasis,
         const int npol,

source/module_hamilt_pw/hamilt_pwdft/operator_pw/meta_pw.cpp

@@ -54,7 +54,7 @@ void Meta<OperatorPW<T, Device>>::act(
     ModuleBase::timer::tick("Operator", "MetaPW");
     if(is_first_node)
     {
-        setmem_complex_op()(this->ctx, tmhpsi, 0, nbasis*nbands);
+        setmem_complex_op()(this->ctx, tmhpsi, 0, nbasis*nbands/npol);
     }
 
     const int current_spin = this->isk[this->ik];

source/module_hamilt_pw/hamilt_pwdft/operator_pw/nonlocal_pw.cpp

@@ -220,7 +220,7 @@ void Nonlocal<OperatorPW<T, Device>>::act(
     ModuleBase::timer::tick("Operator", "NonlocalPW");
     if(is_first_node)
     {
-        setmem_complex_op()(this->ctx, tmhpsi, 0, nbasis*nbands);
+        setmem_complex_op()(this->ctx, tmhpsi, 0, nbasis*nbands/npol);
     }
     if(!PARAM.inp.use_paw)
     {

source/module_hamilt_pw/hamilt_pwdft/operator_pw/veff_pw.cpp

@@ -47,7 +47,7 @@ void Veff<OperatorPW<T, Device>>::act(
     ModuleBase::timer::tick("Operator", "VeffPW");
     if(is_first_node)
     {
-        setmem_complex_op()(this->ctx, tmhpsi, 0, nbasis*nbands);
+        setmem_complex_op()(this->ctx, tmhpsi, 0, nbasis*nbands/npol);
     }
 
     int max_npw = nbasis / npol;