Skip to content

Commit b3cdbb5

Browse files
committed
Update david test spsi_func
1 parent 2049cd9 commit b3cdbb5

File tree

4 files changed

+7
-7
lines changed

4 files changed

+7
-7
lines changed

source/module_hsolver/test/diago_david_float_test.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -116,8 +116,8 @@ class DiagoDavPrepare
116116
hpsi_info info(&psi_iter_wrapper, bands_range, hpsi_out);
117117
phm->ops->hPsi(info);
118118
};
119-
auto spsi_func = [phm](const std::complex<float>* psi_in, std::complex<float>* spsi_out,const int nrow, const int npw, const int nbands){
120-
phm->sPsi(psi_in, spsi_out, nrow, npw, nbands);
119+
auto spsi_func = [phm](const std::complex<float>* psi_in, std::complex<float>* spsi_out,const int ld_psi, const int nbands){
120+
phm->sPsi(psi_in, spsi_out, ld_psi, ld_psi, nbands);
121121
};
122122
dav.diag(hpsi_func,spsi_func, ld_psi, phi.get_pointer(), en, eps, maxiter);
123123

source/module_hsolver/test/diago_david_real_test.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -118,8 +118,8 @@ class DiagoDavPrepare
118118
hpsi_info info(&psi_iter_wrapper, bands_range, hpsi_out);
119119
phm->ops->hPsi(info);
120120
};
121-
auto spsi_func = [phm](const double* psi_in, double* spsi_out,const int nrow, const int npw, const int nbands){
122-
phm->sPsi(psi_in, spsi_out, nrow, npw, nbands);
121+
auto spsi_func = [phm](const double* psi_in, double* spsi_out,const int ld_psi, const int nbands){
122+
phm->sPsi(psi_in, spsi_out, ld_psi, ld_psi, nbands);
123123
};
124124
dav.diag(hpsi_func,spsi_func, ld_psi, phi.get_pointer(), en, eps, maxiter);
125125

source/module_hsolver/test/diago_david_test.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -118,8 +118,8 @@ class DiagoDavPrepare
118118
hpsi_info info(&psi_iter_wrapper, bands_range, hpsi_out);
119119
phm->ops->hPsi(info);
120120
};
121-
auto spsi_func = [phm](const std::complex<double>* psi_in, std::complex<double>* spsi_out,const int nrow, const int npw, const int nbands){
122-
phm->sPsi(psi_in, spsi_out, nrow, npw, nbands);
121+
auto spsi_func = [phm](const std::complex<double>* psi_in, std::complex<double>* spsi_out,const int ld_psi, const int nbands){
122+
phm->sPsi(psi_in, spsi_out, ld_psi, ld_psi, nbands);
123123
};
124124
dav.diag(hpsi_func,spsi_func, ld_psi, phi.get_pointer(), en, eps, maxiter);
125125

source/module_hsolver/test/hsolver_pw_sup.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,7 @@ DiagoDavid<T, Device>::~DiagoDavid() {
154154

155155
template <typename T, typename Device>
156156
int DiagoDavid<T, Device>::diag(const std::function<void(T*, T*, const int, const int)>& hpsi_func,
157-
const std::function<void(T*, T*, const int, const int, const int)>& spsi_func,
157+
const std::function<void(T*, T*, const int, const int)>& spsi_func,
158158
const int ld_psi,
159159
T *psi_in,
160160
Real* eigenvalue_in,

0 commit comments

Comments
 (0)