fix bug

Author: haozhihan

source/module_psi/psi.cpp

@@ -32,6 +32,7 @@ Range::Range(const bool k_first_in, const size_t index_1_in, const size_t range_
 template <typename T, typename Device>
 Psi<T, Device>::Psi()
 {
+    this->npol = this->get_npol();
 }
 
 template <typename T, typename Device>
@@ -51,6 +52,8 @@ Psi<T, Device>::Psi(const int nk_in, const int nbd_in, const int nbs_in, const i
     assert(nbd_in >= 0); // 187_PW_SDFT_ALL_GPU && 187_PW_MD_SDFT_ALL_GPU
     assert(nbs_in > 0);
 
+    this->npol = this->get_npol();
+
     this->k_first = k_first_in;
     this->allocate_inside = true;
 
@@ -88,6 +91,9 @@ Psi<T, Device>::Psi(const int nk_in,
     assert(nbd_in > 0);
     assert(nbs_in > 0);
 
+
+    this->npol = this->get_npol();
+
     this->k_first = k_first_in;
     this->allocate_inside = true;
 
@@ -125,6 +131,9 @@ Psi<T, Device>::Psi(T* psi_pointer,
     // Currently this function only supports nk_in == 1 when called within diagH_subspace_init.
     // assert(nk_in == 1); // NOTE because lr/utils/lr_uril.hpp func & get_psi_spin func
 
+
+    this->npol = this->get_npol();
+
     this->k_first = k_first_in;
     this->allocate_inside = false;
 
@@ -156,6 +165,8 @@ Psi<T, Device>::Psi(const int nk_in,
     // Currently this function only supports nk_in == 1 when called within diagH_subspace_init.
     assert(nk_in == 1);
 
+    this->npol = this->get_npol();
+
     this->k_first = k_first_in;
     this->allocate_inside = true;
 
@@ -185,6 +196,8 @@ Psi<T, Device>::Psi(const int nk_in,
 template <typename T, typename Device>
 Psi<T, Device>::Psi(const Psi& psi_in)
 {
+    this->npol = this->get_npol();
+
     this->ngk = psi_in.ngk;
     this->nk = psi_in.get_nk();
     this->nbands = psi_in.get_nbands();
@@ -211,6 +224,9 @@ template <typename T, typename Device>
 template <typename T_in, typename Device_in>
 Psi<T, Device>::Psi(const Psi<T_in, Device_in>& psi_in)
 {
+
+    this->npol = this->get_npol();
+
     this->ngk = psi_in.get_ngk_pointer();
     this->nk = psi_in.get_nk();
     this->nbands = psi_in.get_nbands();