1
0
mirror of https://github.com/paboyle/Grid.git synced 2024-11-10 07:55:35 +00:00

Clover term compiles, not tested

This commit is contained in:
Guido Cossu 2017-04-28 15:23:34 +01:00
parent 752048f410
commit 5553b8d2b8
4 changed files with 383 additions and 73 deletions

View File

@ -49,7 +49,7 @@ Author: Peter Boyle <pabobyle@ph.ed.ac.uk>
#include <Grid/qcd/action/fermion/WilsonFermion.h> // 4d wilson like
#include <Grid/qcd/action/fermion/WilsonTMFermion.h> // 4d wilson like
#include <Grid/qcd/action/fermion/WilsonCloverFermion.h> // 4d wilson like
#include <Grid/qcd/action/fermion/WilsonCloverFermion.h> // 4d wilson clover fermions
#include <Grid/qcd/action/fermion/WilsonFermion5D.h> // 5d base used by all 5d overlap types
#include <Grid/qcd/action/fermion/ImprovedStaggeredFermion.h>

View File

@ -27,28 +27,35 @@
*************************************************************************************/
/* END LEGAL */
#include <Grid/Grid.h>
#include <Grid/Eigen/Dense>
#include <Grid/qcd/spin/Dirac.h>
namespace Grid {
namespace QCD {
template <class Impl>
void WilsonCloverFermion<Impl>::AddCloverTerm(const FermionField& in,
FermionField& out){
FermionField tmp(out._grid);
tmp = zero;
// the product sigma_munu Fmunu is hermitian
tmp += Bx*(Gamma(Gamma::Algebra::SigmaYZ)*in);
tmp += By*(Gamma(Gamma::Algebra::MinusSigmaXZ)*in);
tmp += Bz*(Gamma(Gamma::Algebra::SigmaXY)*in);
tmp += Ex*(Gamma(Gamma::Algebra::MinusSigmaXT)*in);
tmp += Ey*(Gamma(Gamma::Algebra::MinusSigmaYT)*in);
tmp += Ez*(Gamma(Gamma::Algebra::MinusSigmaZT)*in);
out += tmp*csw; // check signs
}
//WilsonLoop::CloverPlaquette
/////////////////////////////////////////////////////
//// Clover plaquette combination in mu,nu plane with Double Stored U
////////////////////////////////////////////////////
//static void CloverPlaquette(GaugeMat &Q, const std::vector<GaugeMat> &U,
// const int mu, const int nu){
// Q = zero;
// Q += Gimpl::CovShiftBackward(
// U[mu], mu, Gimpl::CovShiftBackward(
// U[nu], nu, Gimpl::CovShiftForward(U[mu], mu, U[nu] )));
// Q += Gimpl::CovShiftForward(
// U[mu], mu, Gimpl::CovShiftForward(
// U[nu], nu, Gimpl::CovShiftBackward(U[mu], mu, U[nu+Nd] )));
// Q += Gimpl::CovShiftBackward(
// U[nu], nu, Gimpl::CovShiftForward(
// U[mu], mu, Gimpl::CovShiftForward(U[nu], nu, U[mu+Nd] )));
// Q += Gimpl::CovShiftForward(
// U[mu], mu, Gimpl::CovShiftBackward(
// U[nu], nu, Gimpl::CovShiftBackward(U[mu], mu, U[nu] )));
// }
// *NOT* EO
template <class Impl>
RealD WilsonCloverFermion<Impl>::M(const FermionField& in, FermionField& out) {
// Wilson term
@ -56,7 +63,7 @@ namespace QCD {
this->Dhop(in, out, DaggerNo);
// Clover term
// apply the sigma and Fmunu
AddCloverTerm(in, out);
Mooee(in, out);
// overall factor
return axpy_norm(out, 4 + this->mass, in, out);
}
@ -68,13 +75,16 @@ namespace QCD {
this->Dhop(in, out, DaggerYes);
// Clover term
// apply the sigma and Fmunu
AddCloverTerm(in, out);
MooeeDag(in, out);
return axpy_norm(out, 4 + this->mass, in, out);
}
template <class Impl>
void WilsonCloverFermion<Impl>::ImportGauge(const GaugeField& _Umu) {
this->ImportGauge(_Umu);
GridBase* grid = _Umu._grid;
assert(Nd==4); //only works in 4 dim
typename Impl::GaugeLinkField Bx(grid), By(grid), Bz(grid), Ex(grid), Ey(grid), Ez(grid);
// Compute the field strength terms
WilsonLoops<Impl>::FieldStrength(Bx, _Umu, Ydir, Zdir);
WilsonLoops<Impl>::FieldStrength(By, _Umu, Zdir, Xdir);
@ -82,31 +92,77 @@ namespace QCD {
WilsonLoops<Impl>::FieldStrength(Ex, _Umu, Tdir, Xdir);
WilsonLoops<Impl>::FieldStrength(Ey, _Umu, Tdir, Ydir);
WilsonLoops<Impl>::FieldStrength(Ez, _Umu, Tdir, Zdir);
// Save the contracted term with sigma
// into a dense matrix site by site
// Invert the Moo, Mee terms (using Eigen)
// Compute the Clover Operator acting on Colour and Spin
CloverTerm = fillClover(Bx)*(Gamma(Gamma::Algebra::SigmaYZ));
CloverTerm += fillClover(By)*(Gamma(Gamma::Algebra::MinusSigmaXZ));
CloverTerm += fillClover(Bz)*(Gamma(Gamma::Algebra::SigmaXY));
CloverTerm += fillClover(Ex)*(Gamma(Gamma::Algebra::MinusSigmaXT));
CloverTerm += fillClover(Ey)*(Gamma(Gamma::Algebra::MinusSigmaYT));
CloverTerm += fillClover(Ez)*(Gamma(Gamma::Algebra::MinusSigmaZT));
CloverTerm *= csw;
int lvol = _Umu._grid->lSites();
int DimRep = Impl::Dimension;
Eigen::MatrixXcd EigenCloverOp = Eigen::MatrixXcd::Zero(Ns*DimRep,Ns*DimRep);
Eigen::MatrixXcd EigenInvCloverOp = Eigen::MatrixXcd::Zero(Ns*DimRep,Ns*DimRep);
std::vector <int> lcoor;
typename SiteCloverType::scalar_object Qx = zero, Qxinv = zero;
for (int site = 0; site < lvol; site++){
grid->LocalIndexToLocalCoor(site,lcoor);
EigenCloverOp=Eigen::MatrixXcd::Zero(Ns*DimRep,Ns*DimRep);
peekLocalSite(Qx,CloverTerm,lcoor);
Qxinv = zero;
for(int j = 0; j < Ns; j++)
for (int k = 0; k < Ns; k++)
for(int a = 0; a < DimRep; a++)
for(int b = 0; b < DimRep; b++)
EigenCloverOp(a+j*DimRep,b+k*DimRep) = Qx()(j,k)(a,b);
EigenInvCloverOp = EigenCloverOp.inverse();
for(int j = 0; j < Ns; j++)
for (int k = 0; k < Ns; k++)
for(int a = 0; a < DimRep; a++)
for(int b = 0; b < DimRep; b++)
Qxinv()(j,k)(a,b) = EigenInvCloverOp(a+j*DimRep,b+k*DimRep);
pokeLocalSite(Qxinv,CloverTermInv,lcoor);
}
}
template<class Impl>
void WilsonCloverFermion<Impl>::Mooee(const FermionField &in, FermionField &out){
this -> MooeeInternal(in, out, DaggerNo, InverseNo);
}
template<class Impl>
void WilsonCloverFermion<Impl>::MooeeDag(const FermionField &in, FermionField &out){
this -> MooeeInternal(in, out, DaggerNo, InverseYes);
}
template<class Impl>
void WilsonCloverFermion<Impl>::MooeeInv(const FermionField &in, FermionField &out){
this -> MooeeInternal(in, out, DaggerNo, InverseYes);
}
template<class Impl>
void WilsonCloverFermion<Impl>::MooeeInvDag(const FermionField &in, FermionField &out){
this -> MooeeInternal(in, out, DaggerNo, InverseYes);
}
template<class Impl>
void WilsonCloverFermion<Impl>::Mooee(const FermionField &in, FermionField &out) {
void WilsonCloverFermion<Impl>::MooeeInternal(const FermionField &in, FermionField &out, int dag, int inv){
out.checkerboard = in.checkerboard;
assert(0); // to be completed
}
CloverFieldType *Clover;
template<class Impl>
void WilsonCloverFermion<Impl>::MooeeDag(const FermionField &in, FermionField &out) {
assert(0); // not implemented yet
}
template<class Impl>
void WilsonCloverFermion<Impl>::MooeeInv(const FermionField &in, FermionField &out) {
assert(0); // not implemented yet
}
template<class Impl>
void WilsonCloverFermion<Impl>::MooeeInvDag(const FermionField &in, FermionField &out) {
assert(0); // not implemented yet
}
Clover = (inv) ? &CloverTermInv : &CloverTerm;
if(dag){ out = adj(*Clover)*in;} else {out = *Clover*in;}
} // MooeeInternal
// Derivative parts
template<class Impl>
@ -128,17 +184,6 @@ namespace QCD {
template<class Impl>
void WilsonCloverFermion<Impl>::MooDeriv(GaugeField&mat, const FermionField&U, const FermionField&V, int dag){
// Compute the 8 terms of the derivative
// Pseudocode
// Using Chroma as a template
// for loop on mu and nu, but upper matrix
// take the outer product factor * U x (sigma_mu_nu V)
// derivative of loops
// end of loop
assert(0); // not implemented yet
}
@ -148,7 +193,10 @@ namespace QCD {
assert(0); // not implemented yet
}
FermOpTemplateInstantiate(WilsonCloverFermion);
FermOpTemplateInstantiate(WilsonCloverFermion); // now only for the fundamental representation
//AdjointFermOpTemplateInstantiate(WilsonCloverFermion);
//TwoIndexFermOpTemplateInstantiate(WilsonCloverFermion);
//GparityFermOpTemplateInstantiate(WilsonCloverFermion);
}
}

View File

@ -31,14 +31,20 @@
#include <Grid/Grid.h>
namespace Grid {
namespace QCD {
namespace Grid
{
namespace QCD
{
template <class Impl>
class WilsonCloverFermion : public WilsonFermion<Impl> {
class WilsonCloverFermion : public WilsonFermion<Impl>
{
public:
// Types definitions
INHERIT_IMPL_TYPES(Impl);
template <typename vtype> using iImplClover = iScalar<iMatrix<iMatrix<vtype, Impl::Dimension>, Ns> >;
typedef iImplClover<Simd> SiteCloverType;
typedef Lattice<SiteCloverType> CloverFieldType;
public:
typedef WilsonFermion<Impl> WilsonBase;
@ -52,42 +58,47 @@ public:
Fgrid,
Hgrid,
_mass, p),
Bx(_Umu._grid),
By(_Umu._grid),
Bz(_Umu._grid),
Ex(_Umu._grid),
Ey(_Umu._grid),
Ez(_Umu._grid)
CloverTerm(&Fgrid),
CloverTermInv(&Fgrid)
{
csw = _csw;
assert(Nd == 4); // require 4 dimensions
}
virtual RealD M(const FermionField& in, FermionField& out);
virtual RealD Mdag(const FermionField& in, FermionField& out);
virtual RealD M(const FermionField &in, FermionField &out);
virtual RealD Mdag(const FermionField &in, FermionField &out);
virtual void Mooee(const FermionField &in, FermionField &out);
virtual void MooeeDag(const FermionField &in, FermionField &out);
virtual void MooeeInv(const FermionField &in, FermionField &out);
virtual void MooeeInvDag(const FermionField &in, FermionField &out);
virtual void MooeeInternal(const FermionField &in, FermionField &out, int dag, int inv);
virtual void MDeriv(GaugeField&mat, const FermionField&U, const FermionField&V, int dag);
virtual void MooDeriv(GaugeField&mat, const FermionField&U, const FermionField&V, int dag);
virtual void MeeDeriv(GaugeField&mat, const FermionField&U, const FermionField&V, int dag);
virtual void MDeriv(GaugeField &mat, const FermionField &U, const FermionField &V, int dag);
virtual void MooDeriv(GaugeField &mat, const FermionField &U, const FermionField &V, int dag);
virtual void MeeDeriv(GaugeField &mat, const FermionField &U, const FermionField &V, int dag);
void ImportGauge(const GaugeField &_Umu);
private:
// here fixing the 4 dimensions, make it more general?
// Field strengths
GaugeLinkField Bx, By, Bz, Ex, Ey, Ez;
RealD csw; // Clover coefficient
CloverFieldType CloverTerm, CloverTermInv; // Clover term
// eventually these two can be compressed into 6x6 blocks instead of the 12x12
// using the DeGrand-Rossi basis for the gamma matrices
CloverFieldType fillClover(const GaugeLinkField& F){
CloverFieldType T(F._grid);
PARALLEL_FOR_LOOP
for (int i = 0; i < CloverTerm._grid->oSites(); i++){
for (int s1 = 0; s1 < Nc; s1++)
for (int s2 = 0; s2 < Nc; s2++)
T._odata[i]()(s1,s2) = F._odata[i]()();
}
return T;
}
// Methods
void AddCloverTerm(const FermionField& in, FermionField& out);
};
}
}

View File

@ -0,0 +1,251 @@
/*************************************************************************************
Grid physics library, www.github.com/paboyle/Grid
Source file: ./benchmarks/Benchmark_wilson.cc
Copyright (C) 2015
Author: Guido Cossu <guido.cossu@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/Grid.h>
using namespace std;
using namespace Grid;
using namespace Grid::QCD;
int main (int argc, char ** argv)
{
Grid_init(&argc,&argv);
std::vector<int> latt_size = GridDefaultLatt();
std::vector<int> simd_layout = GridDefaultSimd(Nd,vComplex::Nsimd());
std::vector<int> mpi_layout = GridDefaultMpi();
GridCartesian Grid(latt_size,simd_layout,mpi_layout);
GridRedBlackCartesian RBGrid(latt_size,simd_layout,mpi_layout);
int threads = GridThread::GetThreads();
std::cout<<GridLogMessage << "Grid is setup to use "<<threads<<" threads"<<std::endl;
std::cout<<GridLogMessage << "Grid floating point word size is REALF"<< sizeof(RealF)<<std::endl;
std::cout<<GridLogMessage << "Grid floating point word size is REALD"<< sizeof(RealD)<<std::endl;
std::cout<<GridLogMessage << "Grid floating point word size is REAL"<< sizeof(Real)<<std::endl;
std::vector<int> seeds({1,2,3,4});
GridParallelRNG pRNG(&Grid);
pRNG.SeedFixedIntegers(seeds);
// pRNG.SeedFixedIntegers(std::vector<int>({45,12,81,9});
typedef typename WilsonCloverFermionR::FermionField FermionField;
typename WilsonCloverFermionR::ImplParams params;
FermionField src (&Grid); random(pRNG,src);
FermionField result(&Grid); result=zero;
FermionField ref(&Grid); ref=zero;
FermionField tmp(&Grid); tmp=zero;
FermionField err(&Grid); tmp=zero;
FermionField phi (&Grid); random(pRNG,phi);
FermionField chi (&Grid); random(pRNG,chi);
LatticeGaugeField Umu(&Grid); SU3::HotConfiguration(pRNG,Umu);
std::vector<LatticeColourMatrix> U(4,&Grid);
double volume=1;
for(int mu=0;mu<Nd;mu++){
volume=volume*latt_size[mu];
}
// Only one non-zero (y)
for(int mu=0;mu<Nd;mu++){
U[mu] = PeekIndex<LorentzIndex>(Umu,mu);
/* Debug force unit
U[mu] = 1.0;
PokeIndex<LorentzIndex>(Umu,U[mu],mu);
*/
}
ref = zero;
RealD mass=0.1;
RealD csw = 1.0;
{ // Simple clover implementation
// ref = ref + mass * src;
}
WilsonCloverFermionR Dwc(Umu,Grid,RBGrid,mass,csw,params);
std::cout<<GridLogMessage<<"=========================================================="<<std::endl;
std::cout<<GridLogMessage<<"= Testing Dhop against cshift implementation "<<std::endl;
std::cout<<GridLogMessage<<"=========================================================="<<std::endl;
std::cout<<GridLogMessage << "Calling Dwc"<<std::endl;
int ncall=1000;
double t0=usecond();
for(int i=0;i<ncall;i++){
Dwc.Dhop(src,result,0);
}
double t1=usecond();
double t2;
double flops=(16*(3*(6+8+8)) + 15*3*2)*volume*ncall; // == 66*16 + == 1146
std::cout<<GridLogMessage << "Called Dwc"<<std::endl;
std::cout<<GridLogMessage << "norm result "<< norm2(result)<<std::endl;
std::cout<<GridLogMessage << "norm ref "<< norm2(ref)<<std::endl;
std::cout<<GridLogMessage << "mflop/s = "<< flops/(t1-t0)<<std::endl;
err = ref-result;
std::cout<<GridLogMessage << "norm diff "<< norm2(err)<<std::endl;
std::cout<<GridLogMessage<<"=========================================================="<<std::endl;
std::cout<<GridLogMessage<<"= Testing that Deo + Doe = Dunprec "<<std::endl;
std::cout<<GridLogMessage<<"=========================================================="<<std::endl;
FermionField src_e (&RBGrid);
FermionField src_o (&RBGrid);
FermionField r_e (&RBGrid);
FermionField r_o (&RBGrid);
FermionField r_eo (&Grid);
pickCheckerboard(Even,src_e,src);
pickCheckerboard(Odd,src_o,src);
Dwc.Meooe(src_e,r_o); std::cout<<GridLogMessage<<"Applied Meo"<<std::endl;
Dwc.Meooe(src_o,r_e); std::cout<<GridLogMessage<<"Applied Moe"<<std::endl;
Dwc.Dhop (src,ref,DaggerNo);
setCheckerboard(r_eo,r_o);
setCheckerboard(r_eo,r_e);
err= ref - r_eo;
std::cout<<GridLogMessage << "EO norm diff "<< norm2(err)<< " "<<norm2(ref)<< " " << norm2(r_eo) <<std::endl;
std::cout<<GridLogMessage<<"=============================================================="<<std::endl;
std::cout<<GridLogMessage<<"= Test Ddagger is the dagger of D by requiring "<<std::endl;
std::cout<<GridLogMessage<<"= < phi | Deo | chi > * = < chi | Deo^dag| phi> "<<std::endl;
std::cout<<GridLogMessage<<"=============================================================="<<std::endl;
FermionField chi_e (&RBGrid);
FermionField chi_o (&RBGrid);
FermionField dchi_e (&RBGrid);
FermionField dchi_o (&RBGrid);
FermionField phi_e (&RBGrid);
FermionField phi_o (&RBGrid);
FermionField dphi_e (&RBGrid);
FermionField dphi_o (&RBGrid);
pickCheckerboard(Even,chi_e,chi);
pickCheckerboard(Odd ,chi_o,chi);
pickCheckerboard(Even,phi_e,phi);
pickCheckerboard(Odd ,phi_o,phi);
Dwc.Meooe(chi_e,dchi_o);
Dwc.Meooe(chi_o,dchi_e);
Dwc.MeooeDag(phi_e,dphi_o);
Dwc.MeooeDag(phi_o,dphi_e);
ComplexD pDce = innerProduct(phi_e,dchi_e);
ComplexD pDco = innerProduct(phi_o,dchi_o);
ComplexD cDpe = innerProduct(chi_e,dphi_e);
ComplexD cDpo = innerProduct(chi_o,dphi_o);
std::cout<<GridLogMessage <<"e "<<pDce<<" "<<cDpe <<std::endl;
std::cout<<GridLogMessage <<"o "<<pDco<<" "<<cDpo <<std::endl;
std::cout<<GridLogMessage <<"pDce - conj(cDpo) "<< pDce-conj(cDpo) <<std::endl;
std::cout<<GridLogMessage <<"pDco - conj(cDpe) "<< pDco-conj(cDpe) <<std::endl;
std::cout<<GridLogMessage <<"e "<<pDce<<" "<<cDpe <<std::endl;
std::cout<<GridLogMessage <<"o "<<pDco<<" "<<cDpo <<std::endl;
std::cout<<GridLogMessage <<"pDce - conj(cDpo) "<< pDce-conj(cDpo) <<std::endl;
std::cout<<GridLogMessage <<"pDco - conj(cDpe) "<< pDco-conj(cDpe) <<std::endl;
std::cout<<GridLogMessage<<"=============================================================="<<std::endl;
std::cout<<GridLogMessage<<"= Test MeeInv Mee = 1 "<<std::endl;
std::cout<<GridLogMessage<<"=============================================================="<<std::endl;
pickCheckerboard(Even,chi_e,chi);
pickCheckerboard(Odd ,chi_o,chi);
Dwc.Mooee(chi_e,src_e);
Dwc.MooeeInv(src_e,phi_e);
Dwc.Mooee(chi_o,src_o);
Dwc.MooeeInv(src_o,phi_o);
setCheckerboard(phi,phi_e);
setCheckerboard(phi,phi_o);
err = phi-chi;
std::cout<<GridLogMessage << "norm diff "<< norm2(err)<< std::endl;
std::cout<<GridLogMessage<<"=============================================================="<<std::endl;
std::cout<<GridLogMessage<<"= Test MeeInvDag MeeDag = 1 "<<std::endl;
std::cout<<GridLogMessage<<"=============================================================="<<std::endl;
pickCheckerboard(Even,chi_e,chi);
pickCheckerboard(Odd ,chi_o,chi);
Dwc.MooeeDag(chi_e,src_e);
Dwc.MooeeInvDag(src_e,phi_e);
Dwc.MooeeDag(chi_o,src_o);
Dwc.MooeeInvDag(src_o,phi_o);
setCheckerboard(phi,phi_e);
setCheckerboard(phi,phi_o);
err = phi-chi;
std::cout<<GridLogMessage << "norm diff "<< norm2(err)<< std::endl;
std::cout<<GridLogMessage<<"=============================================================="<<std::endl;
std::cout<<GridLogMessage<<"= Test MpcDagMpc is Hermitian "<<std::endl;
std::cout<<GridLogMessage<<"=============================================================="<<std::endl;
random(pRNG,phi);
random(pRNG,chi);
pickCheckerboard(Even,chi_e,chi);
pickCheckerboard(Odd ,chi_o,chi);
pickCheckerboard(Even,phi_e,phi);
pickCheckerboard(Odd ,phi_o,phi);
SchurDiagMooeeOperator<WilsonCloverFermionR,FermionField> HermOpEO(Dwc);
HermOpEO.MpcDagMpc(chi_e,dchi_e,t1,t2);
HermOpEO.MpcDagMpc(chi_o,dchi_o,t1,t2);
HermOpEO.MpcDagMpc(phi_e,dphi_e,t1,t2);
HermOpEO.MpcDagMpc(phi_o,dphi_o,t1,t2);
pDce = innerProduct(phi_e,dchi_e);
pDco = innerProduct(phi_o,dchi_o);
cDpe = innerProduct(chi_e,dphi_e);
cDpo = innerProduct(chi_o,dphi_o);
std::cout<<GridLogMessage <<"e "<<pDce<<" "<<cDpe <<std::endl;
std::cout<<GridLogMessage <<"o "<<pDco<<" "<<cDpo <<std::endl;
std::cout<<GridLogMessage <<"pDce - conj(cDpo) "<< pDco-conj(cDpo) <<std::endl;
std::cout<<GridLogMessage <<"pDco - conj(cDpe) "<< pDce-conj(cDpe) <<std::endl;
Grid_finalize();
}