Skip to content

Commit 4fd5094

Browse files
committed
Modify hpsi_func definition in dav tests
1 parent c7613df commit 4fd5094

File tree

3 files changed

+9
-12
lines changed

3 files changed

+9
-12
lines changed

source/module_hsolver/test/diago_david_float_test.cpp

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -108,11 +108,10 @@ class DiagoDavPrepare
108108

109109

110110
auto hpsi_func = [phm](std::complex<float>* psi_in,std::complex<float>* hpsi_out,
111-
const int nband_in, const int nbasis_in,
112-
const int band_index1, const int band_index2)
111+
const int ldPsi, const int nvec)
113112
{
114-
auto psi_iter_wrapper = psi::Psi<std::complex<float>>(psi_in, 1, nband_in, nbasis_in, nullptr);
115-
psi::Range bands_range(1, 0, band_index1, band_index2);
113+
auto psi_iter_wrapper = psi::Psi<std::complex<float>>(psi_in, 1, nvec, ldPsi, nullptr);
114+
psi::Range bands_range(1, 0, 0, nvec-1);
116115
using hpsi_info = typename hamilt::Operator<std::complex<float>>::hpsi_info;
117116
hpsi_info info(&psi_iter_wrapper, bands_range, hpsi_out);
118117
phm->ops->hPsi(info);

source/module_hsolver/test/diago_david_real_test.cpp

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -107,11 +107,10 @@ class DiagoDavPrepare
107107

108108

109109
auto hpsi_func = [phm](double* psi_in,double* hpsi_out,
110-
const int nband_in, const int nbasis_in,
111-
const int band_index1, const int band_index2)
110+
const int ldPsi, const int nvec)
112111
{
113-
auto psi_iter_wrapper = psi::Psi<double>(psi_in, 1, nband_in, nbasis_in, nullptr);
114-
psi::Range bands_range(1, 0, band_index1, band_index2);
112+
auto psi_iter_wrapper = psi::Psi<double>(psi_in, 1, nvec, ldPsi, nullptr);
113+
psi::Range bands_range(1, 0, 0, nvec-1);
115114
using hpsi_info = typename hamilt::Operator<double>::hpsi_info;
116115
hpsi_info info(&psi_iter_wrapper, bands_range, hpsi_out);
117116
phm->ops->hPsi(info);

source/module_hsolver/test/diago_david_test.cpp

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -110,11 +110,10 @@ class DiagoDavPrepare
110110

111111

112112
auto hpsi_func = [phm](std::complex<double>* psi_in,std::complex<double>* hpsi_out,
113-
const int nband_in, const int nbasis_in,
114-
const int band_index1, const int band_index2)
113+
const int ldPsi, const int nvec)
115114
{
116-
auto psi_iter_wrapper = psi::Psi<std::complex<double>>(psi_in, 1, nband_in, nbasis_in, nullptr);
117-
psi::Range bands_range(1, 0, band_index1, band_index2);
115+
auto psi_iter_wrapper = psi::Psi<std::complex<double>>(psi_in, 1, nvec, ldPsi, nullptr);
116+
psi::Range bands_range(1, 0, 0, nvec-1);
118117
using hpsi_info = typename hamilt::Operator<std::complex<double>>::hpsi_info;
119118
hpsi_info info(&psi_iter_wrapper, bands_range, hpsi_out);
120119
phm->ops->hPsi(info);

0 commit comments

Comments
 (0)