portelli/Grid
1
0
Fork 0
mirror of https://github.com/paboyle/Grid.git synced 2025-04-09 05:30:46 +01:00
Code Releases Activity

Changes to Hadrons SchurRB solver to allow for a subtract_guess boolean to be passed

Browse Source
This commit is contained in:
fionnoh 2018-05-31 17:17:16 +01:00
parent f4c6d39238
commit c18074869b
1 changed files with 14 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
extras/Hadrons/Modules/MSolver/RBPrecCG.hpp
View File

@ -116,7 +116,7 @@ std::vector<std::string> TRBPrecCG<FImpl, nBasis>::getReference(void)
template <typename FImpl, int nBasis> template <typename FImpl, int nBasis>
std::vector<std::string> TRBPrecCG<FImpl, nBasis>::getOutput(void) 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; return out;
} }
@ -166,16 +166,21 @@ void TRBPrecCG<FImpl, nBasis>::setup(void)
guesser.reset(new FineGuesser(epack.evec, epack.eval)); guesser.reset(new FineGuesser(epack.evec, epack.eval));
} }
} }
auto solver = [&mat, guesser, this](FermionField &sol, auto makeSolver = [&mat, guesser, this](bool subGuess) {
const FermionField &source) return [&mat, guesser, subGuess, this](FermionField &sol,
{ const FermionField &source) {
ConjugateGradient<FermionField> cg(par().residual, ConjugateGradient<FermionField> cg(par().residual,
par().maxIteration); par().maxIteration);
HADRONS_DEFAULT_SCHUR_SOLVE<FermionField> schurSolver(cg); HADRONS_DEFAULT_SCHUR_SOLVE<FermionField> schurSolver(cg);
schurSolver.subtractGuess(subGuess);
schurSolver(mat, source, sol, *guesser); schurSolver(mat, source, sol, *guesser);
};
}; };
auto solver = makeSolver(false);
envCreate(SolverFn, getName(), Ls, solver); envCreate(SolverFn, getName(), Ls, solver);
auto solver_subtract = makeSolver(true);
envCreate(SolverFn, getName() + "_subtract", Ls, solver_subtract);
} }