mirror of
https://github.com/paboyle/Grid.git
synced 2025-06-14 13:57:07 +01:00
fd933420c6
Added a bounds-check function for the RHMC with arbitrary power
Added a pseudofermion action for the rational ratio with an arbitrary power and a mixed-precision variant of the same. The existing one-flavor rational ratio class now uses the general class under the hood
To support testing of the two-flavor even-odd ratio pseudofermion, separated the functionality of generating the random field and performing the heatbath step, and added a method to obtain the pseudofermion field
Added a new HMC runner start type: CheckpointStartReseed, which reseeds the RNG from scratch, allowing for the creation of new evolution streams from an existing checkpoint. Added log output of seeds used when the RNG is seeded.
EOFA changes:
To support mixed-precision inversion, generalized the class to maintain a separate solver for the L and R operators in the heatbath (separate solvers are already implemented for the other stages)
To support mixed-precision, the action of setting the operator shift coefficients is now maintained in a virtual function. A derived class for mixed-precision solvers ensures the coefficients are applied to both the double and single-prec operators
The ||^2 of the random source is now stored by the heatbath and compared to the initial action when it is computed. These should be equal but may differ if the rational bounds are not chosen correctly, hence serving as a useful and free test
Fixed calculation of M_eofa (previously incomplete and #if'd out)
Added functionality to compute M_eofa^-1 to complement the calculation of M_eofa (both are equally expensive!)
To support testing, separated the functionality of generating the random field and performing the heatbath step, and added a method to obtain the pseudofermion field
Added a test program which computes the G-parity force using the 1 and 2 flavor implementations and compares the result. Test supports DWF, EOFA and DSDR actions, chosen by a command line option.
The Mobius EOFA force test now also checks the rational approximation used for the heatbath
Added a test program for the mixed precision EOFA compared to the double-prec implementation,
G-parity HMC test now applied GPBC in the y direction and not the t direction (GPBC in t are no longer supported) and checkpoints after every configuration
Added a test program which computes the two-flavor G-parity action (via RHMC) with both the 1 and 2 flavor implementations and checks they agree
Added a test program to check the implementation of M_eofa^{-1}
130 lines
4.5 KiB
C++
130 lines
4.5 KiB
C++
#pragma once
|
|
|
|
NAMESPACE_BEGIN(Grid);
|
|
|
|
template<class Field>
|
|
void HighBoundCheck(LinearOperatorBase<Field> &HermOp,
|
|
Field &Phi,
|
|
RealD hi)
|
|
{
|
|
// Eigenvalue bound check at high end
|
|
PowerMethod<Field> power_method;
|
|
auto lambda_max = power_method(HermOp,Phi);
|
|
std::cout << GridLogMessage << "Pseudofermion action lamda_max "<<lambda_max<<"( bound "<<hi<<")"<<std::endl;
|
|
assert( (lambda_max < hi) && " High Bounds Check on operator failed" );
|
|
}
|
|
|
|
template<class Field> void ChebyBoundsCheck(LinearOperatorBase<Field> &HermOp,
|
|
Field &GaussNoise,
|
|
RealD lo,RealD hi)
|
|
{
|
|
int orderfilter = 1000;
|
|
Chebyshev<Field> Cheb(lo,hi,orderfilter);
|
|
|
|
GridBase *FermionGrid = GaussNoise.Grid();
|
|
|
|
Field X(FermionGrid);
|
|
Field Z(FermionGrid);
|
|
|
|
X=GaussNoise;
|
|
RealD Nx = norm2(X);
|
|
Cheb(HermOp,X,Z);
|
|
RealD Nz = norm2(Z);
|
|
|
|
std::cout << "************************* "<<std::endl;
|
|
std::cout << " noise = "<<Nx<<std::endl;
|
|
std::cout << " Cheb x noise = "<<Nz<<std::endl;
|
|
std::cout << " Ratio = "<<Nz/Nx<<std::endl;
|
|
std::cout << "************************* "<<std::endl;
|
|
assert( ((Nz/Nx)<1.0) && " ChebyBoundsCheck ");
|
|
}
|
|
|
|
template<class Field> void InverseSqrtBoundsCheck(int MaxIter,double tol,
|
|
LinearOperatorBase<Field> &HermOp,
|
|
Field &GaussNoise,
|
|
MultiShiftFunction &PowerNegHalf)
|
|
{
|
|
GridBase *FermionGrid = GaussNoise.Grid();
|
|
|
|
Field X(FermionGrid);
|
|
Field Y(FermionGrid);
|
|
Field Z(FermionGrid);
|
|
|
|
X=GaussNoise;
|
|
RealD Nx = norm2(X);
|
|
|
|
ConjugateGradientMultiShift<Field> msCG(MaxIter,PowerNegHalf);
|
|
msCG(HermOp,X,Y);
|
|
msCG(HermOp,Y,Z);
|
|
|
|
RealD Nz = norm2(Z);
|
|
|
|
HermOp.HermOp(Z,Y);
|
|
RealD Ny = norm2(Y);
|
|
|
|
X=X-Y;
|
|
RealD Nd = norm2(X);
|
|
std::cout << "************************* "<<std::endl;
|
|
std::cout << " | noise |^2 = "<<Nx<<std::endl;
|
|
std::cout << " | (MdagM^-1/2)^2 noise |^2 = "<<Nz<<std::endl;
|
|
std::cout << " | MdagM (MdagM^-1/2)^2 noise |^2 = "<<Ny<<std::endl;
|
|
std::cout << " | noise - MdagM (MdagM^-1/2)^2 noise |^2 = "<<Nd<<std::endl;
|
|
std::cout << " | noise - MdagM (MdagM^-1/2)^2 noise|/|noise| = " << std::sqrt(Nd/Nx) << std::endl;
|
|
std::cout << "************************* "<<std::endl;
|
|
assert( (std::sqrt(Nd/Nx)<tol) && " InverseSqrtBoundsCheck ");
|
|
}
|
|
|
|
/* For a HermOp = M^dag M, check the approximation of HermOp^{-1/inv_pow}
|
|
by computing |X - HermOp * [ Hermop^{-1/inv_pow} ]^{inv_pow} X| < tol
|
|
for noise X (aka GaussNoise).
|
|
ApproxNegPow should be the rational approximation for X^{-1/inv_pow}
|
|
*/
|
|
template<class Field> void InversePowerBoundsCheck(int inv_pow,
|
|
int MaxIter,double tol,
|
|
LinearOperatorBase<Field> &HermOp,
|
|
Field &GaussNoise,
|
|
MultiShiftFunction &ApproxNegPow)
|
|
{
|
|
GridBase *FermionGrid = GaussNoise.Grid();
|
|
|
|
Field X(FermionGrid);
|
|
Field Y(FermionGrid);
|
|
Field Z(FermionGrid);
|
|
|
|
Field tmp1(FermionGrid), tmp2(FermionGrid);
|
|
|
|
X=GaussNoise;
|
|
RealD Nx = norm2(X);
|
|
|
|
ConjugateGradientMultiShift<Field> msCG(MaxIter,ApproxNegPow);
|
|
|
|
tmp1 = X;
|
|
|
|
Field* in = &tmp1;
|
|
Field* out = &tmp2;
|
|
for(int i=0;i<inv_pow;i++){ //apply [ Hermop^{-1/inv_pow} ]^{inv_pow} X = HermOp^{-1} X
|
|
msCG(HermOp, *in, *out); //backwards conventions!
|
|
if(i!=inv_pow-1) std::swap(in, out);
|
|
}
|
|
Z = *out;
|
|
|
|
RealD Nz = norm2(Z);
|
|
|
|
HermOp.HermOp(Z,Y);
|
|
RealD Ny = norm2(Y);
|
|
|
|
X=X-Y;
|
|
RealD Nd = norm2(X);
|
|
std::cout << "************************* "<<std::endl;
|
|
std::cout << " | noise |^2 = "<<Nx<<std::endl;
|
|
std::cout << " | (MdagM^-1/" << inv_pow << ")^" << inv_pow << " noise |^2 = "<<Nz<<std::endl;
|
|
std::cout << " | MdagM (MdagM^-1/" << inv_pow << ")^" << inv_pow << " noise |^2 = "<<Ny<<std::endl;
|
|
std::cout << " | noise - MdagM (MdagM^-1/" << inv_pow << ")^" << inv_pow << " noise |^2 = "<<Nd<<std::endl;
|
|
std::cout << " | noise - MdagM (MdagM^-1/" << inv_pow << ")^" << inv_pow << " noise |/| noise | = "<<std::sqrt(Nd/Nx)<<std::endl;
|
|
std::cout << "************************* "<<std::endl;
|
|
assert( (std::sqrt(Nd/Nx)<tol) && " InversePowerBoundsCheck ");
|
|
}
|
|
|
|
NAMESPACE_END(Grid);
|
|
|