mirror of
https://github.com/paboyle/Grid.git
synced 2024-11-10 15:55:37 +00:00
20100d0a40
Useful to compare; raises Intel compiler from 9GFlop/s to 17.5 Gflops.
on ivybridge core. Raises Clang form 14.5 to 17.5
106 lines
3.8 KiB
C++
106 lines
3.8 KiB
C++
#ifndef GRID_QCD_WILSON_DOP_H
|
|
#define GRID_QCD_WILSON_DOP_H
|
|
|
|
|
|
namespace Grid {
|
|
|
|
namespace QCD {
|
|
|
|
// Should be in header?
|
|
const int Xp = 0;
|
|
const int Yp = 1;
|
|
const int Zp = 2;
|
|
const int Tp = 3;
|
|
const int Xm = 4;
|
|
const int Ym = 5;
|
|
const int Zm = 6;
|
|
const int Tm = 7;
|
|
|
|
class WilsonMatrix : public CheckerBoardedSparseMatrixBase<LatticeFermion>
|
|
{
|
|
//NB r=1;
|
|
public:
|
|
static int HandOptDslash;
|
|
|
|
double mass;
|
|
// GridBase * grid; // Inherited
|
|
// GridBase * cbgrid;
|
|
|
|
//Defines the stencils for even and odd
|
|
CartesianStencil Stencil;
|
|
CartesianStencil StencilEven;
|
|
CartesianStencil StencilOdd;
|
|
|
|
// Copy of the gauge field , with even and odd subsets
|
|
LatticeDoubledGaugeField Umu;
|
|
LatticeDoubledGaugeField UmuEven;
|
|
LatticeDoubledGaugeField UmuOdd;
|
|
|
|
static const int npoint=8;
|
|
static const std::vector<int> directions ;
|
|
static const std::vector<int> displacements;
|
|
static const int Xp,Xm,Yp,Ym,Zp,Zm,Tp,Tm;
|
|
|
|
// Comms buffer
|
|
std::vector<vHalfSpinColourVector,alignedAllocator<vHalfSpinColourVector> > comm_buf;
|
|
|
|
// Constructor
|
|
WilsonMatrix(LatticeGaugeField &_Umu,GridCartesian &Fgrid,GridRedBlackCartesian &Hgrid,double _mass);
|
|
|
|
// DoubleStore
|
|
void DoubleStore(LatticeDoubledGaugeField &Uds,const LatticeGaugeField &Umu);
|
|
|
|
// override multiply
|
|
virtual RealD M (const LatticeFermion &in, LatticeFermion &out);
|
|
virtual RealD Mdag (const LatticeFermion &in, LatticeFermion &out);
|
|
|
|
// half checkerboard operaions
|
|
virtual void Meooe (const LatticeFermion &in, LatticeFermion &out);
|
|
virtual void MeooeDag (const LatticeFermion &in, LatticeFermion &out);
|
|
virtual void Mooee (const LatticeFermion &in, LatticeFermion &out);
|
|
virtual void MooeeDag (const LatticeFermion &in, LatticeFermion &out);
|
|
virtual void MooeeInv (const LatticeFermion &in, LatticeFermion &out);
|
|
virtual void MooeeInvDag (const LatticeFermion &in, LatticeFermion &out);
|
|
|
|
// non-hermitian hopping term; half cb or both
|
|
void Dhop (const LatticeFermion &in, LatticeFermion &out,int dag);
|
|
void DhopOE(const LatticeFermion &in, LatticeFermion &out,int dag);
|
|
void DhopEO(const LatticeFermion &in, LatticeFermion &out,int dag);
|
|
void DhopInternal(CartesianStencil & st,LatticeDoubledGaugeField &U,
|
|
const LatticeFermion &in, LatticeFermion &out,int dag);
|
|
|
|
typedef iScalar<iMatrix<vComplex, Nc> > matrix;
|
|
|
|
|
|
};
|
|
|
|
|
|
class DiracOpt {
|
|
public:
|
|
// These ones will need to be package intelligently. WilsonType base class
|
|
// for use by DWF etc..
|
|
static void DhopSite(CartesianStencil &st,LatticeDoubledGaugeField &U,
|
|
std::vector<vHalfSpinColourVector,alignedAllocator<vHalfSpinColourVector> > &buf,
|
|
int ss,const LatticeFermion &in, LatticeFermion &out);
|
|
static void DhopSiteDag(CartesianStencil &st,LatticeDoubledGaugeField &U,
|
|
std::vector<vHalfSpinColourVector,alignedAllocator<vHalfSpinColourVector> > &buf,
|
|
int ss,const LatticeFermion &in, LatticeFermion &out);
|
|
|
|
};
|
|
class DiracOptHand {
|
|
public:
|
|
// These ones will need to be package intelligently. WilsonType base class
|
|
// for use by DWF etc..
|
|
static void DhopSite(CartesianStencil &st,LatticeDoubledGaugeField &U,
|
|
std::vector<vHalfSpinColourVector,alignedAllocator<vHalfSpinColourVector> > &buf,
|
|
int ss,const LatticeFermion &in, LatticeFermion &out);
|
|
static void DhopSiteDag(CartesianStencil &st,LatticeDoubledGaugeField &U,
|
|
std::vector<vHalfSpinColourVector,alignedAllocator<vHalfSpinColourVector> > &buf,
|
|
int ss,const LatticeFermion &in, LatticeFermion &out);
|
|
|
|
};
|
|
|
|
}
|
|
}
|
|
#endif
|