Skip to content

Commit 4314ebb

Browse files
committed
Fix PPCG governance check findings
1 parent f55a88e commit 4314ebb

4 files changed

Lines changed: 10 additions & 4 deletions

File tree

source/source_hsolver/diago_iter_assist.h

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ class DiagoIterAssist
2020
public:
2121
static Real PW_DIAG_THR;
2222
static int PW_DIAG_NMAX;
23+
static int PW_DIAG_NDIM;
2324

2425
static Real LCAO_DIAG_THR;
2526
static int LCAO_DIAG_NMAX;
@@ -153,6 +154,9 @@ typename DiagoIterAssist<T, Device>::Real DiagoIterAssist<T, Device>::avg_iter =
153154
template <typename T, typename Device>
154155
int DiagoIterAssist<T, Device>::PW_DIAG_NMAX = 30;
155156

157+
template <typename T, typename Device>
158+
int DiagoIterAssist<T, Device>::PW_DIAG_NDIM = 4;
159+
156160
template <typename T, typename Device>
157161
typename DiagoIterAssist<T, Device>::Real DiagoIterAssist<T, Device>::PW_DIAG_THR = 1.0e-2;
158162

@@ -175,4 +179,4 @@ template <typename T, typename Device>
175179
T DiagoIterAssist<T, Device>::zero = static_cast<T>(0.0);
176180
} // namespace hsolver
177181

178-
#endif
182+
#endif

source/source_hsolver/diago_params.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@ void setup_diago_params_pw(const int istep,
1515
DiagoIterAssist<T, Device>::need_subspace = ((istep == 0 || istep == 1) && iter == 1) ? false : true;
1616
DiagoIterAssist<T, Device>::SCF_ITER = iter;
1717
DiagoIterAssist<T, Device>::PW_DIAG_THR = ethr;
18+
DiagoIterAssist<T, Device>::PW_DIAG_NDIM = inp.pw_diag_ndim;
1819

1920
if (inp.calculation != "nscf")
2021
{
@@ -41,6 +42,7 @@ void setup_diago_params_sdft(const int istep,
4142

4243
DiagoIterAssist<T, Device>::PW_DIAG_THR = ethr;
4344
DiagoIterAssist<T, Device>::PW_DIAG_NMAX = inp.pw_diag_nmax;
45+
DiagoIterAssist<T, Device>::PW_DIAG_NDIM = inp.pw_diag_ndim;
4446
}
4547

4648
/// Template instantiation for CPU

source/source_hsolver/hsolver_pw.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -456,8 +456,8 @@ void HSolverPW<T, Device>::hamiltSolvePsiK(hamilt::Hamilt<T, Device>* hm,
456456
pre_condition.data(),
457457
this->diag_thr,
458458
this->diag_iter_max,
459-
PARAM.inp.pw_diag_ndim,
460-
PARAM.globalv.gamma_only_pw,
459+
DiagoIterAssist<T, Device>::PW_DIAG_NDIM,
460+
this->wfc_basis->gamma_only,
461461
std::is_same<Device, base_device::DEVICE_CPU>());
462462
}
463463
ModuleBase::timer::end("HSolverPW", "solve_psik");

source/source_hsolver/ppcg/diago_ppcg_diag.hpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -178,7 +178,7 @@ double DiagoPPCG<T, Device>::diag(const HPsiFunc& hpsi_func,
178178
line_minimize(psi_in, hpsi_.data(), spsi_.data(),
179179
p.data(), hp.data(), sp.data(), ncol);
180180

181-
const bool do_rr = (iter % rr_step_ == 0);
181+
const bool do_rr = (iter % rr_step_) == 0;
182182
if (do_rr)
183183
{
184184
// Rayleigh-Ritz: full subspace diagonalization.

0 commit comments

Comments
 (0)