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
|
|
|
|
|
|
|
|
namespace Grid {
|
|
|
|
|
|
|
|
namespace QCD {
|
|
|
|
|
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-07-15 00:00:09 +01:00
|
|
|
void MooeeInternal(const FermionField &in, FermionField &out,int dag,int inv);
|
|
|
|
|
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
|
|
|
|
|
|
|
// 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);
|
|
|
|
|
|
|
|
#define CAYLEY_DPERP_CACHE
|
|
|
|
#undef CAYLEY_DPERP_LINALG
|
2015-06-02 16:57:12 +01:00
|
|
|
|
|
|
|
#endif
|