mirror of
https://github.com/paboyle/Grid.git
synced 2024-11-10 15:55:37 +00:00
3845f267cb
Tanh/Zolo * (Cayley/PartFrac/ContFrac) * (Mobius/Shamir/Wilson) Approx Representation Kernel. All are done with space-time taking part in checkerboarding, Ls uncheckerboarded Have only so far tested the Domain Wall limit of mobius, and at that only checked that it i) Inverts ii) 5dim DW == Ls copies of 4dim D2 iii) MeeInv Mee == 1 iv) Meo+Mee+Moe+Moo == M unprec. v) MpcDagMpc is hermitan vi) Mdag is the adjoint of M between stochastic vectors. That said, the RB schur solve, RB MpcDagMpc solve, Unprec solve all converge and the true residual becomes small; so pretty good tests.
114 lines
3.3 KiB
C++
114 lines
3.3 KiB
C++
#ifndef GRID_QCD_LINALG_UTILS_H
|
|
#define GRID_QCD_LINALG_UTILS_H
|
|
|
|
namespace Grid{
|
|
namespace QCD{
|
|
////////////////////////////////////////////////////////////////////////
|
|
//This file brings additional linear combination assist that is helpful
|
|
//to QCD such as chiral projectors and spin matrices applied to one of the inputs.
|
|
//These routines support five-D chiral fermions and contain s-subslice indexing
|
|
//on the 5d (rb4d) checkerboarded lattices
|
|
////////////////////////////////////////////////////////////////////////
|
|
template<class vobj>
|
|
void axpby_ssp(Lattice<vobj> &z, RealD a,const Lattice<vobj> &x,RealD b,const Lattice<vobj> &y,int s,int sp)
|
|
{
|
|
z.checkerboard = x.checkerboard;
|
|
conformable(x,y);
|
|
conformable(x,z);
|
|
GridBase *grid=x._grid;
|
|
int Ls = grid->_rdimensions[0];
|
|
PARALLEL_FOR_LOOP
|
|
for(int ss=0;ss<grid->oSites();ss+=Ls){ // adds Ls
|
|
vobj tmp = a*x._odata[ss+s]+b*y._odata[ss+sp];
|
|
vstream(z._odata[ss+s],tmp);
|
|
}
|
|
}
|
|
|
|
template<class vobj>
|
|
void ag5xpby_ssp(Lattice<vobj> &z,RealD a,const Lattice<vobj> &x,RealD b,const Lattice<vobj> &y,int s,int sp)
|
|
{
|
|
z.checkerboard = x.checkerboard;
|
|
conformable(x,y);
|
|
conformable(x,z);
|
|
GridBase *grid=x._grid;
|
|
int Ls = grid->_rdimensions[0];
|
|
PARALLEL_FOR_LOOP
|
|
for(int ss=0;ss<grid->oSites();ss+=Ls){ // adds Ls
|
|
vobj tmp;
|
|
multGamma5(tmp(),a*x._odata[ss+s]());
|
|
tmp = tmp + b*y._odata[ss+sp];
|
|
vstream(z._odata[ss+s],tmp);
|
|
}
|
|
}
|
|
|
|
template<class vobj>
|
|
void axpbg5y_ssp(Lattice<vobj> &z,RealD a,const Lattice<vobj> &x,RealD b,const Lattice<vobj> &y,int s,int sp)
|
|
{
|
|
z.checkerboard = x.checkerboard;
|
|
conformable(x,y);
|
|
conformable(x,z);
|
|
GridBase *grid=x._grid;
|
|
int Ls = grid->_rdimensions[0];
|
|
PARALLEL_FOR_LOOP
|
|
for(int ss=0;ss<grid->oSites();ss+=Ls){ // adds Ls
|
|
vobj tmp;
|
|
multGamma5(tmp(),b*y._odata[ss+sp]());
|
|
tmp = tmp + a*x._odata[ss+s];
|
|
vstream(z._odata[ss+s],tmp);
|
|
}
|
|
}
|
|
|
|
template<class vobj>
|
|
void ag5xpbg5y_ssp(Lattice<vobj> &z,RealD a,const Lattice<vobj> &x,RealD b,const Lattice<vobj> &y,int s,int sp)
|
|
{
|
|
z.checkerboard = x.checkerboard;
|
|
conformable(x,y);
|
|
conformable(x,z);
|
|
GridBase *grid=x._grid;
|
|
int Ls = grid->_rdimensions[0];
|
|
PARALLEL_FOR_LOOP
|
|
for(int ss=0;ss<grid->oSites();ss+=Ls){ // adds Ls
|
|
vobj tmp1;
|
|
vobj tmp2;
|
|
tmp1 = a*x._odata[ss+s]+b*y._odata[ss+sp];
|
|
multGamma5(tmp2(),tmp1());
|
|
vstream(z._odata[ss+s],tmp2);
|
|
}
|
|
}
|
|
|
|
template<class vobj>
|
|
void axpby_ssp_pminus(Lattice<vobj> &z,RealD a,const Lattice<vobj> &x,RealD b,const Lattice<vobj> &y,int s,int sp)
|
|
{
|
|
z.checkerboard = x.checkerboard;
|
|
conformable(x,y);
|
|
conformable(x,z);
|
|
GridBase *grid=x._grid;
|
|
int Ls = grid->_rdimensions[0];
|
|
PARALLEL_FOR_LOOP
|
|
for(int ss=0;ss<grid->oSites();ss+=Ls){ // adds Ls
|
|
vobj tmp;
|
|
spProj5m(tmp,y._odata[ss+sp]);
|
|
tmp = a*x._odata[ss+s]+b*tmp;
|
|
vstream(z._odata[ss+s],tmp);
|
|
}
|
|
}
|
|
|
|
template<class vobj>
|
|
void axpby_ssp_pplus(Lattice<vobj> &z,RealD a,const Lattice<vobj> &x,RealD b,const Lattice<vobj> &y,int s,int sp)
|
|
{
|
|
z.checkerboard = x.checkerboard;
|
|
conformable(x,y);
|
|
conformable(x,z);
|
|
GridBase *grid=x._grid;
|
|
int Ls = grid->_rdimensions[0];
|
|
PARALLEL_FOR_LOOP
|
|
for(int ss=0;ss<grid->oSites();ss+=Ls){ // adds Ls
|
|
vobj tmp;
|
|
spProj5p(tmp,y._odata[ss+sp]);
|
|
tmp = a*x._odata[ss+s]+b*tmp;
|
|
vstream(z._odata[ss+s],tmp);
|
|
}
|
|
}
|
|
}}
|
|
#endif
|