2016-01-02 14:51:32 +00:00
|
|
|
/*************************************************************************************
|
|
|
|
|
|
|
|
Grid physics library, www.github.com/paboyle/Grid
|
|
|
|
|
|
|
|
Source file: ./lib/qcd/action/fermion/CayleyFermion5D.h
|
|
|
|
|
|
|
|
Copyright (C) 2015
|
|
|
|
|
|
|
|
Author: Peter Boyle <pabobyle@ph.ed.ac.uk>
|
|
|
|
Author: Peter Boyle <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 */
|
2015-06-02 16:57:12 +01:00
|
|
|
#ifndef GRID_QCD_CAYLEY_FERMION_H
|
|
|
|
#define GRID_QCD_CAYLEY_FERMION_H
|
|
|
|
|
2017-02-22 18:09:33 +00:00
|
|
|
#include <Grid/qcd/action/fermion/WilsonFermion5D.h>
|
|
|
|
|
2015-06-02 16:57:12 +01:00
|
|
|
namespace Grid {
|
|
|
|
|
|
|
|
namespace QCD {
|
|
|
|
|
2016-12-18 02:05:11 +00:00
|
|
|
template<typename T> struct switcheroo {
|
|
|
|
static inline int iscomplex() { return 0; }
|
|
|
|
|
|
|
|
template<class vec>
|
|
|
|
static inline vec mult(vec a, vec b) {
|
|
|
|
return real_mult(a,b);
|
|
|
|
}
|
|
|
|
};
|
|
|
|
template<> struct switcheroo<ComplexD> {
|
|
|
|
static inline int iscomplex() { return 1; }
|
|
|
|
|
|
|
|
template<class vec>
|
|
|
|
static inline vec mult(vec a, vec b) {
|
|
|
|
return a*b;
|
|
|
|
}
|
|
|
|
};
|
|
|
|
template<> struct switcheroo<ComplexF> {
|
|
|
|
static inline int iscomplex() { return 1; }
|
|
|
|
template<class vec>
|
|
|
|
static inline vec mult(vec a, vec b) {
|
|
|
|
return a*b;
|
|
|
|
}
|
|
|
|
};
|
2016-12-09 22:51:32 +00:00
|
|
|
|
|
|
|
|
2015-08-10 20:47:44 +01:00
|
|
|
template<class Impl>
|
|
|
|
class CayleyFermion5D : public WilsonFermion5D<Impl>
|
2015-06-02 16:57:12 +01:00
|
|
|
{
|
|
|
|
public:
|
2015-08-15 23:25:49 +01:00
|
|
|
INHERIT_IMPL_TYPES(Impl);
|
2015-08-10 20:47:44 +01:00
|
|
|
public:
|
2015-06-02 16:57:12 +01:00
|
|
|
|
|
|
|
// override multiply
|
2015-08-10 20:47:44 +01:00
|
|
|
virtual RealD M (const FermionField &in, FermionField &out);
|
|
|
|
virtual RealD Mdag (const FermionField &in, FermionField &out);
|
2015-06-02 16:57:12 +01:00
|
|
|
|
|
|
|
// half checkerboard operations
|
2015-08-10 20:47:44 +01:00
|
|
|
virtual void Meooe (const FermionField &in, FermionField &out);
|
|
|
|
virtual void MeooeDag (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);
|
2016-07-15 00:00:09 +01:00
|
|
|
virtual void Meo5D (const FermionField &psi, FermionField &chi);
|
|
|
|
|
|
|
|
virtual void M5D (const FermionField &psi, FermionField &chi);
|
|
|
|
virtual void M5Ddag(const FermionField &psi, FermionField &chi);
|
|
|
|
|
2016-10-10 23:43:19 +01:00
|
|
|
virtual void Dminus(const FermionField &psi, FermionField &chi);
|
|
|
|
virtual void DminusDag(const FermionField &psi, FermionField &chi);
|
|
|
|
|
2016-07-15 00:00:09 +01:00
|
|
|
/////////////////////////////////////////////////////
|
|
|
|
// Instantiate different versions depending on Impl
|
|
|
|
/////////////////////////////////////////////////////
|
|
|
|
void M5D(const FermionField &psi,
|
|
|
|
const FermionField &phi,
|
|
|
|
FermionField &chi,
|
2016-08-15 22:31:29 +01:00
|
|
|
std::vector<Coeff_t> &lower,
|
|
|
|
std::vector<Coeff_t> &diag,
|
|
|
|
std::vector<Coeff_t> &upper);
|
2016-07-15 00:00:09 +01:00
|
|
|
|
|
|
|
void M5Ddag(const FermionField &psi,
|
|
|
|
const FermionField &phi,
|
|
|
|
FermionField &chi,
|
2016-08-15 22:31:29 +01:00
|
|
|
std::vector<Coeff_t> &lower,
|
|
|
|
std::vector<Coeff_t> &diag,
|
|
|
|
std::vector<Coeff_t> &upper);
|
2016-12-09 22:51:32 +00:00
|
|
|
|
2016-07-15 00:00:09 +01:00
|
|
|
void MooeeInternal(const FermionField &in, FermionField &out,int dag,int inv);
|
2016-12-09 22:51:32 +00:00
|
|
|
void MooeeInternalCompute(int dag, int inv, Vector<iSinglet<Simd> > & Matp, Vector<iSinglet<Simd> > & Matm);
|
|
|
|
|
2016-12-08 17:28:28 +00:00
|
|
|
void MooeeInternalAsm(const FermionField &in, FermionField &out,
|
|
|
|
int LLs, int site,
|
|
|
|
Vector<iSinglet<Simd> > &Matp,
|
|
|
|
Vector<iSinglet<Simd> > &Matm);
|
2016-12-09 22:51:32 +00:00
|
|
|
void MooeeInternalZAsm(const FermionField &in, FermionField &out,
|
|
|
|
int LLs, int site,
|
|
|
|
Vector<iSinglet<Simd> > &Matp,
|
|
|
|
Vector<iSinglet<Simd> > &Matm);
|
2016-12-08 17:28:28 +00:00
|
|
|
|
2016-07-15 00:00:09 +01:00
|
|
|
|
2015-06-04 00:00:45 +01:00
|
|
|
virtual void Instantiatable(void)=0;
|
2015-06-08 12:04:59 +01:00
|
|
|
|
2015-07-31 21:58:35 +01:00
|
|
|
// force terms; five routines; default to Dhop on diagonal
|
2015-08-10 20:47:44 +01:00
|
|
|
virtual void MDeriv (GaugeField &mat,const FermionField &U,const FermionField &V,int dag);
|
|
|
|
virtual void MoeDeriv(GaugeField &mat,const FermionField &U,const FermionField &V,int dag);
|
|
|
|
virtual void MeoDeriv(GaugeField &mat,const FermionField &U,const FermionField &V,int dag);
|
2015-07-31 21:58:35 +01:00
|
|
|
|
2015-06-08 12:04:59 +01:00
|
|
|
// Efficient support for multigrid coarsening
|
2015-08-10 20:47:44 +01:00
|
|
|
virtual void Mdir (const FermionField &in, FermionField &out,int dir,int disp);
|
2015-07-31 21:58:35 +01:00
|
|
|
|
2015-08-10 20:47:44 +01:00
|
|
|
void Meooe5D (const FermionField &in, FermionField &out);
|
|
|
|
void MeooeDag5D (const FermionField &in, FermionField &out);
|
2015-06-08 12:04:59 +01:00
|
|
|
|
2015-06-02 16:57:12 +01:00
|
|
|
// protected:
|
|
|
|
RealD mass;
|
2015-06-03 09:36:26 +01:00
|
|
|
|
2015-06-02 16:57:12 +01:00
|
|
|
// Cayley form Moebius (tanh and zolotarev)
|
2016-08-15 22:31:29 +01:00
|
|
|
std::vector<Coeff_t> omega;
|
|
|
|
std::vector<Coeff_t> bs; // S dependent coeffs
|
|
|
|
std::vector<Coeff_t> cs;
|
|
|
|
std::vector<Coeff_t> as;
|
2015-06-02 16:57:12 +01:00
|
|
|
// For preconditioning Cayley form
|
2016-08-15 22:31:29 +01:00
|
|
|
std::vector<Coeff_t> bee;
|
|
|
|
std::vector<Coeff_t> cee;
|
|
|
|
std::vector<Coeff_t> aee;
|
|
|
|
std::vector<Coeff_t> beo;
|
|
|
|
std::vector<Coeff_t> ceo;
|
|
|
|
std::vector<Coeff_t> aeo;
|
2015-06-02 16:57:12 +01:00
|
|
|
// LDU factorisation of the eeoo matrix
|
2016-08-15 22:31:29 +01:00
|
|
|
std::vector<Coeff_t> lee;
|
|
|
|
std::vector<Coeff_t> leem;
|
|
|
|
std::vector<Coeff_t> uee;
|
|
|
|
std::vector<Coeff_t> ueem;
|
|
|
|
std::vector<Coeff_t> dee;
|
2015-06-02 16:57:12 +01:00
|
|
|
|
2016-12-09 22:51:32 +00:00
|
|
|
// Matrices of 5d ee inverse params
|
|
|
|
Vector<iSinglet<Simd> > MatpInv;
|
|
|
|
Vector<iSinglet<Simd> > MatmInv;
|
|
|
|
Vector<iSinglet<Simd> > MatpInvDag;
|
|
|
|
Vector<iSinglet<Simd> > MatmInvDag;
|
|
|
|
|
2015-06-02 16:57:12 +01:00
|
|
|
// Constructors
|
2015-08-10 20:47:44 +01:00
|
|
|
CayleyFermion5D(GaugeField &_Umu,
|
2015-06-02 16:57:12 +01:00
|
|
|
GridCartesian &FiveDimGrid,
|
|
|
|
GridRedBlackCartesian &FiveDimRedBlackGrid,
|
|
|
|
GridCartesian &FourDimGrid,
|
|
|
|
GridRedBlackCartesian &FourDimRedBlackGrid,
|
2015-08-15 23:25:49 +01:00
|
|
|
RealD _mass,RealD _M5,const ImplParams &p= ImplParams());
|
2015-06-02 16:57:12 +01:00
|
|
|
|
2016-11-29 22:27:55 +00:00
|
|
|
|
|
|
|
|
|
|
|
void CayleyReport(void);
|
|
|
|
void CayleyZeroCounters(void);
|
|
|
|
|
|
|
|
double M5Dflops;
|
|
|
|
double M5Dcalls;
|
|
|
|
double M5Dtime;
|
|
|
|
|
|
|
|
double MooeeInvFlops;
|
|
|
|
double MooeeInvCalls;
|
|
|
|
double MooeeInvTime;
|
2016-10-10 23:43:19 +01:00
|
|
|
|
2015-06-03 09:36:26 +01:00
|
|
|
protected:
|
2015-06-04 00:00:45 +01:00
|
|
|
void SetCoefficientsZolotarev(RealD zolohi,Approx::zolotarev_data *zdata,RealD b,RealD c);
|
|
|
|
void SetCoefficientsTanh(Approx::zolotarev_data *zdata,RealD b,RealD c);
|
2016-08-15 22:31:29 +01:00
|
|
|
void SetCoefficientsInternal(RealD zolo_hi,std::vector<Coeff_t> & gamma,RealD b,RealD c);
|
2015-06-02 16:57:12 +01:00
|
|
|
};
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
2016-07-15 17:19:44 +01:00
|
|
|
#define INSTANTIATE_DPERP(A)\
|
|
|
|
template void CayleyFermion5D< A >::M5D(const FermionField &psi,const FermionField &phi,FermionField &chi,\
|
2016-08-15 22:31:29 +01:00
|
|
|
std::vector<Coeff_t> &lower,std::vector<Coeff_t> &diag,std::vector<Coeff_t> &upper); \
|
2016-07-15 17:19:44 +01:00
|
|
|
template void CayleyFermion5D< A >::M5Ddag(const FermionField &psi,const FermionField &phi,FermionField &chi,\
|
2016-08-15 22:31:29 +01:00
|
|
|
std::vector<Coeff_t> &lower,std::vector<Coeff_t> &diag,std::vector<Coeff_t> &upper); \
|
2016-07-15 17:19:44 +01:00
|
|
|
template void CayleyFermion5D< A >::MooeeInv (const FermionField &psi, FermionField &chi); \
|
|
|
|
template void CayleyFermion5D< A >::MooeeInvDag (const FermionField &psi, FermionField &chi);
|
|
|
|
|
2017-03-30 05:31:34 +01:00
|
|
|
#undef CAYLEY_DPERP_DENSE
|
|
|
|
#define CAYLEY_DPERP_CACHE
|
2016-07-15 17:19:44 +01:00
|
|
|
#undef CAYLEY_DPERP_LINALG
|
2017-03-30 05:31:34 +01:00
|
|
|
#define CAYLEY_DPERP_VEC
|
2015-06-02 16:57:12 +01:00
|
|
|
|
|
|
|
#endif
|