portelli/Grid
1
0
Fork 0
mirror of https://github.com/paboyle/Grid.git synced 2025-06-15 06:17:05 +01:00
Code Releases Activity

Improvement in the CG interface for Repro

Browse Source
This commit is contained in:
Guido Cossu
2016-12-09 05:20:38 +00:00
parent 6ceee102e8
commit ec0c53fa68
3 changed files with 27 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

49
lib/algorithms/iterative/ConjugateGradient.h
View File

@ -47,7 +47,7 @@ struct CG_state {
};
enum CGexec_modes{ Default, ReproducibilityTest };
enum CGexec_mode{ Default, ReproducibilityTest };
/////////////////////////////////////////////////////////////
// Base classes for iterative processes based on operators
@ -67,19 +67,20 @@ class ConjugateGradient : public OperatorFunction<Field> {
CG_state CGState; //to check reproducibility by repeating the CG
ReproducibilityState<typename Field::vector_object> ReprTest; // for the inner proucts
ConjugateGradient(RealD tol, Integer maxit, bool err_on_no_conv = true,
bool ReproducibilityTest = false)
: Tolerance(tol),
MaxIterations(maxit),
ErrorOnNoConverge(err_on_no_conv),
ReproTest(ReproducibilityTest){
if(ReproducibilityTest == true && err_on_no_conv == true){
std::cout << GridLogMessage << "CG: Reproducibility test ON "<<
"and error on convergence ON are incompatible options" << std::endl;
exit(1);
}
};
// Constructor
ConjugateGradient(RealD tol, Integer maxit, CGexec_mode Mode = Default)
: Tolerance(tol),MaxIterations(maxit){
switch(Mode)
{
case Default :
ErrorOnNoConverge = true;
ReproTest = false;
case ReproducibilityTest :
ErrorOnNoConverge = false;
ReproTest = true;
}
};
void operator()(LinearOperatorBase<Field> &Linop, const Field &src,
@ -116,18 +117,12 @@ class ConjugateGradient : public OperatorFunction<Field> {
cp = a;
ssq = norm2(src);
std::cout << GridLogIterative << std::setprecision(4)
<< "ConjugateGradient: guess " << guess << std::endl;
std::cout << GridLogIterative << std::setprecision(4)
<< "ConjugateGradient: src " << ssq << std::endl;
std::cout << GridLogIterative << std::setprecision(4)
<< "ConjugateGradient: mp " << d << std::endl;
std::cout << GridLogIterative << std::setprecision(4)
<< "ConjugateGradient: mmp " << b << std::endl;
std::cout << GridLogIterative << std::setprecision(4)
<< "ConjugateGradient: cp,r " << cp << std::endl;
std::cout << GridLogIterative << std::setprecision(4)
<< "ConjugateGradient: p " << a << std::endl;
std::cout << GridLogIterative << "ConjugateGradient: guess " << guess << std::endl;
std::cout << GridLogIterative << "ConjugateGradient: src " << ssq << std::endl;
std::cout << GridLogIterative << "ConjugateGradient: mp " << d << std::endl;
std::cout << GridLogIterative << "ConjugateGradient: mmp " << b << std::endl;
std::cout << GridLogIterative << "ConjugateGradient: cp,r " << cp << std::endl;
std::cout << GridLogIterative << "ConjugateGradient: p " << a << std::endl;
RealD rsq = Tolerance * Tolerance * ssq;
@ -162,7 +157,7 @@ class ConjugateGradient : public OperatorFunction<Field> {
axpy(r, -a, mmp, r);// new residual r = r_old - a * Ap
cp = norm2(r, ReprTest); //
cp = norm2(r, ReprTest); // bookmarking this norm
if (ReproTest && !CGState.do_repro) {
CGState.residuals.push_back(cp); // save residuals state
std::cout << GridLogIterative << "ReproTest: Saving state" << std::endl;

6
tests/debug/Test_cayley_cg_reproducibility.cc
View File

@ -168,7 +168,7 @@ void TestCGunprec(What & Ddwf,
LatticeFermion result(FGrid); result=zero;
MdagMLinearOperator<What,LatticeFermion> HermOp(Ddwf);
ConjugateGradient<LatticeFermion> CG(1.0e-8,10000, false, true);
ConjugateGradient<LatticeFermion> CG(1.0e-8,10000, ReproducibilityTest);
CG(HermOp,src,result);
}
@ -187,7 +187,7 @@ void TestCGprec(What & Ddwf,
result_o=zero;
SchurDiagMooeeOperator<What,LatticeFermion> HermOpEO(Ddwf);
ConjugateGradient<LatticeFermion> CG(1.0e-8,10000, false, true);
ConjugateGradient<LatticeFermion> CG(1.0e-8,10000, ReproducibilityTest);
CG(HermOpEO,src_o,result_o);
}
@ -203,7 +203,7 @@ void TestCGschur(What & Ddwf,
LatticeFermion src (FGrid); random(*RNG5,src);
LatticeFermion result(FGrid); result=zero;
ConjugateGradient<LatticeFermion> CG(1.0e-8,10000, false, true);
ConjugateGradient<LatticeFermion> CG(1.0e-8,10000, ReproducibilityTest);
SchurRedBlackDiagMooeeSolve<LatticeFermion> SchurSolver(CG);
SchurSolver(Ddwf,src,result);
}

4
tests/hmc/Test_hmc_WilsonFermionGauge.cc
View File

@ -64,9 +64,9 @@ class HmcRunner : public NerscHmcRunner {
FermionAction FermOp(U, *FGrid, *FrbGrid, mass);
// To enable the CG reproducibility tests use
ConjugateGradient<FermionField> CG(1.0e-8, 10000, false, true);
//ConjugateGradient<FermionField> CG(1.0e-8, 10000, ReproducibilityTest);
// This is the plain version
//ConjugateGradient<FermionField> CG(1.0e-8, 10000);
ConjugateGradient<FermionField> CG(1.0e-8, 10000);
TwoFlavourPseudoFermionAction<ImplPolicy> Nf2(FermOp, CG, CG);