mirror of
https://github.com/paboyle/Grid.git
synced 2024-11-10 15:55:37 +00:00
311 lines
9.6 KiB
C++
311 lines
9.6 KiB
C++
/*************************************************************************************
|
|
|
|
Grid physics library, www.github.com/paboyle/Grid
|
|
|
|
Source file: ./lib/qcd/action/fermion/WilsonFermion.cc
|
|
|
|
Copyright (C) 2015
|
|
|
|
Author: Peter Boyle <pabobyle@ph.ed.ac.uk>
|
|
Author: Peter Boyle <paboyle@ph.ed.ac.uk>
|
|
Author: Peter Boyle <peterboyle@Peters-MacBook-Pro-2.local>
|
|
Author: paboyle <paboyle@ph.ed.ac.uk>
|
|
|
|
This program is free software; you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation; either version 2 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License along
|
|
with this program; if not, write to the Free Software Foundation, Inc.,
|
|
51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
|
|
See the full license in the file "LICENSE" in the top level distribution
|
|
directory
|
|
*************************************************************************************/
|
|
/* END LEGAL */
|
|
#include <Grid.h>
|
|
|
|
namespace Grid {
|
|
namespace QCD {
|
|
|
|
const std::vector<int> WilsonFermionStatic::directions({0, 1, 2, 3, 0, 1, 2, 3});
|
|
const std::vector<int> WilsonFermionStatic::displacements({1, 1, 1, 1, -1, -1, -1, -1});
|
|
|
|
int WilsonFermionStatic::HandOptDslash;
|
|
|
|
/////////////////////////////////
|
|
// Constructor and gauge import
|
|
/////////////////////////////////
|
|
|
|
template <class Impl>
|
|
WilsonFermion<Impl>::WilsonFermion(GaugeField &_Umu, GridCartesian &Fgrid,
|
|
GridRedBlackCartesian &Hgrid, RealD _mass,
|
|
const ImplParams &p)
|
|
: Kernels(p),
|
|
_grid(&Fgrid),
|
|
_cbgrid(&Hgrid),
|
|
Stencil(&Fgrid, npoint, Even, directions, displacements),
|
|
StencilEven(&Hgrid, npoint, Even, directions,
|
|
displacements), // source is Even
|
|
StencilOdd(&Hgrid, npoint, Odd, directions,
|
|
displacements), // source is Odd
|
|
mass(_mass),
|
|
Lebesgue(_grid),
|
|
LebesgueEvenOdd(_cbgrid),
|
|
Umu(&Fgrid),
|
|
UmuEven(&Hgrid),
|
|
UmuOdd(&Hgrid) {
|
|
// Allocate the required comms buffer
|
|
ImportGauge(_Umu);
|
|
}
|
|
|
|
template <class Impl>
|
|
void WilsonFermion<Impl>::ImportGauge(const GaugeField &_Umu) {
|
|
GaugeField HUmu(_Umu._grid);
|
|
HUmu = _Umu * (-0.5);
|
|
Impl::DoubleStore(GaugeGrid(), Umu, HUmu);
|
|
pickCheckerboard(Even, UmuEven, Umu);
|
|
pickCheckerboard(Odd, UmuOdd, Umu);
|
|
}
|
|
|
|
/////////////////////////////
|
|
// Implement the interface
|
|
/////////////////////////////
|
|
|
|
template <class Impl>
|
|
RealD WilsonFermion<Impl>::M(const FermionField &in, FermionField &out) {
|
|
out.checkerboard = in.checkerboard;
|
|
Dhop(in, out, DaggerNo);
|
|
return axpy_norm(out, 4 + mass, in, out);
|
|
}
|
|
|
|
template <class Impl>
|
|
RealD WilsonFermion<Impl>::Mdag(const FermionField &in, FermionField &out) {
|
|
out.checkerboard = in.checkerboard;
|
|
Dhop(in, out, DaggerYes);
|
|
return axpy_norm(out, 4 + mass, in, out);
|
|
}
|
|
|
|
template <class Impl>
|
|
void WilsonFermion<Impl>::Meooe(const FermionField &in, FermionField &out) {
|
|
if (in.checkerboard == Odd) {
|
|
DhopEO(in, out, DaggerNo);
|
|
} else {
|
|
DhopOE(in, out, DaggerNo);
|
|
}
|
|
}
|
|
template <class Impl>
|
|
void WilsonFermion<Impl>::MeooeDag(const FermionField &in, FermionField &out) {
|
|
if (in.checkerboard == Odd) {
|
|
DhopEO(in, out, DaggerYes);
|
|
} else {
|
|
DhopOE(in, out, DaggerYes);
|
|
}
|
|
}
|
|
|
|
template <class Impl>
|
|
void WilsonFermion<Impl>::Mooee(const FermionField &in, FermionField &out) {
|
|
out.checkerboard = in.checkerboard;
|
|
typename FermionField::scalar_type scal(4.0 + mass);
|
|
out = scal * in;
|
|
}
|
|
|
|
template <class Impl>
|
|
void WilsonFermion<Impl>::MooeeDag(const FermionField &in, FermionField &out) {
|
|
out.checkerboard = in.checkerboard;
|
|
Mooee(in, out);
|
|
}
|
|
|
|
template <class Impl>
|
|
void WilsonFermion<Impl>::MooeeInv(const FermionField &in, FermionField &out) {
|
|
out.checkerboard = in.checkerboard;
|
|
out = (1.0 / (4.0 + mass)) * in;
|
|
}
|
|
|
|
template <class Impl>
|
|
void WilsonFermion<Impl>::MooeeInvDag(const FermionField &in,
|
|
FermionField &out) {
|
|
out.checkerboard = in.checkerboard;
|
|
MooeeInv(in, out);
|
|
}
|
|
|
|
///////////////////////////////////
|
|
// Internal
|
|
///////////////////////////////////
|
|
|
|
template <class Impl>
|
|
void WilsonFermion<Impl>::DerivInternal(StencilImpl &st, DoubledGaugeField &U,
|
|
GaugeField &mat, const FermionField &A,
|
|
const FermionField &B, int dag) {
|
|
assert((dag == DaggerNo) || (dag == DaggerYes));
|
|
|
|
Compressor compressor(dag);
|
|
|
|
FermionField Btilde(B._grid);
|
|
FermionField Atilde(B._grid);
|
|
Atilde = A;
|
|
|
|
st.HaloExchange(B, compressor);
|
|
|
|
for (int mu = 0; mu < Nd; mu++) {
|
|
////////////////////////////////////////////////////////////////////////
|
|
// Flip gamma (1+g)<->(1-g) if dag
|
|
////////////////////////////////////////////////////////////////////////
|
|
int gamma = mu;
|
|
if (!dag) gamma += Nd;
|
|
|
|
////////////////////////
|
|
// Call the single hop
|
|
////////////////////////
|
|
PARALLEL_FOR_LOOP
|
|
for (int sss = 0; sss < B._grid->oSites(); sss++) {
|
|
Kernels::DiracOptDhopDir(st, U, st.CommBuf(), sss, sss, B, Btilde, mu, gamma);
|
|
}
|
|
|
|
//////////////////////////////////////////////////
|
|
// spin trace outer product
|
|
//////////////////////////////////////////////////
|
|
Impl::InsertForce4D(mat, Btilde, Atilde, mu);
|
|
}
|
|
}
|
|
|
|
template <class Impl>
|
|
void WilsonFermion<Impl>::DhopDeriv(GaugeField &mat, const FermionField &U,
|
|
const FermionField &V, int dag) {
|
|
conformable(U._grid, _grid);
|
|
conformable(U._grid, V._grid);
|
|
conformable(U._grid, mat._grid);
|
|
|
|
mat.checkerboard = U.checkerboard;
|
|
|
|
DerivInternal(Stencil, Umu, mat, U, V, dag);
|
|
}
|
|
|
|
template <class Impl>
|
|
void WilsonFermion<Impl>::DhopDerivOE(GaugeField &mat, const FermionField &U,
|
|
const FermionField &V, int dag) {
|
|
conformable(U._grid, _cbgrid);
|
|
conformable(U._grid, V._grid);
|
|
conformable(U._grid, mat._grid);
|
|
|
|
assert(V.checkerboard == Even);
|
|
assert(U.checkerboard == Odd);
|
|
mat.checkerboard = Odd;
|
|
|
|
DerivInternal(StencilEven, UmuOdd, mat, U, V, dag);
|
|
}
|
|
|
|
template <class Impl>
|
|
void WilsonFermion<Impl>::DhopDerivEO(GaugeField &mat, const FermionField &U,
|
|
const FermionField &V, int dag) {
|
|
conformable(U._grid, _cbgrid);
|
|
conformable(U._grid, V._grid);
|
|
conformable(U._grid, mat._grid);
|
|
|
|
assert(V.checkerboard == Odd);
|
|
assert(U.checkerboard == Even);
|
|
mat.checkerboard = Even;
|
|
|
|
DerivInternal(StencilOdd, UmuEven, mat, U, V, dag);
|
|
}
|
|
|
|
template <class Impl>
|
|
void WilsonFermion<Impl>::Dhop(const FermionField &in, FermionField &out,
|
|
int dag) {
|
|
conformable(in._grid, _grid); // verifies full grid
|
|
conformable(in._grid, out._grid);
|
|
|
|
out.checkerboard = in.checkerboard;
|
|
|
|
DhopInternal(Stencil, Lebesgue, Umu, in, out, dag);
|
|
}
|
|
|
|
template <class Impl>
|
|
void WilsonFermion<Impl>::DhopOE(const FermionField &in, FermionField &out,
|
|
int dag) {
|
|
conformable(in._grid, _cbgrid); // verifies half grid
|
|
conformable(in._grid, out._grid); // drops the cb check
|
|
|
|
assert(in.checkerboard == Even);
|
|
out.checkerboard = Odd;
|
|
|
|
DhopInternal(StencilEven, LebesgueEvenOdd, UmuOdd, in, out, dag);
|
|
}
|
|
|
|
template <class Impl>
|
|
void WilsonFermion<Impl>::DhopEO(const FermionField &in, FermionField &out,
|
|
int dag) {
|
|
conformable(in._grid, _cbgrid); // verifies half grid
|
|
conformable(in._grid, out._grid); // drops the cb check
|
|
|
|
assert(in.checkerboard == Odd);
|
|
out.checkerboard = Even;
|
|
|
|
DhopInternal(StencilOdd, LebesgueEvenOdd, UmuEven, in, out, dag);
|
|
}
|
|
|
|
template <class Impl>
|
|
void WilsonFermion<Impl>::Mdir(const FermionField &in, FermionField &out,
|
|
int dir, int disp) {
|
|
DhopDir(in, out, dir, disp);
|
|
}
|
|
|
|
template <class Impl>
|
|
void WilsonFermion<Impl>::DhopDir(const FermionField &in, FermionField &out,
|
|
int dir, int disp) {
|
|
int skip = (disp == 1) ? 0 : 1;
|
|
int dirdisp = dir + skip * 4;
|
|
int gamma = dir + (1 - skip) * 4;
|
|
|
|
DhopDirDisp(in, out, dirdisp, gamma, DaggerNo);
|
|
};
|
|
|
|
template <class Impl>
|
|
void WilsonFermion<Impl>::DhopDirDisp(const FermionField &in, FermionField &out,
|
|
int dirdisp, int gamma, int dag) {
|
|
Compressor compressor(dag);
|
|
|
|
Stencil.HaloExchange(in, compressor);
|
|
|
|
PARALLEL_FOR_LOOP
|
|
for (int sss = 0; sss < in._grid->oSites(); sss++) {
|
|
Kernels::DiracOptDhopDir(Stencil, Umu, Stencil.CommBuf(), sss, sss, in, out, dirdisp, gamma);
|
|
}
|
|
};
|
|
|
|
template <class Impl>
|
|
void WilsonFermion<Impl>::DhopInternal(StencilImpl &st, LebesgueOrder &lo,
|
|
DoubledGaugeField &U,
|
|
const FermionField &in,
|
|
FermionField &out, int dag) {
|
|
assert((dag == DaggerNo) || (dag == DaggerYes));
|
|
|
|
Compressor compressor(dag);
|
|
st.HaloExchange(in, compressor);
|
|
|
|
if (dag == DaggerYes) {
|
|
PARALLEL_FOR_LOOP
|
|
for (int sss = 0; sss < in._grid->oSites(); sss++) {
|
|
Kernels::DiracOptDhopSiteDag(st, lo, U, st.CommBuf(), sss, sss, 1, 1, in, out);
|
|
}
|
|
} else {
|
|
PARALLEL_FOR_LOOP
|
|
for (int sss = 0; sss < in._grid->oSites(); sss++) {
|
|
Kernels::DiracOptDhopSite(st, lo, U, st.CommBuf(), sss, sss, 1, 1, in, out);
|
|
}
|
|
}
|
|
};
|
|
|
|
FermOpTemplateInstantiate(WilsonFermion);
|
|
AdjointFermOpTemplateInstantiate(WilsonFermion);
|
|
TwoIndexFermOpTemplateInstantiate(WilsonFermion);
|
|
GparityFermOpTemplateInstantiate(WilsonFermion);
|
|
}
|
|
}
|