Skip to content

Commit 1c2f523

Browse files
committed
update get_ngk usage
1 parent c716bb7 commit 1c2f523

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

source/module_hamilt_general/operator.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ typename Operator<T, Device>::hpsi_info Operator<T, Device>::hPsi(hpsi_info& inp
7373
op->act(psi_wrapper, *this->hpsi, nbands);
7474
break;
7575
default:
76-
op->act(nbands, psi_input->get_nbasis(), psi_input->npol, tmpsi_in, this->hpsi->get_pointer(), psi_input->get_ngk(op->ik), is_first_node);
76+
op->act(nbands, psi_input->get_nbasis(), psi_input->npol, tmpsi_in, this->hpsi->get_pointer(), psi_input->get_current_nbas(), is_first_node);
7777
break;
7878
}
7979
};

source/module_hamilt_pw/hamilt_pwdft/operator_pw/velocity_pw.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ void Velocity::act
4747
) const
4848
{
4949
ModuleBase::timer::tick("Operator", "Velocity");
50-
const int npw = psi_in->get_ngk(this->ik);
50+
const int npw = psi_in->get_current_nbas();
5151
const int max_npw = psi_in->get_nbasis() / psi_in->npol;
5252
const int npol = psi_in->npol;
5353
const std::complex<double>* tmpsi_in = psi0;

source/module_psi/psi.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,7 @@ Psi<T, Device>::Psi(T* psi_pointer, const int nk_in, const int nbd_in, const int
106106
}
107107

108108
template <typename T, typename Device>
109-
Psi<T, Device>::Psi(const Psi& psi_in, const int nk_in, int nband_in)
109+
Psi<T, Device>::Psi(const Psi& psi_in, const int nk_in, const int nband_in)
110110
{
111111
assert(nk_in <= psi_in.get_nk() && nk_in > 0);
112112
assert(nband_in <= psi_in.get_nbands() && nband_in > 0);

0 commit comments

Comments
 (0)