mirror of
https://github.com/paboyle/Grid.git
synced 2025-04-09 21:50:45 +01:00
Namespace, indenting
This commit is contained in:
paboyle
parent
19234fb40e
commit
fe0467df1e
@ -1,4 +1,4 @@
|
|||||||
/*************************************************************************************
|
/*************************************************************************************
|
||||||
|
|
||||||
Grid physics library, www.github.com/paboyle/Grid
|
Grid physics library, www.github.com/paboyle/Grid
|
||||||
|
|
||||||
@ -24,175 +24,173 @@ Author: Peter Boyle <paboyle@ph.ed.ac.uk>
|
|||||||
51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
||||||
|
|
||||||
See the full license in the file "LICENSE" in the top level distribution directory
|
See the full license in the file "LICENSE" in the top level distribution directory
|
||||||
*************************************************************************************/
|
*************************************************************************************/
|
||||||
/* END LEGAL */
|
/* END LEGAL */
|
||||||
#ifndef GRID_QCD_CAYLEY_FERMION_H
|
#ifndef GRID_QCD_CAYLEY_FERMION_H
|
||||||
#define GRID_QCD_CAYLEY_FERMION_H
|
#define GRID_QCD_CAYLEY_FERMION_H
|
||||||
|
|
||||||
#include <Grid/qcd/action/fermion/WilsonFermion5D.h>
|
#include <Grid/qcd/action/fermion/WilsonFermion5D.h>
|
||||||
|
|
||||||
namespace Grid {
|
NAMESPACE_BEGIN(Grid);
|
||||||
|
|
||||||
namespace QCD {
|
template<typename T> struct switcheroo {
|
||||||
|
static inline int iscomplex() { return 0; }
|
||||||
template<typename T> struct switcheroo {
|
|
||||||
static inline int iscomplex() { return 0; }
|
|
||||||
|
|
||||||
template<class vec>
|
|
||||||
static inline vec mult(vec a, vec b) {
|
|
||||||
return real_mult(a,b);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
template<> struct switcheroo<ComplexD> {
|
|
||||||
static inline int iscomplex() { return 1; }
|
|
||||||
|
|
||||||
template<class vec>
|
|
||||||
static inline vec mult(vec a, vec b) {
|
|
||||||
return a*b;
|
|
||||||
}
|
|
||||||
};
|
|
||||||
template<> struct switcheroo<ComplexF> {
|
|
||||||
static inline int iscomplex() { return 1; }
|
|
||||||
template<class vec>
|
|
||||||
static inline vec mult(vec a, vec b) {
|
|
||||||
return a*b;
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
|
|
||||||
template<class Impl>
|
|
||||||
class CayleyFermion5D : public WilsonFermion5D<Impl>
|
|
||||||
{
|
|
||||||
public:
|
|
||||||
INHERIT_IMPL_TYPES(Impl);
|
|
||||||
public:
|
|
||||||
|
|
||||||
// override multiply
|
|
||||||
virtual RealD M (const FermionField &in, FermionField &out);
|
|
||||||
virtual RealD Mdag (const FermionField &in, FermionField &out);
|
|
||||||
|
|
||||||
// half checkerboard operations
|
|
||||||
virtual void Meooe (const FermionField &in, FermionField &out);
|
|
||||||
virtual void MeooeDag (const FermionField &in, FermionField &out);
|
|
||||||
virtual void Mooee (const FermionField &in, FermionField &out);
|
|
||||||
virtual void MooeeDag (const FermionField &in, FermionField &out);
|
|
||||||
virtual void MooeeInv (const FermionField &in, FermionField &out);
|
|
||||||
virtual void MooeeInvDag (const FermionField &in, FermionField &out);
|
|
||||||
virtual void Meo5D (const FermionField &psi, FermionField &chi);
|
|
||||||
|
|
||||||
virtual void M5D (const FermionField &psi, FermionField &chi);
|
|
||||||
virtual void M5Ddag(const FermionField &psi, FermionField &chi);
|
|
||||||
|
|
||||||
virtual void Dminus(const FermionField &psi, FermionField &chi);
|
|
||||||
virtual void DminusDag(const FermionField &psi, FermionField &chi);
|
|
||||||
|
|
||||||
/////////////////////////////////////////////////////
|
|
||||||
// Instantiate different versions depending on Impl
|
|
||||||
/////////////////////////////////////////////////////
|
|
||||||
void M5D(const FermionField &psi,
|
|
||||||
const FermionField &phi,
|
|
||||||
FermionField &chi,
|
|
||||||
std::vector<Coeff_t> &lower,
|
|
||||||
std::vector<Coeff_t> &diag,
|
|
||||||
std::vector<Coeff_t> &upper);
|
|
||||||
|
|
||||||
void M5Ddag(const FermionField &psi,
|
|
||||||
const FermionField &phi,
|
|
||||||
FermionField &chi,
|
|
||||||
std::vector<Coeff_t> &lower,
|
|
||||||
std::vector<Coeff_t> &diag,
|
|
||||||
std::vector<Coeff_t> &upper);
|
|
||||||
|
|
||||||
void MooeeInternal(const FermionField &in, FermionField &out,int dag,int inv);
|
|
||||||
void MooeeInternalCompute(int dag, int inv, Vector<iSinglet<Simd> > & Matp, Vector<iSinglet<Simd> > & Matm);
|
|
||||||
|
|
||||||
void MooeeInternalAsm(const FermionField &in, FermionField &out,
|
|
||||||
int LLs, int site,
|
|
||||||
Vector<iSinglet<Simd> > &Matp,
|
|
||||||
Vector<iSinglet<Simd> > &Matm);
|
|
||||||
void MooeeInternalZAsm(const FermionField &in, FermionField &out,
|
|
||||||
int LLs, int site,
|
|
||||||
Vector<iSinglet<Simd> > &Matp,
|
|
||||||
Vector<iSinglet<Simd> > &Matm);
|
|
||||||
|
|
||||||
|
|
||||||
virtual void Instantiatable(void)=0;
|
|
||||||
|
|
||||||
// force terms; five routines; default to Dhop on diagonal
|
|
||||||
virtual void MDeriv (GaugeField &mat,const FermionField &U,const FermionField &V,int dag);
|
|
||||||
virtual void MoeDeriv(GaugeField &mat,const FermionField &U,const FermionField &V,int dag);
|
|
||||||
virtual void MeoDeriv(GaugeField &mat,const FermionField &U,const FermionField &V,int dag);
|
|
||||||
|
|
||||||
// Efficient support for multigrid coarsening
|
|
||||||
virtual void Mdir (const FermionField &in, FermionField &out,int dir,int disp);
|
|
||||||
|
|
||||||
void Meooe5D (const FermionField &in, FermionField &out);
|
|
||||||
void MeooeDag5D (const FermionField &in, FermionField &out);
|
|
||||||
|
|
||||||
// protected:
|
|
||||||
RealD mass;
|
|
||||||
|
|
||||||
// Cayley form Moebius (tanh and zolotarev)
|
|
||||||
std::vector<Coeff_t> omega;
|
|
||||||
std::vector<Coeff_t> bs; // S dependent coeffs
|
|
||||||
std::vector<Coeff_t> cs;
|
|
||||||
std::vector<Coeff_t> as;
|
|
||||||
// For preconditioning Cayley form
|
|
||||||
std::vector<Coeff_t> bee;
|
|
||||||
std::vector<Coeff_t> cee;
|
|
||||||
std::vector<Coeff_t> aee;
|
|
||||||
std::vector<Coeff_t> beo;
|
|
||||||
std::vector<Coeff_t> ceo;
|
|
||||||
std::vector<Coeff_t> aeo;
|
|
||||||
// LDU factorisation of the eeoo matrix
|
|
||||||
std::vector<Coeff_t> lee;
|
|
||||||
std::vector<Coeff_t> leem;
|
|
||||||
std::vector<Coeff_t> uee;
|
|
||||||
std::vector<Coeff_t> ueem;
|
|
||||||
std::vector<Coeff_t> dee;
|
|
||||||
|
|
||||||
// Matrices of 5d ee inverse params
|
|
||||||
Vector<iSinglet<Simd> > MatpInv;
|
|
||||||
Vector<iSinglet<Simd> > MatmInv;
|
|
||||||
Vector<iSinglet<Simd> > MatpInvDag;
|
|
||||||
Vector<iSinglet<Simd> > MatmInvDag;
|
|
||||||
|
|
||||||
// Constructors
|
|
||||||
CayleyFermion5D(GaugeField &_Umu,
|
|
||||||
GridCartesian &FiveDimGrid,
|
|
||||||
GridRedBlackCartesian &FiveDimRedBlackGrid,
|
|
||||||
GridCartesian &FourDimGrid,
|
|
||||||
GridRedBlackCartesian &FourDimRedBlackGrid,
|
|
||||||
RealD _mass,RealD _M5,const ImplParams &p= ImplParams());
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
void CayleyReport(void);
|
|
||||||
void CayleyZeroCounters(void);
|
|
||||||
|
|
||||||
double M5Dflops;
|
|
||||||
double M5Dcalls;
|
|
||||||
double M5Dtime;
|
|
||||||
|
|
||||||
double MooeeInvFlops;
|
|
||||||
double MooeeInvCalls;
|
|
||||||
double MooeeInvTime;
|
|
||||||
|
|
||||||
protected:
|
|
||||||
virtual void SetCoefficientsZolotarev(RealD zolohi,Approx::zolotarev_data *zdata,RealD b,RealD c);
|
|
||||||
virtual void SetCoefficientsTanh(Approx::zolotarev_data *zdata,RealD b,RealD c);
|
|
||||||
virtual void SetCoefficientsInternal(RealD zolo_hi,std::vector<Coeff_t> & gamma,RealD b,RealD c);
|
|
||||||
};
|
|
||||||
|
|
||||||
|
template<class vec>
|
||||||
|
static inline vec mult(vec a, vec b) {
|
||||||
|
return real_mult(a,b);
|
||||||
}
|
}
|
||||||
}
|
};
|
||||||
#define INSTANTIATE_DPERP(A)\
|
template<> struct switcheroo<ComplexD> {
|
||||||
template void CayleyFermion5D< A >::M5D(const FermionField &psi,const FermionField &phi,FermionField &chi,\
|
static inline int iscomplex() { return 1; }
|
||||||
std::vector<Coeff_t> &lower,std::vector<Coeff_t> &diag,std::vector<Coeff_t> &upper); \
|
|
||||||
template void CayleyFermion5D< A >::M5Ddag(const FermionField &psi,const FermionField &phi,FermionField &chi,\
|
template<class vec>
|
||||||
std::vector<Coeff_t> &lower,std::vector<Coeff_t> &diag,std::vector<Coeff_t> &upper); \
|
static inline vec mult(vec a, vec b) {
|
||||||
template void CayleyFermion5D< A >::MooeeInv (const FermionField &psi, FermionField &chi); \
|
return a*b;
|
||||||
template void CayleyFermion5D< A >::MooeeInvDag (const FermionField &psi, FermionField &chi);
|
}
|
||||||
|
};
|
||||||
|
template<> struct switcheroo<ComplexF> {
|
||||||
|
static inline int iscomplex() { return 1; }
|
||||||
|
template<class vec>
|
||||||
|
static inline vec mult(vec a, vec b) {
|
||||||
|
return a*b;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
template<class Impl>
|
||||||
|
class CayleyFermion5D : public WilsonFermion5D<Impl>
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
INHERIT_IMPL_TYPES(Impl);
|
||||||
|
public:
|
||||||
|
|
||||||
|
// override multiply
|
||||||
|
virtual RealD M (const FermionField &in, FermionField &out);
|
||||||
|
virtual RealD Mdag (const FermionField &in, FermionField &out);
|
||||||
|
|
||||||
|
// half checkerboard operations
|
||||||
|
virtual void Meooe (const FermionField &in, FermionField &out);
|
||||||
|
virtual void MeooeDag (const FermionField &in, FermionField &out);
|
||||||
|
virtual void Mooee (const FermionField &in, FermionField &out);
|
||||||
|
virtual void MooeeDag (const FermionField &in, FermionField &out);
|
||||||
|
virtual void MooeeInv (const FermionField &in, FermionField &out);
|
||||||
|
virtual void MooeeInvDag (const FermionField &in, FermionField &out);
|
||||||
|
virtual void Meo5D (const FermionField &psi, FermionField &chi);
|
||||||
|
|
||||||
|
virtual void M5D (const FermionField &psi, FermionField &chi);
|
||||||
|
virtual void M5Ddag(const FermionField &psi, FermionField &chi);
|
||||||
|
|
||||||
|
virtual void Dminus(const FermionField &psi, FermionField &chi);
|
||||||
|
virtual void DminusDag(const FermionField &psi, FermionField &chi);
|
||||||
|
|
||||||
|
/////////////////////////////////////////////////////
|
||||||
|
// Instantiate different versions depending on Impl
|
||||||
|
/////////////////////////////////////////////////////
|
||||||
|
void M5D(const FermionField &psi,
|
||||||
|
const FermionField &phi,
|
||||||
|
FermionField &chi,
|
||||||
|
std::vector<Coeff_t> &lower,
|
||||||
|
std::vector<Coeff_t> &diag,
|
||||||
|
std::vector<Coeff_t> &upper);
|
||||||
|
|
||||||
|
void M5Ddag(const FermionField &psi,
|
||||||
|
const FermionField &phi,
|
||||||
|
FermionField &chi,
|
||||||
|
std::vector<Coeff_t> &lower,
|
||||||
|
std::vector<Coeff_t> &diag,
|
||||||
|
std::vector<Coeff_t> &upper);
|
||||||
|
|
||||||
|
void MooeeInternal(const FermionField &in, FermionField &out,int dag,int inv);
|
||||||
|
void MooeeInternalCompute(int dag, int inv, Vector<iSinglet<Simd> > & Matp, Vector<iSinglet<Simd> > & Matm);
|
||||||
|
|
||||||
|
void MooeeInternalAsm(const FermionField &in, FermionField &out,
|
||||||
|
int LLs, int site,
|
||||||
|
Vector<iSinglet<Simd> > &Matp,
|
||||||
|
Vector<iSinglet<Simd> > &Matm);
|
||||||
|
void MooeeInternalZAsm(const FermionField &in, FermionField &out,
|
||||||
|
int LLs, int site,
|
||||||
|
Vector<iSinglet<Simd> > &Matp,
|
||||||
|
Vector<iSinglet<Simd> > &Matm);
|
||||||
|
|
||||||
|
|
||||||
|
virtual void Instantiatable(void)=0;
|
||||||
|
|
||||||
|
// force terms; five routines; default to Dhop on diagonal
|
||||||
|
virtual void MDeriv (GaugeField &mat,const FermionField &U,const FermionField &V,int dag);
|
||||||
|
virtual void MoeDeriv(GaugeField &mat,const FermionField &U,const FermionField &V,int dag);
|
||||||
|
virtual void MeoDeriv(GaugeField &mat,const FermionField &U,const FermionField &V,int dag);
|
||||||
|
|
||||||
|
// Efficient support for multigrid coarsening
|
||||||
|
virtual void Mdir (const FermionField &in, FermionField &out,int dir,int disp);
|
||||||
|
|
||||||
|
void Meooe5D (const FermionField &in, FermionField &out);
|
||||||
|
void MeooeDag5D (const FermionField &in, FermionField &out);
|
||||||
|
|
||||||
|
// protected:
|
||||||
|
RealD mass;
|
||||||
|
|
||||||
|
// Cayley form Moebius (tanh and zolotarev)
|
||||||
|
std::vector<Coeff_t> omega;
|
||||||
|
std::vector<Coeff_t> bs; // S dependent coeffs
|
||||||
|
std::vector<Coeff_t> cs;
|
||||||
|
std::vector<Coeff_t> as;
|
||||||
|
// For preconditioning Cayley form
|
||||||
|
std::vector<Coeff_t> bee;
|
||||||
|
std::vector<Coeff_t> cee;
|
||||||
|
std::vector<Coeff_t> aee;
|
||||||
|
std::vector<Coeff_t> beo;
|
||||||
|
std::vector<Coeff_t> ceo;
|
||||||
|
std::vector<Coeff_t> aeo;
|
||||||
|
// LDU factorisation of the eeoo matrix
|
||||||
|
std::vector<Coeff_t> lee;
|
||||||
|
std::vector<Coeff_t> leem;
|
||||||
|
std::vector<Coeff_t> uee;
|
||||||
|
std::vector<Coeff_t> ueem;
|
||||||
|
std::vector<Coeff_t> dee;
|
||||||
|
|
||||||
|
// Matrices of 5d ee inverse params
|
||||||
|
Vector<iSinglet<Simd> > MatpInv;
|
||||||
|
Vector<iSinglet<Simd> > MatmInv;
|
||||||
|
Vector<iSinglet<Simd> > MatpInvDag;
|
||||||
|
Vector<iSinglet<Simd> > MatmInvDag;
|
||||||
|
|
||||||
|
// Constructors
|
||||||
|
CayleyFermion5D(GaugeField &_Umu,
|
||||||
|
GridCartesian &FiveDimGrid,
|
||||||
|
GridRedBlackCartesian &FiveDimRedBlackGrid,
|
||||||
|
GridCartesian &FourDimGrid,
|
||||||
|
GridRedBlackCartesian &FourDimRedBlackGrid,
|
||||||
|
RealD _mass,RealD _M5,const ImplParams &p= ImplParams());
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
void CayleyReport(void);
|
||||||
|
void CayleyZeroCounters(void);
|
||||||
|
|
||||||
|
double M5Dflops;
|
||||||
|
double M5Dcalls;
|
||||||
|
double M5Dtime;
|
||||||
|
|
||||||
|
double MooeeInvFlops;
|
||||||
|
double MooeeInvCalls;
|
||||||
|
double MooeeInvTime;
|
||||||
|
|
||||||
|
protected:
|
||||||
|
virtual void SetCoefficientsZolotarev(RealD zolohi,Approx::zolotarev_data *zdata,RealD b,RealD c);
|
||||||
|
virtual void SetCoefficientsTanh(Approx::zolotarev_data *zdata,RealD b,RealD c);
|
||||||
|
virtual void SetCoefficientsInternal(RealD zolo_hi,std::vector<Coeff_t> & gamma,RealD b,RealD c);
|
||||||
|
};
|
||||||
|
|
||||||
|
NAMESPACE_END(Grid);
|
||||||
|
|
||||||
|
#define INSTANTIATE_DPERP(A) \
|
||||||
|
template void CayleyFermion5D< A >::M5D(const FermionField &psi,const FermionField &phi,FermionField &chi, \
|
||||||
|
std::vector<Coeff_t> &lower,std::vector<Coeff_t> &diag,std::vector<Coeff_t> &upper); \
|
||||||
|
template void CayleyFermion5D< A >::M5Ddag(const FermionField &psi,const FermionField &phi,FermionField &chi, \
|
||||||
|
std::vector<Coeff_t> &lower,std::vector<Coeff_t> &diag,std::vector<Coeff_t> &upper); \
|
||||||
|
template void CayleyFermion5D< A >::MooeeInv (const FermionField &psi, FermionField &chi); \
|
||||||
|
template void CayleyFermion5D< A >::MooeeInvDag (const FermionField &psi, FermionField &chi);
|
||||||
|
|
||||||
#undef CAYLEY_DPERP_DENSE
|
#undef CAYLEY_DPERP_DENSE
|
||||||
#define CAYLEY_DPERP_CACHE
|
#define CAYLEY_DPERP_CACHE
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user