mirror of
https://github.com/paboyle/Grid.git
synced 2025-04-06 20:25:56 +01:00
Changes to Hadrons SchurRB solver to allow for a subtract_guess boolean to be passed
This commit is contained in:
parent
f4c6d39238
commit
c18074869b
@ -116,7 +116,7 @@ std::vector<std::string> TRBPrecCG<FImpl, nBasis>::getReference(void)
|
||||
template <typename FImpl, int nBasis>
|
||||
std::vector<std::string> TRBPrecCG<FImpl, nBasis>::getOutput(void)
|
||||
{
|
||||
std::vector<std::string> out = {getName()};
|
||||
std::vector<std::string> out = {getName(), getName() + "_subtract"};
|
||||
|
||||
return out;
|
||||
}
|
||||
@ -166,16 +166,21 @@ void TRBPrecCG<FImpl, nBasis>::setup(void)
|
||||
guesser.reset(new FineGuesser(epack.evec, epack.eval));
|
||||
}
|
||||
}
|
||||
auto solver = [&mat, guesser, this](FermionField &sol,
|
||||
const FermionField &source)
|
||||
{
|
||||
ConjugateGradient<FermionField> cg(par().residual,
|
||||
par().maxIteration);
|
||||
HADRONS_DEFAULT_SCHUR_SOLVE<FermionField> schurSolver(cg);
|
||||
|
||||
schurSolver(mat, source, sol, *guesser);
|
||||
auto makeSolver = [&mat, guesser, this](bool subGuess) {
|
||||
return [&mat, guesser, subGuess, this](FermionField &sol,
|
||||
const FermionField &source) {
|
||||
ConjugateGradient<FermionField> cg(par().residual,
|
||||
par().maxIteration);
|
||||
HADRONS_DEFAULT_SCHUR_SOLVE<FermionField> schurSolver(cg);
|
||||
schurSolver.subtractGuess(subGuess);
|
||||
schurSolver(mat, source, sol, *guesser);
|
||||
};
|
||||
};
|
||||
|
||||
auto solver = makeSolver(false);
|
||||
envCreate(SolverFn, getName(), Ls, solver);
|
||||
auto solver_subtract = makeSolver(true);
|
||||
envCreate(SolverFn, getName() + "_subtract", Ls, solver_subtract);
|
||||
}
|
||||
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user