2015-05-31 15:09:02 +01:00
|
|
|
#ifndef GRID_QCD_DHOP_H
|
|
|
|
#define GRID_QCD_DHOP_H
|
|
|
|
|
|
|
|
namespace Grid {
|
|
|
|
|
|
|
|
namespace QCD {
|
|
|
|
|
|
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
2015-08-10 20:47:44 +01:00
|
|
|
// Helper routines that implement Wilson stencil for a single site.
|
2015-08-15 23:25:49 +01:00
|
|
|
// Common to both the WilsonFermion and WilsonFermion5D
|
2015-05-31 15:09:02 +01:00
|
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
2015-08-10 20:47:44 +01:00
|
|
|
|
2015-08-15 23:25:49 +01:00
|
|
|
template<class Impl> class WilsonKernels : public FermionOperator<Impl> {
|
2015-08-10 20:47:44 +01:00
|
|
|
public:
|
|
|
|
|
2015-08-15 23:25:49 +01:00
|
|
|
INHERIT_IMPL_TYPES(Impl);
|
|
|
|
typedef FermionOperator<Impl> Base;
|
|
|
|
|
2015-08-10 20:47:44 +01:00
|
|
|
public:
|
2015-11-06 11:30:17 +00:00
|
|
|
void DiracOptDhopSite(StencilImpl &st,DoubledGaugeField &U,
|
2015-08-15 23:25:49 +01:00
|
|
|
std::vector<SiteHalfSpinor,alignedAllocator<SiteHalfSpinor> > &buf,
|
|
|
|
int sF,int sU,const FermionField &in, FermionField &out);
|
2015-06-30 15:03:11 +01:00
|
|
|
|
2015-11-06 11:30:17 +00:00
|
|
|
void DiracOptDhopSiteDag(StencilImpl &st,DoubledGaugeField &U,
|
2015-08-15 23:25:49 +01:00
|
|
|
std::vector<SiteHalfSpinor,alignedAllocator<SiteHalfSpinor> > &buf,
|
|
|
|
int sF,int sU,const FermionField &in,FermionField &out);
|
|
|
|
|
2015-11-06 11:30:17 +00:00
|
|
|
void DiracOptDhopDir(StencilImpl &st,DoubledGaugeField &U,
|
2015-08-15 23:25:49 +01:00
|
|
|
std::vector<SiteHalfSpinor,alignedAllocator<SiteHalfSpinor> > &buf,
|
|
|
|
int sF,int sU,const FermionField &in, FermionField &out,int dirdisp,int gamma);
|
2015-11-06 11:45:51 +00:00
|
|
|
#if defined(AVX512) || defined(IMCI)
|
|
|
|
void DiracOptAsmDhopSite(CartesianStencil &st,DoubledGaugeField &U,
|
|
|
|
std::vector<SiteHalfSpinor,alignedAllocator<SiteHalfSpinor> > &buf,
|
|
|
|
int sF,int sU,const FermionField &in, FermionField &out,uint64_t *);
|
|
|
|
#else
|
|
|
|
void DiracOptAsmDhopSite(CartesianStencil &st,DoubledGaugeField &U,
|
|
|
|
std::vector<SiteHalfSpinor,alignedAllocator<SiteHalfSpinor> > &buf,
|
|
|
|
int sF,int sU,const FermionField &in, FermionField &out,uint64_t *p){
|
|
|
|
DiracOptDhopSite(st,U,buf,sF,sU,in,out); // will template override for Wilson Nc=3
|
|
|
|
}
|
|
|
|
#endif
|
2015-09-08 17:45:30 +01:00
|
|
|
#define HANDOPT
|
|
|
|
#ifdef HANDOPT
|
2015-11-06 11:30:17 +00:00
|
|
|
void DiracOptHandDhopSite(StencilImpl &st,DoubledGaugeField &U,
|
2015-09-08 17:45:30 +01:00
|
|
|
std::vector<SiteHalfSpinor,alignedAllocator<SiteHalfSpinor> > &buf,
|
|
|
|
int sF,int sU,const FermionField &in, FermionField &out);
|
|
|
|
|
2015-11-06 11:30:17 +00:00
|
|
|
void DiracOptHandDhopSiteDag(StencilImpl &st,DoubledGaugeField &U,
|
2015-09-08 17:45:30 +01:00
|
|
|
std::vector<SiteHalfSpinor,alignedAllocator<SiteHalfSpinor> > &buf,
|
|
|
|
int sF,int sU,const FermionField &in, FermionField &out);
|
|
|
|
#else
|
2015-08-15 23:25:49 +01:00
|
|
|
|
2015-11-06 11:30:17 +00:00
|
|
|
void DiracOptHandDhopSite(StencilImpl &st,DoubledGaugeField &U,
|
2015-08-15 23:25:49 +01:00
|
|
|
std::vector<SiteHalfSpinor,alignedAllocator<SiteHalfSpinor> > &buf,
|
2015-09-08 17:45:30 +01:00
|
|
|
int sF,int sU,const FermionField &in, FermionField &out)
|
|
|
|
{
|
2015-08-15 23:25:49 +01:00
|
|
|
DiracOptDhopSite(st,U,buf,sF,sU,in,out); // will template override for Wilson Nc=3
|
|
|
|
}
|
|
|
|
|
2015-11-06 11:30:17 +00:00
|
|
|
void DiracOptHandDhopSiteDag(StencilImpl &st,DoubledGaugeField &U,
|
2015-08-15 23:25:49 +01:00
|
|
|
std::vector<SiteHalfSpinor,alignedAllocator<SiteHalfSpinor> > &buf,
|
2015-09-08 17:45:30 +01:00
|
|
|
int sF,int sU,const FermionField &in, FermionField &out)
|
|
|
|
{
|
2015-08-15 23:25:49 +01:00
|
|
|
DiracOptDhopSiteDag(st,U,buf,sF,sU,in,out); // will template override for Wilson Nc=3
|
|
|
|
}
|
2015-09-08 17:45:30 +01:00
|
|
|
#endif
|
2015-08-15 23:25:49 +01:00
|
|
|
|
|
|
|
WilsonKernels(const ImplParams &p= ImplParams()) : Base(p) {};
|
2015-08-10 20:47:44 +01:00
|
|
|
|
|
|
|
};
|
2015-05-31 15:09:02 +01:00
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
#endif
|