portelli/Grid
1
0
Fork 0
mirror of https://github.com/paboyle/Grid.git synced 2025-05-10 12:35:57 +01:00
Code Releases Activity

Mixed precision changes

Browse Source
This commit is contained in:
Quadro 2021-06-01 13:41:02 -04:00
parent 9164cfbfc6
commit 9e51fa5681
1 changed files with 26 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
Grid/qcd/action/pseudofermion/DomainDecomposedBoundaryTwoFlavourPseudoFermion.h
View File

@ -32,27 +32,25 @@ NAMESPACE_BEGIN(Grid);
/////////////////////////////////////// ///////////////////////////////////////
// Two flavour ratio // Two flavour ratio
/////////////////////////////////////// ///////////////////////////////////////
template<class Impl> template<class ImplD,class ImplF>
class DomainDecomposedBoundaryTwoFlavourPseudoFermion : public Action<typename Impl::GaugeField> { class DomainDecomposedBoundaryTwoFlavourPseudoFermion : public Action<typename ImplD::GaugeField> {
public: public:
INHERIT_IMPL_TYPES(Impl); INHERIT_IMPL_TYPES(ImplD);
private: private:
SchurFactoredFermionOperator<Impl> & DenOp;// the basic operator SchurFactoredFermionOperator<ImplD,ImplF> & DenOp;// the basic operator
RealD ActionStoppingCondition;
OperatorFunction<FermionField> &DerivativeSolver; RealD DerivativeStoppingCondition;
OperatorFunction<FermionField> &ActionSolver;
FermionField Phi; // the pseudo fermion field for this trajectory FermionField Phi; // the pseudo fermion field for this trajectory
RealD refresh_action; RealD refresh_action;
public: public:
DomainDecomposedBoundaryTwoFlavourPseudoFermion(SchurFactoredFermionOperator<Impl> &_DenOp, DomainDecomposedBoundaryTwoFlavourPseudoFermion(SchurFactoredFermionOperator<ImplD,ImplF> &_DenOp,RealD _DerivativeTol, RealD _ActionTol )
OperatorFunction<FermionField> & DS, : DenOp(_DenOp),
OperatorFunction<FermionField> & AS DerivativeStoppingCondition(_DerivativeTol),
) : DenOp(_DenOp), ActionStoppingCondition(_ActionTol),
DerivativeSolver(DS), ActionSolver(AS), Phi(_DenOp.FermionGrid()) {};
Phi(_DenOp.FermOp.FermionGrid()) {};
virtual std::string action_name(){return "DomainDecomposedBoundaryTwoFlavourPseudoFermion";} virtual std::string action_name(){return "DomainDecomposedBoundaryTwoFlavourPseudoFermion";}
@ -78,9 +76,10 @@ public:
// //
RealD scale = std::sqrt(0.5); RealD scale = std::sqrt(0.5);
DenOp.tol=ActionStoppingCondition;
DenOp.ImportGauge(U); DenOp.ImportGauge(U);
FermionField eta(DenOp.FermOp.FermionGrid()); FermionField eta(DenOp.FermionGrid());
gaussian(pRNG,eta); eta=eta*scale; gaussian(pRNG,eta); eta=eta*scale;
@ -95,9 +94,10 @@ public:
////////////////////////////////////////////////////// //////////////////////////////////////////////////////
virtual RealD S(const GaugeField &U) { virtual RealD S(const GaugeField &U) {
DenOp.tol=ActionStoppingCondition;
DenOp.ImportGauge(U); DenOp.ImportGauge(U);
FermionField X(DenOp.FermOp.FermionGrid()); FermionField X(DenOp.FermionGrid());
DenOp.RInv(Phi,X); DenOp.RInv(Phi,X);
@ -108,10 +108,11 @@ public:
virtual void deriv(const GaugeField &U,GaugeField & dSdU) virtual void deriv(const GaugeField &U,GaugeField & dSdU)
{ {
DenOp.tol=DerivativeStoppingCondition;
DenOp.ImportGauge(U); DenOp.ImportGauge(U);
GridBase *fgrid = DenOp.FermOp.FermionGrid(); GridBase *fgrid = DenOp.FermionGrid();
GridBase *ugrid = DenOp.FermOp.GaugeGrid(); GridBase *ugrid = DenOp.GaugeGrid();
FermionField X(fgrid); FermionField X(fgrid);
FermionField Y(fgrid); FermionField Y(fgrid);
@ -127,24 +128,29 @@ public:
FermionField RinvDagRinv_Phi(fgrid); FermionField RinvDagRinv_Phi(fgrid);
// R^-1 term // R^-1 term
DumpSliceNorm("Phi",Phi);
DenOp.dBoundaryBar(Phi,tmp); DenOp.dBoundaryBar(Phi,tmp);
DumpSliceNorm("Ddb Phi",tmp);
DenOp.Dinverse(tmp,DiDdb_Phi); // Vector C DenOp.Dinverse(tmp,DiDdb_Phi); // Vector C
DumpSliceNorm("DiDdb Phi",DiDdb_Phi);
Rinv_Phi = Phi - DiDdb_Phi; Rinv_Phi = Phi - DiDdb_Phi;
DenOp.ProjectBoundaryBar(Rinv_Phi); DenOp.ProjectBoundaryBar(Rinv_Phi);
DumpSliceNorm("Rinv Phi",Rinv_Phi);
// R^-dagger R^-1 term // R^-dagger R^-1 term
DenOp.DinverseDag(Rinv_Phi,DidRinv_Phi); // Vector D DenOp.DinverseDag(Rinv_Phi,DidRinv_Phi); // Vector D
DumpSliceNorm("DidRinv Phi",DidRinv_Phi);
DenOp.dBoundaryBarDag(DidRinv_Phi,DdbdDidRinv_Phi); DenOp.dBoundaryBarDag(DidRinv_Phi,DdbdDidRinv_Phi);
RinvDagRinv_Phi = Rinv_Phi - DdbdDidRinv_Phi; RinvDagRinv_Phi = Rinv_Phi - DdbdDidRinv_Phi;
DenOp.ProjectBoundaryBar(RinvDagRinv_Phi); DenOp.ProjectBoundaryBar(RinvDagRinv_Phi);
X = DiDdb_Phi; X = DiDdb_Phi;
Y = DidRinv_Phi; Y = DidRinv_Phi;
DenOp.FermOp.MDeriv(force,Y,X,DaggerNo); dSdU=force; DenOp.PeriodicFermOpD.MDeriv(force,Y,X,DaggerNo); dSdU=force;
DenOp.FermOp.MDeriv(force,X,Y,DaggerYes); dSdU=dSdU+force; DenOp.PeriodicFermOpD.MDeriv(force,X,Y,DaggerYes); dSdU=dSdU+force;
dSdU *= -1.0; dSdU *= -1.0;
}; };
}; };