fix error introduced by esolver_ks's psi, which should be esolver_ks_pw's new psi (stp.psi)

Author: mohanchen

source/source_esolver/esolver_sdft_pw.cpp

@@ -177,7 +177,7 @@ void ESolver_SDFT_PW<T, Device>::hamilt2rho_single(UnitCell& ucell, int istep, i
     hsolver_pw_sdft_obj.solve(ucell,
                               this->p_hamilt,
                               this->stp.psi_t[0],
-                              this->psi[0],
+                              this->stp.psi_cpu[0],
                               this->pelec,
                               this->pw_wfc,
                               this->stowf,
@@ -279,7 +279,7 @@ void ESolver_SDFT_PW<T, Device>::after_all_runners(UnitCell& ucell)
             this->pw_wfc,
             &this->kv,
             this->pelec,
-            reinterpret_cast<psi::Psi<std::complex<double>>*>(this->psi),
+            reinterpret_cast<psi::Psi<std::complex<double>>*>(this->stp.psi_cpu),
             reinterpret_cast<hamilt::Hamilt<std::complex<double>>*>(this->p_hamilt),
             this->stoche,
             reinterpret_cast<Stochastic_WF<std::complex<double>, base_device::DEVICE_CPU>*>(&stowf));