mirror of
https://github.com/paboyle/Grid.git
synced 2024-09-20 17:25:37 +01:00
aeb7442d8f
Allows multi-precision work and paves the way for alternate BC's and such like allowing for example G-parity which is important for K pipi programme. In particular, can drive an extra flavour index into the fermion fields using template types.
432 lines
10 KiB
C++
432 lines
10 KiB
C++
#include <Grid.h>
|
|
namespace Grid {
|
|
namespace QCD {
|
|
|
|
template<class Impl>
|
|
void WilsonKernels<Impl>::DiracOptDhopSite(CartesianStencil &st,DoubledGaugeField &U,
|
|
std::vector<SiteHalfSpinor,alignedAllocator<SiteHalfSpinor> > &buf,
|
|
int sF,int sU,const FermionField &in, FermionField &out)
|
|
{
|
|
SiteHalfSpinor tmp;
|
|
SiteHalfSpinor chi;
|
|
SiteSpinor result;
|
|
SiteHalfSpinor Uchi;
|
|
int offset,local,perm, ptype;
|
|
|
|
// Xp
|
|
int ss = sF;
|
|
offset = st._offsets [Xp][ss];
|
|
local = st._is_local[Xp][ss];
|
|
perm = st._permute[Xp][ss];
|
|
|
|
ptype = st._permute_type[Xp];
|
|
if ( local && perm ) {
|
|
spProjXp(tmp,in._odata[offset]);
|
|
permute(chi,tmp,ptype);
|
|
} else if ( local ) {
|
|
spProjXp(chi,in._odata[offset]);
|
|
} else {
|
|
chi=buf[offset];
|
|
}
|
|
Impl::multLink(Uchi,U._odata[sU],chi,Xp);
|
|
spReconXp(result,Uchi);
|
|
|
|
// Yp
|
|
offset = st._offsets [Yp][ss];
|
|
local = st._is_local[Yp][ss];
|
|
perm = st._permute[Yp][ss];
|
|
ptype = st._permute_type[Yp];
|
|
if ( local && perm ) {
|
|
spProjYp(tmp,in._odata[offset]);
|
|
permute(chi,tmp,ptype);
|
|
} else if ( local ) {
|
|
spProjYp(chi,in._odata[offset]);
|
|
} else {
|
|
chi=buf[offset];
|
|
}
|
|
Impl::multLink(Uchi,U._odata[sU],chi,Yp);
|
|
accumReconYp(result,Uchi);
|
|
|
|
// Zp
|
|
offset = st._offsets [Zp][ss];
|
|
local = st._is_local[Zp][ss];
|
|
perm = st._permute[Zp][ss];
|
|
ptype = st._permute_type[Zp];
|
|
if ( local && perm ) {
|
|
spProjZp(tmp,in._odata[offset]);
|
|
permute(chi,tmp,ptype);
|
|
} else if ( local ) {
|
|
spProjZp(chi,in._odata[offset]);
|
|
} else {
|
|
chi=buf[offset];
|
|
}
|
|
Impl::multLink(Uchi,U._odata[sU],chi,Zp);
|
|
accumReconZp(result,Uchi);
|
|
|
|
// Tp
|
|
offset = st._offsets [Tp][ss];
|
|
local = st._is_local[Tp][ss];
|
|
perm = st._permute[Tp][ss];
|
|
ptype = st._permute_type[Tp];
|
|
if ( local && perm ) {
|
|
spProjTp(tmp,in._odata[offset]);
|
|
permute(chi,tmp,ptype);
|
|
} else if ( local ) {
|
|
spProjTp(chi,in._odata[offset]);
|
|
} else {
|
|
chi=buf[offset];
|
|
}
|
|
Impl::multLink(Uchi,U._odata[sU],chi,Tp);
|
|
accumReconTp(result,Uchi);
|
|
|
|
// Xm
|
|
offset = st._offsets [Xm][ss];
|
|
local = st._is_local[Xm][ss];
|
|
perm = st._permute[Xm][ss];
|
|
ptype = st._permute_type[Xm];
|
|
|
|
if ( local && perm ) {
|
|
spProjXm(tmp,in._odata[offset]);
|
|
permute(chi,tmp,ptype);
|
|
} else if ( local ) {
|
|
spProjXm(chi,in._odata[offset]);
|
|
} else {
|
|
chi=buf[offset];
|
|
}
|
|
Impl::multLink(Uchi,U._odata[sU],chi,Xm);
|
|
accumReconXm(result,Uchi);
|
|
|
|
// Ym
|
|
offset = st._offsets [Ym][ss];
|
|
local = st._is_local[Ym][ss];
|
|
perm = st._permute[Ym][ss];
|
|
ptype = st._permute_type[Ym];
|
|
|
|
if ( local && perm ) {
|
|
spProjYm(tmp,in._odata[offset]);
|
|
permute(chi,tmp,ptype);
|
|
} else if ( local ) {
|
|
spProjYm(chi,in._odata[offset]);
|
|
} else {
|
|
chi=buf[offset];
|
|
}
|
|
Impl::multLink(Uchi,U._odata[sU],chi,Ym);
|
|
accumReconYm(result,Uchi);
|
|
|
|
// Zm
|
|
offset = st._offsets [Zm][ss];
|
|
local = st._is_local[Zm][ss];
|
|
perm = st._permute[Zm][ss];
|
|
ptype = st._permute_type[Zm];
|
|
if ( local && perm ) {
|
|
spProjZm(tmp,in._odata[offset]);
|
|
permute(chi,tmp,ptype);
|
|
} else if ( local ) {
|
|
spProjZm(chi,in._odata[offset]);
|
|
} else {
|
|
chi=buf[offset];
|
|
}
|
|
Impl::multLink(Uchi,U._odata[sU],chi,Zm);
|
|
accumReconZm(result,Uchi);
|
|
|
|
// Tm
|
|
offset = st._offsets [Tm][ss];
|
|
local = st._is_local[Tm][ss];
|
|
perm = st._permute[Tm][ss];
|
|
ptype = st._permute_type[Tm];
|
|
if ( local && perm ) {
|
|
spProjTm(tmp,in._odata[offset]);
|
|
permute(chi,tmp,ptype);
|
|
} else if ( local ) {
|
|
spProjTm(chi,in._odata[offset]);
|
|
} else {
|
|
chi=buf[offset];
|
|
}
|
|
Impl::multLink(Uchi,U._odata[sU],chi,Tm);
|
|
accumReconTm(result,Uchi);
|
|
|
|
vstream(out._odata[ss],result*(-0.5));
|
|
};
|
|
|
|
template<class Impl>
|
|
void WilsonKernels<Impl>::DiracOptDhopSiteDag(CartesianStencil &st,DoubledGaugeField &U,
|
|
std::vector<SiteHalfSpinor,alignedAllocator<SiteHalfSpinor> > &buf,
|
|
int sF,int sU,const FermionField &in, FermionField &out)
|
|
{
|
|
SiteHalfSpinor tmp;
|
|
SiteHalfSpinor chi;
|
|
SiteSpinor result;
|
|
SiteHalfSpinor Uchi;
|
|
int offset,local,perm, ptype;
|
|
|
|
// Xp
|
|
int ss=sF;
|
|
offset = st._offsets [Xm][ss];
|
|
local = st._is_local[Xm][ss];
|
|
perm = st._permute[Xm][ss];
|
|
|
|
ptype = st._permute_type[Xm];
|
|
if ( local && perm ) {
|
|
spProjXp(tmp,in._odata[offset]);
|
|
permute(chi,tmp,ptype);
|
|
} else if ( local ) {
|
|
spProjXp(chi,in._odata[offset]);
|
|
} else {
|
|
chi=buf[offset];
|
|
}
|
|
Impl::multLink(Uchi,U._odata[sU],chi,Xm);
|
|
spReconXp(result,Uchi);
|
|
|
|
// Yp
|
|
offset = st._offsets [Ym][ss];
|
|
local = st._is_local[Ym][ss];
|
|
perm = st._permute[Ym][ss];
|
|
ptype = st._permute_type[Ym];
|
|
if ( local && perm ) {
|
|
spProjYp(tmp,in._odata[offset]);
|
|
permute(chi,tmp,ptype);
|
|
} else if ( local ) {
|
|
spProjYp(chi,in._odata[offset]);
|
|
} else {
|
|
chi=buf[offset];
|
|
}
|
|
Impl::multLink(Uchi,U._odata[sU],chi,Ym);
|
|
accumReconYp(result,Uchi);
|
|
|
|
// Zp
|
|
offset = st._offsets [Zm][ss];
|
|
local = st._is_local[Zm][ss];
|
|
perm = st._permute[Zm][ss];
|
|
ptype = st._permute_type[Zm];
|
|
if ( local && perm ) {
|
|
spProjZp(tmp,in._odata[offset]);
|
|
permute(chi,tmp,ptype);
|
|
} else if ( local ) {
|
|
spProjZp(chi,in._odata[offset]);
|
|
} else {
|
|
chi=buf[offset];
|
|
}
|
|
Impl::multLink(Uchi,U._odata[sU],chi,Zm);
|
|
accumReconZp(result,Uchi);
|
|
|
|
// Tp
|
|
offset = st._offsets [Tm][ss];
|
|
local = st._is_local[Tm][ss];
|
|
perm = st._permute[Tm][ss];
|
|
ptype = st._permute_type[Tm];
|
|
if ( local && perm ) {
|
|
spProjTp(tmp,in._odata[offset]);
|
|
permute(chi,tmp,ptype);
|
|
} else if ( local ) {
|
|
spProjTp(chi,in._odata[offset]);
|
|
} else {
|
|
chi=buf[offset];
|
|
}
|
|
Impl::multLink(Uchi,U._odata[sU],chi,Tm);
|
|
accumReconTp(result,Uchi);
|
|
|
|
// Xm
|
|
offset = st._offsets [Xp][ss];
|
|
local = st._is_local[Xp][ss];
|
|
perm = st._permute[Xp][ss];
|
|
ptype = st._permute_type[Xp];
|
|
|
|
if ( local && perm ) {
|
|
spProjXm(tmp,in._odata[offset]);
|
|
permute(chi,tmp,ptype);
|
|
} else if ( local ) {
|
|
spProjXm(chi,in._odata[offset]);
|
|
} else {
|
|
chi=buf[offset];
|
|
}
|
|
Impl::multLink(Uchi,U._odata[sU],chi,Xp);
|
|
accumReconXm(result,Uchi);
|
|
|
|
// Ym
|
|
offset = st._offsets [Yp][ss];
|
|
local = st._is_local[Yp][ss];
|
|
perm = st._permute[Yp][ss];
|
|
ptype = st._permute_type[Yp];
|
|
|
|
if ( local && perm ) {
|
|
spProjYm(tmp,in._odata[offset]);
|
|
permute(chi,tmp,ptype);
|
|
} else if ( local ) {
|
|
spProjYm(chi,in._odata[offset]);
|
|
} else {
|
|
chi=buf[offset];
|
|
}
|
|
Impl::multLink(Uchi,U._odata[sU],chi,Yp);
|
|
accumReconYm(result,Uchi);
|
|
|
|
// Zm
|
|
offset = st._offsets [Zp][ss];
|
|
local = st._is_local[Zp][ss];
|
|
perm = st._permute[Zp][ss];
|
|
ptype = st._permute_type[Zp];
|
|
if ( local && perm ) {
|
|
spProjZm(tmp,in._odata[offset]);
|
|
permute(chi,tmp,ptype);
|
|
} else if ( local ) {
|
|
spProjZm(chi,in._odata[offset]);
|
|
} else {
|
|
chi=buf[offset];
|
|
}
|
|
Impl::multLink(Uchi,U._odata[sU],chi,Zp);
|
|
accumReconZm(result,Uchi);
|
|
|
|
// Tm
|
|
offset = st._offsets [Tp][ss];
|
|
local = st._is_local[Tp][ss];
|
|
perm = st._permute[Tp][ss];
|
|
ptype = st._permute_type[Tp];
|
|
if ( local && perm ) {
|
|
spProjTm(tmp,in._odata[offset]);
|
|
permute(chi,tmp,ptype);
|
|
} else if ( local ) {
|
|
spProjTm(chi,in._odata[offset]);
|
|
} else {
|
|
chi=buf[offset];
|
|
}
|
|
Impl::multLink(Uchi,U._odata[sU],chi,Tp);
|
|
accumReconTm(result,Uchi);
|
|
|
|
vstream(out._odata[ss],result*(-0.5));
|
|
}
|
|
|
|
template<class Impl>
|
|
void WilsonKernels<Impl>::DiracOptDhopDir(CartesianStencil &st,DoubledGaugeField &U,
|
|
std::vector<SiteHalfSpinor,alignedAllocator<SiteHalfSpinor> > &buf,
|
|
int sF,int sU,const FermionField &in, FermionField &out,int dir,int gamma)
|
|
{
|
|
SiteHalfSpinor tmp;
|
|
SiteHalfSpinor chi;
|
|
SiteSpinor result;
|
|
SiteHalfSpinor Uchi;
|
|
int offset,local,perm, ptype;
|
|
int ss=sF;
|
|
|
|
offset = st._offsets [dir][ss];
|
|
local = st._is_local[dir][ss];
|
|
perm = st._permute[dir][ss];
|
|
ptype = st._permute_type[dir];
|
|
|
|
// Xp
|
|
if(gamma==Xp){
|
|
if ( local && perm ) {
|
|
spProjXp(tmp,in._odata[offset]);
|
|
permute(chi,tmp,ptype);
|
|
} else if ( local ) {
|
|
spProjXp(chi,in._odata[offset]);
|
|
} else {
|
|
chi=buf[offset];
|
|
}
|
|
Impl::multLink(Uchi,U._odata[sU],chi,dir);
|
|
spReconXp(result,Uchi);
|
|
}
|
|
|
|
// Yp
|
|
if ( gamma==Yp ){
|
|
if ( local && perm ) {
|
|
spProjYp(tmp,in._odata[offset]);
|
|
permute(chi,tmp,ptype);
|
|
} else if ( local ) {
|
|
spProjYp(chi,in._odata[offset]);
|
|
} else {
|
|
chi=buf[offset];
|
|
}
|
|
Impl::multLink(Uchi,U._odata[sU],chi,dir);
|
|
spReconYp(result,Uchi);
|
|
}
|
|
|
|
// Zp
|
|
if ( gamma ==Zp ){
|
|
if ( local && perm ) {
|
|
spProjZp(tmp,in._odata[offset]);
|
|
permute(chi,tmp,ptype);
|
|
} else if ( local ) {
|
|
spProjZp(chi,in._odata[offset]);
|
|
} else {
|
|
chi=buf[offset];
|
|
}
|
|
Impl::multLink(Uchi,U._odata[sU],chi,dir);
|
|
spReconZp(result,Uchi);
|
|
}
|
|
|
|
// Tp
|
|
if ( gamma ==Tp ){
|
|
if ( local && perm ) {
|
|
spProjTp(tmp,in._odata[offset]);
|
|
permute(chi,tmp,ptype);
|
|
} else if ( local ) {
|
|
spProjTp(chi,in._odata[offset]);
|
|
} else {
|
|
chi=buf[offset];
|
|
}
|
|
Impl::multLink(Uchi,U._odata[sU],chi,dir);
|
|
spReconTp(result,Uchi);
|
|
}
|
|
|
|
// Xm
|
|
if ( gamma==Xm ){
|
|
if ( local && perm ) {
|
|
spProjXm(tmp,in._odata[offset]);
|
|
permute(chi,tmp,ptype);
|
|
} else if ( local ) {
|
|
spProjXm(chi,in._odata[offset]);
|
|
} else {
|
|
chi=buf[offset];
|
|
}
|
|
Impl::multLink(Uchi,U._odata[sU],chi,dir);
|
|
spReconXm(result,Uchi);
|
|
}
|
|
|
|
// Ym
|
|
if ( gamma == Ym ){
|
|
if ( local && perm ) {
|
|
spProjYm(tmp,in._odata[offset]);
|
|
permute(chi,tmp,ptype);
|
|
} else if ( local ) {
|
|
spProjYm(chi,in._odata[offset]);
|
|
} else {
|
|
chi=buf[offset];
|
|
}
|
|
Impl::multLink(Uchi,U._odata[sU],chi,dir);
|
|
spReconYm(result,Uchi);
|
|
}
|
|
|
|
// Zm
|
|
if ( gamma == Zm ){
|
|
if ( local && perm ) {
|
|
spProjZm(tmp,in._odata[offset]);
|
|
permute(chi,tmp,ptype);
|
|
} else if ( local ) {
|
|
spProjZm(chi,in._odata[offset]);
|
|
} else {
|
|
chi=buf[offset];
|
|
}
|
|
Impl::multLink(Uchi,U._odata[sU],chi,dir);
|
|
spReconZm(result,Uchi);
|
|
}
|
|
|
|
// Tm
|
|
if ( gamma==Tm ) {
|
|
if ( local && perm ) {
|
|
spProjTm(tmp,in._odata[offset]);
|
|
permute(chi,tmp,ptype);
|
|
} else if ( local ) {
|
|
spProjTm(chi,in._odata[offset]);
|
|
} else {
|
|
chi=buf[offset];
|
|
}
|
|
Impl::multLink(Uchi,U._odata[sU],chi,dir);
|
|
spReconTm(result,Uchi);
|
|
}
|
|
|
|
vstream(out._odata[ss],result*(-0.5));
|
|
}
|
|
|
|
FermOpTemplateInstantiate(WilsonKernels);
|
|
|
|
}}
|