mirror of
				https://github.com/paboyle/Grid.git
				synced 2025-10-24 17:54:47 +01:00 
			
		
		
		
	Compare commits
	
		
			20 Commits
		
	
	
		
			d72e914cf0
			...
			feature/bo
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
|  | 6815e138b4 | ||
|  | e29b97b3ea | ||
|  | ad2b699d2b | ||
|  | b8a7004365 | ||
|  | 994512048e | ||
| 78bae9417c | |||
| dd170ead01 | |||
| 014704856f | |||
|  | ee92e08edb | ||
|  | c1dcee9328 | ||
|  | 6b150961fe | ||
|  | 5bafcaedfa | ||
|  | bfeceae708 | ||
|  | eacb66591f | ||
|  | fadaa85626 | ||
|  | 02a5b0d786 | ||
|  | 0e2141442a | ||
|  | 769eb0eecb | ||
| 85e35c4da1 | |||
| 477b794bc5 | 
| @@ -419,14 +419,15 @@ until convergence | |||||||
| 	} | 	} | ||||||
|       } |       } | ||||||
|  |  | ||||||
|       if ( Nconv < Nstop ) |       if ( Nconv < Nstop ) { | ||||||
| 	std::cout << GridLogIRL << "Nconv ("<<Nconv<<") < Nstop ("<<Nstop<<")"<<std::endl; | 	std::cout << GridLogIRL << "Nconv ("<<Nconv<<") < Nstop ("<<Nstop<<")"<<std::endl; | ||||||
|  | 	std::cout << GridLogIRL << "returning Nstop vectors, the last "<< Nstop-Nconv << "of which might meet convergence criterion only approximately" <<std::endl; | ||||||
|  |       } | ||||||
|       eval=eval2; |       eval=eval2; | ||||||
|        |        | ||||||
|       //Keep only converged |       //Keep only converged | ||||||
|       eval.resize(Nconv);// Nstop? |       eval.resize(Nstop);// was Nconv | ||||||
|       evec.resize(Nconv,grid);// Nstop? |       evec.resize(Nstop,grid);// was Nconv | ||||||
|       basisSortInPlace(evec,eval,reverse); |       basisSortInPlace(evec,eval,reverse); | ||||||
|        |        | ||||||
|     } |     } | ||||||
|   | |||||||
| @@ -124,6 +124,11 @@ public: | |||||||
|   RealD                _b; |   RealD                _b; | ||||||
|   RealD                _c; |   RealD                _c; | ||||||
|  |  | ||||||
|  |   // possible boost | ||||||
|  |   std::vector<ComplexD> qmu; | ||||||
|  |   void set_qmu(std::vector<ComplexD> _qmu) { qmu=_qmu; assert(qmu.size()==Nd);}; | ||||||
|  |   void addQmu(const FermionField &in, FermionField &out, int dag); | ||||||
|  |    | ||||||
|   // Cayley form Moebius (tanh and zolotarev) |   // Cayley form Moebius (tanh and zolotarev) | ||||||
|   Vector<Coeff_t> omega; |   Vector<Coeff_t> omega; | ||||||
|   Vector<Coeff_t> bs;    // S dependent coeffs |   Vector<Coeff_t> bs;    // S dependent coeffs | ||||||
|   | |||||||
| @@ -60,6 +60,50 @@ public: | |||||||
|   //      virtual void   Instantiatable(void)=0; |   //      virtual void   Instantiatable(void)=0; | ||||||
|   virtual void   Instantiatable(void) =0; |   virtual void   Instantiatable(void) =0; | ||||||
|  |  | ||||||
|  |   void FreePropagator(const FermionField &in,FermionField &out,RealD mass,std::vector<Complex> boundary, std::vector<double> twist) | ||||||
|  |   { | ||||||
|  |     std::cout << "Free Propagator for PartialFraction"<<std::endl; | ||||||
|  |     FermionField in_k(in.Grid()); | ||||||
|  |     FermionField prop_k(in.Grid()); | ||||||
|  |      | ||||||
|  |     FFT theFFT((GridCartesian *) in.Grid()); | ||||||
|  |  | ||||||
|  |     //phase for boundary condition | ||||||
|  |     ComplexField coor(in.Grid()); | ||||||
|  |     ComplexField ph(in.Grid());  ph = Zero(); | ||||||
|  |     FermionField in_buf(in.Grid()); in_buf = Zero(); | ||||||
|  |     typedef typename Simd::scalar_type Scalar; | ||||||
|  |     Scalar ci(0.0,1.0); | ||||||
|  |     assert(twist.size() == Nd);//check that twist is Nd | ||||||
|  |     assert(boundary.size() == Nd);//check that boundary conditions is Nd | ||||||
|  |     int shift = 0; | ||||||
|  |     for(unsigned int nu = 0; nu < Nd; nu++) | ||||||
|  |       { | ||||||
|  | 	// Shift coordinate lattice index by 1 to account for 5th dimension. | ||||||
|  | 	LatticeCoordinate(coor, nu + shift); | ||||||
|  | 	double boundary_phase = ::acos(real(boundary[nu])); | ||||||
|  | 	ph = ph + boundary_phase*coor*((1./(in.Grid()->_fdimensions[nu+shift]))); | ||||||
|  | 	//momenta for propagator shifted by twist+boundary | ||||||
|  | 	twist[nu] = twist[nu] + boundary_phase/((2.0*M_PI)); | ||||||
|  |       } | ||||||
|  |     in_buf = exp(ci*ph*(-1.0))*in; | ||||||
|  |  | ||||||
|  |     theFFT.FFT_all_dim(in_k,in,FFT::forward); | ||||||
|  |     this->MomentumSpacePropagatorHw(prop_k,in_k,mass,twist); | ||||||
|  |     theFFT.FFT_all_dim(out,prop_k,FFT::backward); | ||||||
|  |      | ||||||
|  |     //phase for boundary condition | ||||||
|  |     out = out * exp(ci*ph); | ||||||
|  |   }; | ||||||
|  |  | ||||||
|  |   virtual void FreePropagator(const FermionField &in,FermionField &out,RealD mass) { | ||||||
|  |     std::vector<double> twist(Nd,0.0); //default: periodic boundarys in all directions | ||||||
|  |     std::vector<Complex> boundary; | ||||||
|  |     for(int i=0;i<Nd;i++) boundary.push_back(1);//default: periodic boundary conditions | ||||||
|  |     FreePropagator(in,out,mass,boundary,twist); | ||||||
|  |   }; | ||||||
|  |  | ||||||
|  |    | ||||||
|   // Efficient support for multigrid coarsening |   // Efficient support for multigrid coarsening | ||||||
|   virtual void  Mdir (const FermionField &in, FermionField &out,int dir,int disp); |   virtual void  Mdir (const FermionField &in, FermionField &out,int dir,int disp); | ||||||
|   virtual void  MdirAll(const FermionField &in, std::vector<FermionField> &out); |   virtual void  MdirAll(const FermionField &in, std::vector<FermionField> &out); | ||||||
|   | |||||||
| @@ -39,7 +39,7 @@ class PartialFractionFermion5D : public WilsonFermion5D<Impl> | |||||||
| public: | public: | ||||||
|   INHERIT_IMPL_TYPES(Impl); |   INHERIT_IMPL_TYPES(Impl); | ||||||
|  |  | ||||||
|   const int part_frac_chroma_convention=1; |   const int part_frac_chroma_convention=0; | ||||||
|  |  | ||||||
|   void   Meooe_internal(const FermionField &in, FermionField &out,int dag); |   void   Meooe_internal(const FermionField &in, FermionField &out,int dag); | ||||||
|   void   Mooee_internal(const FermionField &in, FermionField &out,int dag); |   void   Mooee_internal(const FermionField &in, FermionField &out,int dag); | ||||||
| @@ -83,12 +83,63 @@ public: | |||||||
| 			   GridRedBlackCartesian &FourDimRedBlackGrid, | 			   GridRedBlackCartesian &FourDimRedBlackGrid, | ||||||
| 			   RealD _mass,RealD M5,const ImplParams &p= ImplParams()); | 			   RealD _mass,RealD M5,const ImplParams &p= ImplParams()); | ||||||
|  |  | ||||||
|  |   PartialFractionFermion5D(GaugeField &_Umu, | ||||||
|  | 			   GridCartesian         &FiveDimGrid, | ||||||
|  | 			   GridRedBlackCartesian &FiveDimRedBlackGrid, | ||||||
|  | 			   GridCartesian         &FourDimGrid, | ||||||
|  | 			   GridRedBlackCartesian &FourDimRedBlackGrid, | ||||||
|  | 			   RealD _mass,RealD M5,std::vector<RealD> &_qmu,const ImplParams &p= ImplParams()); | ||||||
|  |  | ||||||
|  |   void FreePropagator(const FermionField &in,FermionField &out,RealD mass,std::vector<Complex> boundary, std::vector<double> twist) | ||||||
|  |   { | ||||||
|  |     std::cout << "Free Propagator for PartialFraction"<<std::endl; | ||||||
|  |     FermionField in_k(in.Grid()); | ||||||
|  |     FermionField prop_k(in.Grid()); | ||||||
|  |      | ||||||
|  |     FFT theFFT((GridCartesian *) in.Grid()); | ||||||
|  |  | ||||||
|  |     //phase for boundary condition | ||||||
|  |     ComplexField coor(in.Grid()); | ||||||
|  |     ComplexField ph(in.Grid());  ph = Zero(); | ||||||
|  |     FermionField in_buf(in.Grid()); in_buf = Zero(); | ||||||
|  |     typedef typename Simd::scalar_type Scalar; | ||||||
|  |     Scalar ci(0.0,1.0); | ||||||
|  |     assert(twist.size() == Nd);//check that twist is Nd | ||||||
|  |     assert(boundary.size() == Nd);//check that boundary conditions is Nd | ||||||
|  |     int shift = 0; | ||||||
|  |     for(unsigned int nu = 0; nu < Nd; nu++) | ||||||
|  |       { | ||||||
|  | 	// Shift coordinate lattice index by 1 to account for 5th dimension. | ||||||
|  | 	LatticeCoordinate(coor, nu + shift); | ||||||
|  | 	double boundary_phase = ::acos(real(boundary[nu])); | ||||||
|  | 	ph = ph + boundary_phase*coor*((1./(in.Grid()->_fdimensions[nu+shift]))); | ||||||
|  | 	//momenta for propagator shifted by twist+boundary | ||||||
|  | 	twist[nu] = twist[nu] + boundary_phase/((2.0*M_PI)); | ||||||
|  |       } | ||||||
|  |     in_buf = exp(ci*ph*(-1.0))*in; | ||||||
|  |  | ||||||
|  |     theFFT.FFT_all_dim(in_k,in,FFT::forward); | ||||||
|  |     this->MomentumSpacePropagatorHw(prop_k,in_k,mass,twist); | ||||||
|  |     theFFT.FFT_all_dim(out,prop_k,FFT::backward); | ||||||
|  |      | ||||||
|  |     //phase for boundary condition | ||||||
|  |     out = out * exp(ci*ph); | ||||||
|  |   }; | ||||||
|  |  | ||||||
|  |   virtual void FreePropagator(const FermionField &in,FermionField &out,RealD mass) { | ||||||
|  |     std::vector<double> twist(Nd,0.0); //default: periodic boundarys in all directions | ||||||
|  |     std::vector<Complex> boundary; | ||||||
|  |     for(int i=0;i<Nd;i++) boundary.push_back(1);//default: periodic boundary conditions | ||||||
|  |     FreePropagator(in,out,mass,boundary,twist); | ||||||
|  |   }; | ||||||
|  |    | ||||||
| protected: | protected: | ||||||
|  |  | ||||||
|   virtual void SetCoefficientsTanh(Approx::zolotarev_data *zdata,RealD scale); |   virtual void SetCoefficientsTanh(Approx::zolotarev_data *zdata,RealD scale); | ||||||
|   virtual void SetCoefficientsZolotarev(RealD zolo_hi,Approx::zolotarev_data *zdata); |   virtual void SetCoefficientsZolotarev(RealD zolo_hi,Approx::zolotarev_data *zdata); | ||||||
|  |  | ||||||
|   // Part frac |   // Part frac | ||||||
|  |   std::vector<RealD> qmu; | ||||||
|   RealD mass; |   RealD mass; | ||||||
|   RealD dw_diag; |   RealD dw_diag; | ||||||
|   RealD R; |   RealD R; | ||||||
|   | |||||||
| @@ -49,6 +49,7 @@ CayleyFermion5D<Impl>::CayleyFermion5D(GaugeField &_Umu, | |||||||
| 			FourDimRedBlackGrid,_M5,p), | 			FourDimRedBlackGrid,_M5,p), | ||||||
|   mass_plus(_mass), mass_minus(_mass) |   mass_plus(_mass), mass_minus(_mass) | ||||||
| { | { | ||||||
|  |   // qmu defaults to zero size; | ||||||
| } | } | ||||||
|  |  | ||||||
| /////////////////////////////////////////////////////////////// | /////////////////////////////////////////////////////////////// | ||||||
| @@ -270,6 +271,34 @@ void CayleyFermion5D<Impl>::MeooeDag5D    (const FermionField &psi, FermionField | |||||||
|   M5Ddag(psi,psi,Din,lower,diag,upper); |   M5Ddag(psi,psi,Din,lower,diag,upper); | ||||||
| } | } | ||||||
|  |  | ||||||
|  | template<class Impl> | ||||||
|  | void CayleyFermion5D<Impl>::addQmu(const FermionField &psi,FermionField &chi, int dag) | ||||||
|  | { | ||||||
|  |   if ( qmu.size() ) { | ||||||
|  |  | ||||||
|  |     Gamma::Algebra Gmu [] = { | ||||||
|  |       Gamma::Algebra::GammaX, | ||||||
|  |       Gamma::Algebra::GammaY, | ||||||
|  |       Gamma::Algebra::GammaZ, | ||||||
|  |       Gamma::Algebra::GammaT | ||||||
|  |     }; | ||||||
|  |     std::vector<ComplexD> coeff(Nd); | ||||||
|  |     ComplexD ci(0,1); | ||||||
|  |  | ||||||
|  |     assert(qmu.size()==Nd); | ||||||
|  |  | ||||||
|  |     for(int mu=0;mu<Nd;mu++){ | ||||||
|  |        coeff[mu] = ci*qmu[mu]; | ||||||
|  |        if ( dag ) coeff[mu] = conjugate(coeff[mu]); | ||||||
|  |     } | ||||||
|  |  | ||||||
|  |     chi = chi + Gamma(Gmu[0])*psi*coeff[0]; | ||||||
|  |     for(int mu=1;mu<Nd;mu++){ | ||||||
|  |       chi = chi + Gamma(Gmu[mu])*psi*coeff[mu]; | ||||||
|  |     } | ||||||
|  |   } | ||||||
|  | } | ||||||
|  |  | ||||||
| template<class Impl> | template<class Impl> | ||||||
| void CayleyFermion5D<Impl>::M    (const FermionField &psi, FermionField &chi) | void CayleyFermion5D<Impl>::M    (const FermionField &psi, FermionField &chi) | ||||||
| { | { | ||||||
| @@ -279,6 +308,10 @@ void CayleyFermion5D<Impl>::M    (const FermionField &psi, FermionField &chi) | |||||||
|   Meooe5D(psi,Din); |   Meooe5D(psi,Din); | ||||||
|  |  | ||||||
|   this->DW(Din,chi,DaggerNo); |   this->DW(Din,chi,DaggerNo); | ||||||
|  |  | ||||||
|  |   // add i q_mu gamma_mu here | ||||||
|  |   addQmu(Din,chi,DaggerNo); | ||||||
|  |    | ||||||
|   // ((b D_W + D_w hop terms +1) on s-diag |   // ((b D_W + D_w hop terms +1) on s-diag | ||||||
|   axpby(chi,1.0,1.0,chi,psi);  |   axpby(chi,1.0,1.0,chi,psi);  | ||||||
|    |    | ||||||
| @@ -296,6 +329,9 @@ void CayleyFermion5D<Impl>::Mdag (const FermionField &psi, FermionField &chi) | |||||||
|   // Apply Dw |   // Apply Dw | ||||||
|   this->DW(psi,Din,DaggerYes);  |   this->DW(psi,Din,DaggerYes);  | ||||||
|  |  | ||||||
|  |   // add -i conj(q_mu) gamma_mu here ... if qmu is real, gammm_5 hermitian, otherwise not. | ||||||
|  |   addQmu(psi,Din,DaggerYes); | ||||||
|  |    | ||||||
|   MeooeDag5D(Din,chi); |   MeooeDag5D(Din,chi); | ||||||
|    |    | ||||||
|   M5Ddag(psi,chi); |   M5Ddag(psi,chi); | ||||||
|   | |||||||
| @@ -42,13 +42,13 @@ template<class Impl> | |||||||
| void ContinuedFractionFermion5D<Impl>::SetCoefficientsZolotarev(RealD zolo_hi,Approx::zolotarev_data *zdata) | void ContinuedFractionFermion5D<Impl>::SetCoefficientsZolotarev(RealD zolo_hi,Approx::zolotarev_data *zdata) | ||||||
| { | { | ||||||
|   // How to check Ls matches?? |   // How to check Ls matches?? | ||||||
|   //      std::cout<<GridLogMessage << Ls << " Ls"<<std::endl; |   std::cout<<GridLogMessage << zdata->n  << " - n"<<std::endl; | ||||||
|   //      std::cout<<GridLogMessage << zdata->n  << " - n"<<std::endl; |   std::cout<<GridLogMessage << zdata->da << " -da "<<std::endl; | ||||||
|   //      std::cout<<GridLogMessage << zdata->da << " -da "<<std::endl; |   std::cout<<GridLogMessage << zdata->db << " -db"<<std::endl; | ||||||
|   //      std::cout<<GridLogMessage << zdata->db << " -db"<<std::endl; |   std::cout<<GridLogMessage << zdata->dn << " -dn"<<std::endl; | ||||||
|   //      std::cout<<GridLogMessage << zdata->dn << " -dn"<<std::endl; |   std::cout<<GridLogMessage << zdata->dd << " -dd"<<std::endl; | ||||||
|   //      std::cout<<GridLogMessage << zdata->dd << " -dd"<<std::endl; |  | ||||||
|   int Ls = this->Ls; |   int Ls = this->Ls; | ||||||
|  |   std::cout<<GridLogMessage << Ls << " Ls"<<std::endl; | ||||||
|   assert(zdata->db==Ls);// Beta has Ls coeffs |   assert(zdata->db==Ls);// Beta has Ls coeffs | ||||||
|  |  | ||||||
|   R=(1+this->mass)/(1-this->mass); |   R=(1+this->mass)/(1-this->mass); | ||||||
| @@ -320,7 +320,7 @@ ContinuedFractionFermion5D<Impl>::ContinuedFractionFermion5D( | |||||||
|       int Ls = this->Ls; |       int Ls = this->Ls; | ||||||
|       conformable(solution5d.Grid(),this->FermionGrid()); |       conformable(solution5d.Grid(),this->FermionGrid()); | ||||||
|       conformable(exported4d.Grid(),this->GaugeGrid()); |       conformable(exported4d.Grid(),this->GaugeGrid()); | ||||||
|       ExtractSlice(exported4d, solution5d, Ls-1, Ls-1); |       ExtractSlice(exported4d, solution5d, Ls-1, 0); | ||||||
|     } |     } | ||||||
|     template<class Impl> |     template<class Impl> | ||||||
|     void ContinuedFractionFermion5D<Impl>::ImportPhysicalFermionSource(const FermionField &input4d,FermionField &imported5d) |     void ContinuedFractionFermion5D<Impl>::ImportPhysicalFermionSource(const FermionField &input4d,FermionField &imported5d) | ||||||
| @@ -330,7 +330,7 @@ ContinuedFractionFermion5D<Impl>::ContinuedFractionFermion5D( | |||||||
|       conformable(input4d.Grid()   ,this->GaugeGrid()); |       conformable(input4d.Grid()   ,this->GaugeGrid()); | ||||||
|       FermionField tmp(this->FermionGrid()); |       FermionField tmp(this->FermionGrid()); | ||||||
|       tmp=Zero(); |       tmp=Zero(); | ||||||
|       InsertSlice(input4d, tmp, Ls-1, Ls-1); |       InsertSlice(input4d, tmp, Ls-1, 0); | ||||||
|       tmp=Gamma(Gamma::Algebra::Gamma5)*tmp; |       tmp=Gamma(Gamma::Algebra::Gamma5)*tmp; | ||||||
|       this->Dminus(tmp,imported5d); |       this->Dminus(tmp,imported5d); | ||||||
|     } |     } | ||||||
|   | |||||||
| @@ -255,6 +255,45 @@ void   PartialFractionFermion5D<Impl>::M_internal(const FermionField &psi, Fermi | |||||||
|   } |   } | ||||||
| 	 | 	 | ||||||
|   { |   { | ||||||
|  |     // The 'conventional' Cayley overlap operator is | ||||||
|  |     // | ||||||
|  |     // Dov = (1+m)/2 + (1-m)/2 g5 sgn Hw | ||||||
|  |     // | ||||||
|  |     // | ||||||
|  |     // With massless limit 1/2(1+g5 sgnHw) | ||||||
|  |     // | ||||||
|  |     // Luscher shows quite neatly that 1+g5 sgn Hw has tree level propagator i qslash +O(a^2) | ||||||
|  |     // | ||||||
|  |     // However, the conventional normalisation has both a leading order factor of 2 in Zq | ||||||
|  |     // at tree level AND a mass dependent (1-m) that are convenient to absorb. | ||||||
|  |     // | ||||||
|  |     // In WilsonFermion5DImplementation.h, the tree level propagator for Hw is | ||||||
|  |     // | ||||||
|  |     // num = -i sin kmu gmu | ||||||
|  |     // | ||||||
|  |     // denom ( sqrt(sk^2 + (2shk^2 - 1)^2 | ||||||
|  |     //    b_k = sk2 - M5; | ||||||
|  |     //      | ||||||
|  |     //    w_k = sqrt(sk + b_k*b_k); | ||||||
|  |     // | ||||||
|  |     //    denom= ( w_k + b_k + mass*mass) ; | ||||||
|  |     // | ||||||
|  |     //    denom= one/denom; | ||||||
|  |     //    out = num*denom; | ||||||
|  |     // | ||||||
|  |     // Chroma, and Grid define partial fraction via 4d operator | ||||||
|  |     // | ||||||
|  |     //   Dpf = 2/(1-m) x Dov = (1+m)/(1-m) + g5 sgn Hw | ||||||
|  |     // | ||||||
|  |     // Now since: | ||||||
|  |     // | ||||||
|  |     //      (1+m)/(1-m) = (1-m)/(1-m) + 2m/(1-m) = 1 + 2m/(1-m) | ||||||
|  |     // | ||||||
|  |     // This corresponds to a modified mass parameter | ||||||
|  |     // | ||||||
|  |     // It has an annoying  | ||||||
|  |     // | ||||||
|  |     //  | ||||||
|     double R=(1+this->mass)/(1-this->mass); |     double R=(1+this->mass)/(1-this->mass); | ||||||
|     //R g5 psi[Ls] + p[0] H |     //R g5 psi[Ls] + p[0] H | ||||||
|     ag5xpbg5y_ssp(chi,R*scale,psi,p[nblock]*scale/amax,D,Ls-1,Ls-1); |     ag5xpbg5y_ssp(chi,R*scale,psi,p[nblock]*scale/amax,D,Ls-1,Ls-1); | ||||||
| @@ -264,6 +303,28 @@ void   PartialFractionFermion5D<Impl>::M_internal(const FermionField &psi, Fermi | |||||||
|       double pp = p[nblock-1-b]; |       double pp = p[nblock-1-b]; | ||||||
|       axpby_ssp(chi,1.0,chi,-sqrt(amax*pp)*scale*sign,psi,Ls-1,s); |       axpby_ssp(chi,1.0,chi,-sqrt(amax*pp)*scale*sign,psi,Ls-1,s); | ||||||
|     } |     } | ||||||
|  |  | ||||||
|  |     if ( qmu.size() ) { | ||||||
|  |  | ||||||
|  |       FermionField qslash_psi(psi.Grid()); | ||||||
|  |        | ||||||
|  |       Gamma::Algebra Gmu [] = { | ||||||
|  | 			 Gamma::Algebra::GammaX, | ||||||
|  | 			 Gamma::Algebra::GammaY, | ||||||
|  | 			 Gamma::Algebra::GammaZ, | ||||||
|  | 			 Gamma::Algebra::GammaT | ||||||
|  |       }; | ||||||
|  |       ComplexD ci(0,1); | ||||||
|  |       assert(qmu.size()==Nd); | ||||||
|  |       qslash_psi = Gamma(Gmu[0])*psi; | ||||||
|  |       for(int mu=1;mu<Nd;mu++){ | ||||||
|  | 	qslash_psi = Gamma(Gmu[mu])*psi; | ||||||
|  |       } | ||||||
|  |       //      RealD coeff = 1.0; | ||||||
|  |       qslash_psi = Gamma(Gamma::Algebra::Gamma5)*qslash_psi*ci ; // i g5 qslash -- 1-m factor??? | ||||||
|  |       axpby_ssp(chi,1.0,chi,1.0, qslash_psi,Ls-1,Ls-1); | ||||||
|  |     } | ||||||
|  |      | ||||||
|   } |   } | ||||||
|  |  | ||||||
| } | } | ||||||
| @@ -411,7 +472,7 @@ void  PartialFractionFermion5D<Impl>::SetCoefficientsZolotarev(RealD zolo_hi,App | |||||||
|       int Ls = this->Ls; |       int Ls = this->Ls; | ||||||
|       conformable(solution5d.Grid(),this->FermionGrid()); |       conformable(solution5d.Grid(),this->FermionGrid()); | ||||||
|       conformable(exported4d.Grid(),this->GaugeGrid()); |       conformable(exported4d.Grid(),this->GaugeGrid()); | ||||||
|       ExtractSlice(exported4d, solution5d, Ls-1, Ls-1); |       ExtractSlice(exported4d, solution5d, Ls-1, 0); | ||||||
|     } |     } | ||||||
|     template<class Impl> |     template<class Impl> | ||||||
|     void PartialFractionFermion5D<Impl>::ImportPhysicalFermionSource(const FermionField &input4d,FermionField &imported5d) |     void PartialFractionFermion5D<Impl>::ImportPhysicalFermionSource(const FermionField &input4d,FermionField &imported5d) | ||||||
| @@ -421,7 +482,8 @@ void  PartialFractionFermion5D<Impl>::SetCoefficientsZolotarev(RealD zolo_hi,App | |||||||
|       conformable(input4d.Grid()   ,this->GaugeGrid()); |       conformable(input4d.Grid()   ,this->GaugeGrid()); | ||||||
|       FermionField tmp(this->FermionGrid()); |       FermionField tmp(this->FermionGrid()); | ||||||
|       tmp=Zero(); |       tmp=Zero(); | ||||||
|       InsertSlice(input4d, tmp, Ls-1, Ls-1); |       std::cout << " importing to slice " << Ls-1 <<std::endl; | ||||||
|  |       InsertSlice(input4d, tmp, Ls-1, 0); | ||||||
|       tmp=Gamma(Gamma::Algebra::Gamma5)*tmp; |       tmp=Gamma(Gamma::Algebra::Gamma5)*tmp; | ||||||
|       this->Dminus(tmp,imported5d); |       this->Dminus(tmp,imported5d); | ||||||
|     } |     } | ||||||
| @@ -442,7 +504,7 @@ PartialFractionFermion5D<Impl>::PartialFractionFermion5D(GaugeField &_Umu, | |||||||
|  |  | ||||||
| { | { | ||||||
|   int Ls = this->Ls; |   int Ls = this->Ls; | ||||||
|  |   qmu.resize(0); | ||||||
|   assert((Ls&0x1)==1); // Odd Ls required |   assert((Ls&0x1)==1); // Odd Ls required | ||||||
|   int nrational=Ls-1; |   int nrational=Ls-1; | ||||||
|  |  | ||||||
| @@ -460,6 +522,22 @@ PartialFractionFermion5D<Impl>::PartialFractionFermion5D(GaugeField &_Umu, | |||||||
|   Approx::zolotarev_free(zdata); |   Approx::zolotarev_free(zdata); | ||||||
|  |  | ||||||
| } | } | ||||||
|  | template<class Impl> | ||||||
|  | PartialFractionFermion5D<Impl>::PartialFractionFermion5D(GaugeField &_Umu, | ||||||
|  | 							 GridCartesian         &FiveDimGrid, | ||||||
|  | 							 GridRedBlackCartesian &FiveDimRedBlackGrid, | ||||||
|  | 							 GridCartesian         &FourDimGrid, | ||||||
|  | 							 GridRedBlackCartesian &FourDimRedBlackGrid, | ||||||
|  | 							 RealD _mass,RealD M5, | ||||||
|  | 							 std::vector<RealD> &_qmu, | ||||||
|  | 							 const ImplParams &p) | ||||||
|  |   : PartialFractionFermion5D<Impl>(_Umu, | ||||||
|  | 			     FiveDimGrid,FiveDimRedBlackGrid, | ||||||
|  | 			     FourDimGrid,FourDimRedBlackGrid, | ||||||
|  | 			     _mass,M5,p) | ||||||
|  | { | ||||||
|  |   qmu=_qmu; | ||||||
|  | } | ||||||
|  |  | ||||||
| NAMESPACE_END(Grid); | NAMESPACE_END(Grid); | ||||||
|  |  | ||||||
|   | |||||||
| @@ -423,7 +423,6 @@ void WilsonKernels<Impl>::DhopDirKernel( StencilImpl &st, DoubledGaugeField &U,S | |||||||
| #define KERNEL_CALL(A) KERNEL_CALLNB(A); accelerator_barrier(); | #define KERNEL_CALL(A) KERNEL_CALLNB(A); accelerator_barrier(); | ||||||
|  |  | ||||||
| #define KERNEL_CALL_EXT(A)						\ | #define KERNEL_CALL_EXT(A)						\ | ||||||
|   const uint64_t    NN = Nsite*Ls;					\ |  | ||||||
|   const uint64_t    sz = st.surface_list.size();			\ |   const uint64_t    sz = st.surface_list.size();			\ | ||||||
|   auto ptr = &st.surface_list[0];					\ |   auto ptr = &st.surface_list[0];					\ | ||||||
|   accelerator_forNB( ss, sz, Simd::Nsimd(), {				\ |   accelerator_forNB( ss, sz, Simd::Nsimd(), {				\ | ||||||
|   | |||||||
| @@ -40,18 +40,20 @@ Lattice<iScalar<iScalar<iScalar<Vec> > > > Determinant(const Lattice<iScalar<iSc | |||||||
|   GridBase *grid=Umu.Grid(); |   GridBase *grid=Umu.Grid(); | ||||||
|   auto lvol = grid->lSites(); |   auto lvol = grid->lSites(); | ||||||
|   Lattice<iScalar<iScalar<iScalar<Vec> > > > ret(grid); |   Lattice<iScalar<iScalar<iScalar<Vec> > > > ret(grid); | ||||||
|  |   typedef typename Vec::scalar_type scalar; | ||||||
|   autoView(Umu_v,Umu,CpuRead); |   autoView(Umu_v,Umu,CpuRead); | ||||||
|   autoView(ret_v,ret,CpuWrite); |   autoView(ret_v,ret,CpuWrite); | ||||||
|   thread_for(site,lvol,{ |   thread_for(site,lvol,{ | ||||||
|     Eigen::MatrixXcd EigenU = Eigen::MatrixXcd::Zero(N,N); |     Eigen::MatrixXcd EigenU = Eigen::MatrixXcd::Zero(N,N); | ||||||
|     Coordinate lcoor; |     Coordinate lcoor; | ||||||
|     grid->LocalIndexToLocalCoor(site, lcoor); |     grid->LocalIndexToLocalCoor(site, lcoor); | ||||||
|     iScalar<iScalar<iMatrix<ComplexD, N> > > Us; |     iScalar<iScalar<iMatrix<scalar, N> > > Us; | ||||||
|     peekLocalSite(Us, Umu_v, lcoor); |     peekLocalSite(Us, Umu_v, lcoor); | ||||||
|     for(int i=0;i<N;i++){ |     for(int i=0;i<N;i++){ | ||||||
|       for(int j=0;j<N;j++){ |       for(int j=0;j<N;j++){ | ||||||
| 	EigenU(i,j) = Us()()(i,j); | 	scalar tmp= Us()()(i,j); | ||||||
|  | 	ComplexD ztmp(real(tmp),imag(tmp)); | ||||||
|  | 	EigenU(i,j)=ztmp; | ||||||
|       }} |       }} | ||||||
|     ComplexD detD  = EigenU.determinant(); |     ComplexD detD  = EigenU.determinant(); | ||||||
|     typename Vec::scalar_type det(detD.real(),detD.imag()); |     typename Vec::scalar_type det(detD.real(),detD.imag()); | ||||||
|   | |||||||
| @@ -705,7 +705,7 @@ public: | |||||||
| 	} | 	} | ||||||
|       } |       } | ||||||
|     } |     } | ||||||
|     std::cout << "BuildSurfaceList size is "<<surface_list.size()<<std::endl; |     std::cout << GridLogDebug << "BuildSurfaceList size is "<<surface_list.size()<<std::endl; | ||||||
|   } |   } | ||||||
|   /// Introduce a block structure and switch off comms on boundaries |   /// Introduce a block structure and switch off comms on boundaries | ||||||
|   void DirichletBlock(const Coordinate &dirichlet_block) |   void DirichletBlock(const Coordinate &dirichlet_block) | ||||||
|   | |||||||
| @@ -55,7 +55,7 @@ template<class vtype, int N> accelerator_inline iVector<vtype, N> Exponentiate(c | |||||||
|  |  | ||||||
|  |  | ||||||
| // Specialisation: Cayley-Hamilton exponential for SU(3) | // Specialisation: Cayley-Hamilton exponential for SU(3) | ||||||
| #ifndef GRID_ACCELERATED | #if 0 | ||||||
| template<class vtype, typename std::enable_if< GridTypeMapper<vtype>::TensorLevel == 0>::type * =nullptr>  | template<class vtype, typename std::enable_if< GridTypeMapper<vtype>::TensorLevel == 0>::type * =nullptr>  | ||||||
| accelerator_inline iMatrix<vtype,3> Exponentiate(const iMatrix<vtype,3> &arg, RealD alpha  , Integer Nexp = DEFAULT_MAT_EXP ) | accelerator_inline iMatrix<vtype,3> Exponentiate(const iMatrix<vtype,3> &arg, RealD alpha  , Integer Nexp = DEFAULT_MAT_EXP ) | ||||||
| { | { | ||||||
|   | |||||||
							
								
								
									
										224
									
								
								HMC/FTHMC2p1f.cc
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										224
									
								
								HMC/FTHMC2p1f.cc
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,224 @@ | |||||||
|  | /************************************************************************************* | ||||||
|  |  | ||||||
|  | Grid physics library, www.github.com/paboyle/Grid | ||||||
|  |  | ||||||
|  | Copyright (C) 2023 | ||||||
|  |  | ||||||
|  | Author: Peter Boyle <pabobyle@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/Grid.h> | ||||||
|  | #include <Grid/qcd/smearing/GaugeConfigurationMasked.h> | ||||||
|  | #include <Grid/qcd/smearing/JacobianAction.h> | ||||||
|  |  | ||||||
|  | using namespace Grid; | ||||||
|  |  | ||||||
|  | int main(int argc, char **argv) | ||||||
|  | { | ||||||
|  |   std::cout << std::setprecision(12); | ||||||
|  |    | ||||||
|  |   Grid_init(&argc, &argv); | ||||||
|  |   int threads = GridThread::GetThreads(); | ||||||
|  |   // here make a routine to print all the relevant information on the run | ||||||
|  |   std::cout << GridLogMessage << "Grid is setup to use " << threads << " threads" << std::endl; | ||||||
|  |  | ||||||
|  |    // Typedefs to simplify notation | ||||||
|  |   typedef WilsonImplR FermionImplPolicy; | ||||||
|  |   typedef MobiusFermionD FermionAction; | ||||||
|  |   typedef typename FermionAction::FermionField FermionField; | ||||||
|  |  | ||||||
|  |   typedef Grid::XmlReader       Serialiser; | ||||||
|  |  | ||||||
|  |   //:::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::: | ||||||
|  |   IntegratorParameters MD; | ||||||
|  |   //  typedef GenericHMCRunner<LeapFrog> HMCWrapper; | ||||||
|  |   //  MD.name    = std::string("Leap Frog"); | ||||||
|  |   //  typedef GenericHMCRunner<ForceGradient> HMCWrapper; | ||||||
|  |   //  MD.name    = std::string("Force Gradient"); | ||||||
|  |   typedef GenericHMCRunner<MinimumNorm2> HMCWrapper; | ||||||
|  |   MD.name    = std::string("MinimumNorm2"); | ||||||
|  |   MD.MDsteps = 12; | ||||||
|  |   MD.trajL   = 1.0; | ||||||
|  |  | ||||||
|  |   HMCparameters HMCparams; | ||||||
|  |   HMCparams.StartTrajectory  = 0; | ||||||
|  |   HMCparams.Trajectories     = 200; | ||||||
|  |   HMCparams.NoMetropolisUntil=  20; | ||||||
|  |   // "[HotStart, ColdStart, TepidStart, CheckpointStart]\n"; | ||||||
|  |   HMCparams.StartingType     =std::string("HotStart"); | ||||||
|  |   HMCparams.MD = MD; | ||||||
|  |   HMCWrapper TheHMC(HMCparams); | ||||||
|  |  | ||||||
|  |   // Grid from the command line arguments --grid and --mpi | ||||||
|  |   TheHMC.Resources.AddFourDimGrid("gauge"); // use default simd lanes decomposition | ||||||
|  |  | ||||||
|  |   CheckpointerParameters CPparams; | ||||||
|  |   CPparams.config_prefix = "ckpoint_EODWF_lat"; | ||||||
|  |   CPparams.smeared_prefix = "ckpoint_EODWF_lat_smr"; | ||||||
|  |   CPparams.rng_prefix    = "ckpoint_EODWF_rng"; | ||||||
|  |   CPparams.saveInterval  = 1; | ||||||
|  |   CPparams.saveSmeared   = true; | ||||||
|  |   CPparams.format        = "IEEE64BIG"; | ||||||
|  |   TheHMC.Resources.LoadNerscCheckpointer(CPparams); | ||||||
|  |  | ||||||
|  |   RNGModuleParameters RNGpar; | ||||||
|  |   RNGpar.serial_seeds = "1 2 3 4 5"; | ||||||
|  |   RNGpar.parallel_seeds = "6 7 8 9 10"; | ||||||
|  |   TheHMC.Resources.SetRNGSeeds(RNGpar); | ||||||
|  |  | ||||||
|  |   // Construct observables | ||||||
|  |   // here there is too much indirection | ||||||
|  |   typedef PlaquetteMod<HMCWrapper::ImplPolicy> PlaqObs; | ||||||
|  |   TheHMC.Resources.AddObservable<PlaqObs>(); | ||||||
|  |   ////////////////////////////////////////////// | ||||||
|  |  | ||||||
|  |   const int Ls      = 16; | ||||||
|  |   Real beta         = 2.13; | ||||||
|  |   Real light_mass   = 0.01; | ||||||
|  |   Real strange_mass = 0.04; | ||||||
|  |   Real pv_mass      = 1.0; | ||||||
|  |   RealD M5  = 1.8; | ||||||
|  |   RealD b   = 1.0; // Scale factor two | ||||||
|  |   RealD c   = 0.0; | ||||||
|  |  | ||||||
|  |   OneFlavourRationalParams OFRp; | ||||||
|  |   OFRp.lo       = 1.0e-2; | ||||||
|  |   OFRp.hi       = 64; | ||||||
|  |   OFRp.MaxIter  = 10000; | ||||||
|  |   OFRp.tolerance= 1.0e-10; | ||||||
|  |   OFRp.degree   = 14; | ||||||
|  |   OFRp.precision= 40; | ||||||
|  |  | ||||||
|  |   std::vector<Real> hasenbusch({ 0.1 }); | ||||||
|  |  | ||||||
|  |   auto GridPtr   = TheHMC.Resources.GetCartesian(); | ||||||
|  |   auto GridRBPtr = TheHMC.Resources.GetRBCartesian(); | ||||||
|  |   auto FGrid     = SpaceTimeGrid::makeFiveDimGrid(Ls,GridPtr); | ||||||
|  |   auto FrbGrid   = SpaceTimeGrid::makeFiveDimRedBlackGrid(Ls,GridPtr); | ||||||
|  |  | ||||||
|  |   IwasakiGaugeActionR GaugeAction(beta); | ||||||
|  |  | ||||||
|  |   // temporarily need a gauge field | ||||||
|  |   LatticeGaugeField U(GridPtr); | ||||||
|  |   LatticeGaugeField Uhot(GridPtr); | ||||||
|  |  | ||||||
|  |   // These lines are unecessary if BC are all periodic | ||||||
|  |   std::vector<Complex> boundary = {1,1,1,-1}; | ||||||
|  |   FermionAction::ImplParams Params(boundary); | ||||||
|  |  | ||||||
|  |   double StoppingCondition = 1e-10; | ||||||
|  |   double MaxCGIterations = 30000; | ||||||
|  |   ConjugateGradient<FermionField>  CG(StoppingCondition,MaxCGIterations); | ||||||
|  |  | ||||||
|  |   bool ApplySmearing = true; | ||||||
|  |    | ||||||
|  |   //////////////////////////////////// | ||||||
|  |   // Collect actions | ||||||
|  |   //////////////////////////////////// | ||||||
|  |   ActionLevel<HMCWrapper::Field> Level1(1); | ||||||
|  |   ActionLevel<HMCWrapper::Field> Level2(2); | ||||||
|  |   ActionLevel<HMCWrapper::Field> Level3(4); | ||||||
|  |  | ||||||
|  |   //////////////////////////////////// | ||||||
|  |   // Strange action | ||||||
|  |   //////////////////////////////////// | ||||||
|  |  | ||||||
|  |   MobiusEOFAFermionD Strange_Op_L (U , *FGrid , *FrbGrid , *GridPtr , *GridRBPtr , strange_mass, strange_mass, pv_mass, 0.0, -1, M5, b, c); | ||||||
|  |   MobiusEOFAFermionD Strange_Op_R (U , *FGrid , *FrbGrid , *GridPtr , *GridRBPtr , pv_mass, strange_mass,      pv_mass, -1.0, 1, M5, b, c); | ||||||
|  |   ExactOneFlavourRatioPseudoFermionAction<FermionImplPolicy>  | ||||||
|  |     EOFA(Strange_Op_L, Strange_Op_R,  | ||||||
|  | 	 CG, | ||||||
|  | 	 CG, CG, | ||||||
|  | 	 CG, CG,  | ||||||
|  | 	 OFRp, false); | ||||||
|  |  | ||||||
|  |   EOFA.is_smeared = ApplySmearing; | ||||||
|  |   Level1.push_back(&EOFA); | ||||||
|  |  | ||||||
|  |   //////////////////////////////////// | ||||||
|  |   // up down action | ||||||
|  |   //////////////////////////////////// | ||||||
|  |   std::vector<Real> light_den; | ||||||
|  |   std::vector<Real> light_num; | ||||||
|  |  | ||||||
|  |   int n_hasenbusch = hasenbusch.size(); | ||||||
|  |   light_den.push_back(light_mass); | ||||||
|  |   for(int h=0;h<n_hasenbusch;h++){ | ||||||
|  |     light_den.push_back(hasenbusch[h]); | ||||||
|  |     light_num.push_back(hasenbusch[h]); | ||||||
|  |   } | ||||||
|  |   light_num.push_back(pv_mass); | ||||||
|  |  | ||||||
|  |   std::vector<FermionAction *> Numerators; | ||||||
|  |   std::vector<FermionAction *> Denominators; | ||||||
|  |   std::vector<TwoFlavourEvenOddRatioPseudoFermionAction<FermionImplPolicy> *> Quotients; | ||||||
|  |  | ||||||
|  |   for(int h=0;h<n_hasenbusch+1;h++){ | ||||||
|  |     std::cout << GridLogMessage << " 2f quotient Action  "<< light_num[h] << " / " << light_den[h]<< std::endl; | ||||||
|  |     Numerators.push_back  (new FermionAction(U,*FGrid,*FrbGrid,*GridPtr,*GridRBPtr,light_num[h],M5,b,c, Params)); | ||||||
|  |     Denominators.push_back(new FermionAction(U,*FGrid,*FrbGrid,*GridPtr,*GridRBPtr,light_den[h],M5,b,c, Params)); | ||||||
|  |     Quotients.push_back   (new TwoFlavourEvenOddRatioPseudoFermionAction<FermionImplPolicy>(*Numerators[h],*Denominators[h],CG,CG)); | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   for(int h=0;h<n_hasenbusch+1;h++){ | ||||||
|  |     Quotients[h]->is_smeared = ApplySmearing; | ||||||
|  |     Level1.push_back(Quotients[h]); | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   ///////////////////////////////////////////////////////////// | ||||||
|  |   // lnDetJacobianAction | ||||||
|  |   ///////////////////////////////////////////////////////////// | ||||||
|  |   double rho = 0.1;  // smearing parameter | ||||||
|  |   int Nsmear = 1;    // number of smearing levels - must be multiple of 2Nd | ||||||
|  |   int Nstep  = 8*Nsmear;    // number of smearing levels - must be multiple of 2Nd | ||||||
|  |   Smear_Stout<HMCWrapper::ImplPolicy> Stout(rho); | ||||||
|  |   SmearedConfigurationMasked<HMCWrapper::ImplPolicy> SmearingPolicy(GridPtr, Nstep, Stout); | ||||||
|  |   JacobianAction<HMCWrapper::ImplPolicy> Jacobian(&SmearingPolicy); | ||||||
|  |   if( ApplySmearing ) Level2.push_back(&Jacobian); | ||||||
|  |   std::cout << GridLogMessage << " Built the Jacobian "<< std::endl; | ||||||
|  |  | ||||||
|  |  | ||||||
|  |   ///////////////////////////////////////////////////////////// | ||||||
|  |   // Gauge action | ||||||
|  |   ///////////////////////////////////////////////////////////// | ||||||
|  |   //  GaugeAction.is_smeared = ApplySmearing; | ||||||
|  |   GaugeAction.is_smeared = true; | ||||||
|  |   Level3.push_back(&GaugeAction); | ||||||
|  |  | ||||||
|  |   std::cout << GridLogMessage << " ************************************************"<< std::endl; | ||||||
|  |   std::cout << GridLogMessage << " Action complete -- NO FERMIONS FOR NOW -- FIXME"<< std::endl; | ||||||
|  |   std::cout << GridLogMessage << " ************************************************"<< std::endl; | ||||||
|  |   std::cout << GridLogMessage <<  std::endl; | ||||||
|  |   std::cout << GridLogMessage <<  std::endl; | ||||||
|  |  | ||||||
|  |  | ||||||
|  |   std::cout << GridLogMessage << " Running the FT HMC "<< std::endl; | ||||||
|  |  | ||||||
|  |   TheHMC.TheAction.push_back(Level1); | ||||||
|  |   TheHMC.TheAction.push_back(Level2); | ||||||
|  |   TheHMC.TheAction.push_back(Level3); | ||||||
|  |  | ||||||
|  |   TheHMC.Run(SmearingPolicy); // for smearing | ||||||
|  |  | ||||||
|  |   Grid_finalize(); | ||||||
|  | } // main | ||||||
|  |  | ||||||
|  |  | ||||||
|  |  | ||||||
| @@ -146,6 +146,8 @@ NAMESPACE_END(Grid); | |||||||
| int main(int argc, char **argv) { | int main(int argc, char **argv) { | ||||||
|   using namespace Grid; |   using namespace Grid; | ||||||
|  |  | ||||||
|  |   std::cout << " Grid Initialise "<<std::endl; | ||||||
|  |    | ||||||
|   Grid_init(&argc, &argv); |   Grid_init(&argc, &argv); | ||||||
|  |  | ||||||
|   CartesianCommunicator::BarrierWorld(); |   CartesianCommunicator::BarrierWorld(); | ||||||
| @@ -170,24 +172,24 @@ int main(int argc, char **argv) { | |||||||
|   IntegratorParameters MD; |   IntegratorParameters MD; | ||||||
|   //  typedef GenericHMCRunner<LeapFrog> HMCWrapper; |   //  typedef GenericHMCRunner<LeapFrog> HMCWrapper; | ||||||
|   //  MD.name    = std::string("Leap Frog"); |   //  MD.name    = std::string("Leap Frog"); | ||||||
|   typedef GenericHMCRunner<ForceGradient> HMCWrapper; |   //  typedef GenericHMCRunner<ForceGradient> HMCWrapper; | ||||||
|   MD.name    = std::string("Force Gradient"); |   //  MD.name    = std::string("Force Gradient"); | ||||||
|   //typedef GenericHMCRunner<MinimumNorm2> HMCWrapper; |   typedef GenericHMCRunner<MinimumNorm2> HMCWrapper; | ||||||
|   // MD.name    = std::string("MinimumNorm2"); |   MD.name    = std::string("MinimumNorm2"); | ||||||
|   // TrajL = 2 |   // TrajL = 2 | ||||||
|   // 4/2 => 0.6 dH |   // 4/2 => 0.6 dH | ||||||
|   // 3/3 => 0.8 dH .. depth 3, slower |   // 3/3 => 0.8 dH .. depth 3, slower | ||||||
|   //MD.MDsteps =  4; |   //MD.MDsteps =  4; | ||||||
|   MD.MDsteps =  12; |   MD.MDsteps =  14; | ||||||
|   MD.trajL   = 0.5; |   MD.trajL   = 0.5; | ||||||
|  |  | ||||||
|   HMCparameters HMCparams; |   HMCparameters HMCparams; | ||||||
|   HMCparams.StartTrajectory  = 1077; |   HMCparams.StartTrajectory  = 1077; | ||||||
|   HMCparams.Trajectories     = 1; |   HMCparams.Trajectories     = 20; | ||||||
|   HMCparams.NoMetropolisUntil=  0; |   HMCparams.NoMetropolisUntil=  0; | ||||||
|   // "[HotStart, ColdStart, TepidStart, CheckpointStart]\n"; |   // "[HotStart, ColdStart, TepidStart, CheckpointStart]\n"; | ||||||
|   //  HMCparams.StartingType     =std::string("ColdStart"); |   HMCparams.StartingType     =std::string("ColdStart"); | ||||||
|   HMCparams.StartingType     =std::string("CheckpointStart"); |   //  HMCparams.StartingType     =std::string("CheckpointStart"); | ||||||
|   HMCparams.MD = MD; |   HMCparams.MD = MD; | ||||||
|   HMCWrapper TheHMC(HMCparams); |   HMCWrapper TheHMC(HMCparams); | ||||||
|  |  | ||||||
| @@ -223,7 +225,7 @@ int main(int argc, char **argv) { | |||||||
|   Real pv_mass      = 1.0; |   Real pv_mass      = 1.0; | ||||||
|   //  std::vector<Real> hasenbusch({ 0.01, 0.045, 0.108, 0.25, 0.51 , pv_mass }); |   //  std::vector<Real> hasenbusch({ 0.01, 0.045, 0.108, 0.25, 0.51 , pv_mass }); | ||||||
|   //  std::vector<Real> hasenbusch({ light_mass, 0.01, 0.045, 0.108, 0.25, 0.51 , pv_mass }); |   //  std::vector<Real> hasenbusch({ light_mass, 0.01, 0.045, 0.108, 0.25, 0.51 , pv_mass }); | ||||||
|   std::vector<Real> hasenbusch({ 0.005, 0.0145, 0.045, 0.108, 0.25, 0.51 , pv_mass }); // Updated |   std::vector<Real> hasenbusch({ 0.005, 0.0145, 0.045, 0.108, 0.25, 0.51 }); // Updated | ||||||
|   //  std::vector<Real> hasenbusch({ light_mass, 0.0145, 0.045, 0.108, 0.25, 0.51 , 0.75 , pv_mass }); |   //  std::vector<Real> hasenbusch({ light_mass, 0.0145, 0.045, 0.108, 0.25, 0.51 , 0.75 , pv_mass }); | ||||||
|  |  | ||||||
|   auto GridPtr   = TheHMC.Resources.GetCartesian(); |   auto GridPtr   = TheHMC.Resources.GetCartesian(); | ||||||
| @@ -275,10 +277,10 @@ int main(int argc, char **argv) { | |||||||
|  |  | ||||||
|   //  double StoppingCondition = 1e-14; |   //  double StoppingCondition = 1e-14; | ||||||
|   //  double MDStoppingCondition = 1e-9; |   //  double MDStoppingCondition = 1e-9; | ||||||
|   double StoppingCondition = 1e-8; |   double StoppingCondition = 1e-9; | ||||||
|   double MDStoppingCondition = 1e-7; |   double MDStoppingCondition = 1e-8; | ||||||
|   double MDStoppingConditionLoose = 1e-7; |   double MDStoppingConditionLoose = 1e-8; | ||||||
|   double MDStoppingConditionStrange = 1e-7; |   double MDStoppingConditionStrange = 1e-8; | ||||||
|   double MaxCGIterations = 300000; |   double MaxCGIterations = 300000; | ||||||
|   ConjugateGradient<FermionField>  CG(StoppingCondition,MaxCGIterations); |   ConjugateGradient<FermionField>  CG(StoppingCondition,MaxCGIterations); | ||||||
|   ConjugateGradient<FermionField>  MDCG(MDStoppingCondition,MaxCGIterations); |   ConjugateGradient<FermionField>  MDCG(MDStoppingCondition,MaxCGIterations); | ||||||
|   | |||||||
							
								
								
									
										44
									
								
								systems/Lumi/benchmarks/bench2.slurm
									
									
									
									
									
										Executable file
									
								
							
							
						
						
									
										44
									
								
								systems/Lumi/benchmarks/bench2.slurm
									
									
									
									
									
										Executable file
									
								
							| @@ -0,0 +1,44 @@ | |||||||
|  | #!/bin/bash -l | ||||||
|  | #SBATCH --job-name=bench_lehner | ||||||
|  | #SBATCH --partition=small-g | ||||||
|  | #SBATCH --nodes=2 | ||||||
|  | #SBATCH --ntasks-per-node=8 | ||||||
|  | #SBATCH --cpus-per-task=7 | ||||||
|  | #SBATCH --gpus-per-node=8 | ||||||
|  | #SBATCH --time=00:10:00 | ||||||
|  | #SBATCH --account=project_465000546 | ||||||
|  | #SBATCH --gpu-bind=none | ||||||
|  | #SBATCH --exclusive | ||||||
|  | #SBATCH --mem=0 | ||||||
|  |  | ||||||
|  | CPU_BIND="map_cpu:48,56,32,40,16,24,1,8" | ||||||
|  | echo $CPU_BIND | ||||||
|  |  | ||||||
|  | cat << EOF > select_gpu | ||||||
|  | #!/bin/bash | ||||||
|  | export GPU_MAP=(0 1 2 3 4 5 6 7) | ||||||
|  | export GPU=\${GPU_MAP[\$SLURM_LOCALID]} | ||||||
|  | export HIP_VISIBLE_DEVICES=\$GPU | ||||||
|  | unset ROCR_VISIBLE_DEVICES | ||||||
|  | echo RANK \$SLURM_LOCALID using GPU \$GPU     | ||||||
|  | exec \$* | ||||||
|  | EOF | ||||||
|  |  | ||||||
|  | chmod +x ./select_gpu | ||||||
|  |  | ||||||
|  | root=/scratch/project_465000546/boylepet/Grid/systems/Lumi | ||||||
|  | source ${root}/sourceme.sh | ||||||
|  |  | ||||||
|  | export OMP_NUM_THREADS=7 | ||||||
|  | export MPICH_GPU_SUPPORT_ENABLED=1 | ||||||
|  | export MPICH_SMP_SINGLE_COPY_MODE=XPMEM | ||||||
|  |  | ||||||
|  | for vol in 16.16.16.64 32.32.32.64  32.32.32.128 | ||||||
|  | do | ||||||
|  | srun --cpu-bind=${CPU_BIND} ./select_gpu ./Benchmark_dwf_fp32 --mpi 2.2.2.2 --accelerator-threads 8 --comms-overlap --shm 2048 --shm-mpi 0 --grid $vol  > log.shm0.ov.$vol | ||||||
|  | #srun --cpu-bind=${CPU_BIND} ./select_gpu ./Benchmark_dwf_fp32 --mpi 2.2.2.2 --accelerator-threads 8 --comms-overlap --shm 2048 --shm-mpi 1 --grid $vol  > log.shm1.ov.$vol | ||||||
|  |  | ||||||
|  | srun --cpu-bind=${CPU_BIND} ./select_gpu ./Benchmark_dwf_fp32 --mpi 2.2.2.2 --accelerator-threads 8 --comms-sequential --shm 2048 --shm-mpi 0 --grid $vol  > log.shm0.seq.$vol | ||||||
|  | #srun --cpu-bind=${CPU_BIND} ./select_gpu ./Benchmark_dwf_fp32 --mpi 2.2.2.2 --accelerator-threads 8 --comms-sequential --shm 2048 --shm-mpi 1 --grid $vol > log.shm1.seq.$vol | ||||||
|  | done | ||||||
|  |  | ||||||
| @@ -3,30 +3,28 @@ spack load gmp | |||||||
| spack load mpfr | spack load mpfr | ||||||
| CLIME=`spack find --paths c-lime | grep c-lime| cut -c 15-` | CLIME=`spack find --paths c-lime | grep c-lime| cut -c 15-` | ||||||
| GMP=`spack find --paths gmp | grep gmp | cut -c 12-` | GMP=`spack find --paths gmp | grep gmp | cut -c 12-` | ||||||
| MPFR=`spack find --paths mpfr | grep mpfr | cut -c 12-` | MPFR=`spack find --paths mpfr | grep mpfr | cut -c 13-` | ||||||
| echo clime $CLIME | echo clime X$CLIME | ||||||
| echo gmp $GMP | echo gmp X$GMP | ||||||
| echo mpfr $MPFR | echo mpfr X$MPFR | ||||||
|  |  | ||||||
| ../../configure --enable-comms=mpi-auto \ | ../../configure \ | ||||||
|  | --enable-comms=mpi-auto \ | ||||||
| --with-lime=$CLIME \ | --with-lime=$CLIME \ | ||||||
| --enable-unified=no \ | --enable-unified=no \ | ||||||
| --enable-shm=nvlink \ | --enable-shm=nvlink \ | ||||||
| --enable-tracing=timer \ |  | ||||||
| --enable-accelerator=hip \ | --enable-accelerator=hip \ | ||||||
| --enable-gen-simd-width=64 \ | --enable-gen-simd-width=64 \ | ||||||
| --enable-simd=GPU \ | --enable-simd=GPU \ | ||||||
| --disable-accelerator-cshift \ | --enable-accelerator-cshift \ | ||||||
| --with-gmp=$OLCF_GMP_ROOT \ | --with-gmp=$GMP \ | ||||||
|  | --with-mpfr=$MPFR \ | ||||||
| --with-fftw=$FFTW_DIR/.. \ | --with-fftw=$FFTW_DIR/.. \ | ||||||
| --with-mpfr=/opt/cray/pe/gcc/mpfr/3.1.4/ \ |  | ||||||
| --disable-fermion-reps \ | --disable-fermion-reps \ | ||||||
| --disable-gparity \ | --disable-gparity \ | ||||||
| CXX=hipcc MPICXX=mpicxx \ | CXX=hipcc MPICXX=mpicxx \ | ||||||
| CXXFLAGS="-fPIC -I{$ROCM_PATH}/include/ -std=c++14 -I${MPICH_DIR}/include -L/lib64 --amdgpu-target=gfx90a" \ |   CXXFLAGS="-fPIC --offload-arch=gfx90a -I/opt/rocm/include/ -std=c++14 -I/opt/cray/pe/mpich/8.1.23/ofi/gnu/9.1/include" \ | ||||||
|  LDFLAGS="-L/lib64 -L/opt/rocm-5.2.0/lib/ -L${MPICH_DIR}/lib -lmpi -L${CRAY_MPICH_ROOTDIR}/gtl/lib -lmpi_gtl_hsa -lamdhip64 --amdgpu-target=gfx90a " |   LDFLAGS="-L/opt/cray/pe/mpich/8.1.23/ofi/gnu/9.1/lib -lmpi -L/opt/cray/pe/mpich/8.1.23/gtl/lib -lmpi_gtl_hsa -lamdhip64 -fopenmp"  | ||||||
|  |  | ||||||
|  |  | ||||||
| #--enable-simd=GPU-RRII \ |  | ||||||
|  |  | ||||||
|  |  | ||||||
|   | |||||||
| @@ -1 +1,5 @@ | |||||||
| module load CrayEnv LUMI/22.12 partition/G  cray-fftw/3.3.10.1 | source ~/spack/share/spack/setup-env.sh | ||||||
|  | module load CrayEnv LUMI/22.12 partition/G  cray-fftw/3.3.10.1 rocm | ||||||
|  | spack load c-lime | ||||||
|  | spack load gmp | ||||||
|  | spack load mpfr | ||||||
|   | |||||||
| @@ -1,4 +1,4 @@ | |||||||
| BREW=/opt/local/ | BREW=/opt/local/ | ||||||
| MPICXX=mpicxx CXX=c++-12 ../../configure --enable-simd=GEN --enable-comms=mpi-auto --enable-unified=yes --prefix $HOME/QCD/GridInstall --with-lime=/Users/peterboyle/QCD/SciDAC/install/ --with-openssl=$BREW --disable-fermion-reps --disable-gparity --disable-debug | CXX=mpicxx-openmpi-mp ../../configure --enable-simd=GEN --enable-comms=mpi --enable-unified=yes --prefix $HOME/QCD/GridInstall --with-lime=/Users/peterboyle/QCD/SciDAC/install/ --with-openssl=$BREW --disable-fermion-reps --disable-gparity --disable-debug | ||||||
|  |  | ||||||
|  |  | ||||||
|   | |||||||
							
								
								
									
										307
									
								
								tests/core/Test_fft_pf.cc
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										307
									
								
								tests/core/Test_fft_pf.cc
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,307 @@ | |||||||
|  |     /************************************************************************************* | ||||||
|  |  | ||||||
|  |     grid` physics library, www.github.com/paboyle/Grid  | ||||||
|  |  | ||||||
|  |     Source file: ./tests/Test_cshift.cc | ||||||
|  |  | ||||||
|  |     Copyright (C) 2015 | ||||||
|  |  | ||||||
|  | Author: Azusa Yamaguchi <ayamaguc@staffmail.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 */ | ||||||
|  | #include <Grid/Grid.h> | ||||||
|  |  | ||||||
|  | using namespace Grid; | ||||||
|  |  | ||||||
|  | int main (int argc, char ** argv) | ||||||
|  | { | ||||||
|  |   Grid_init(&argc,&argv); | ||||||
|  |  | ||||||
|  |   int threads = GridThread::GetThreads(); | ||||||
|  |   std::cout<<GridLogMessage << "Grid is setup to use "<<threads<<" threads"<<std::endl; | ||||||
|  |  | ||||||
|  |   Coordinate latt_size   = GridDefaultLatt(); | ||||||
|  |   Coordinate simd_layout( { vComplexD::Nsimd(),1,1,1}); | ||||||
|  |   Coordinate mpi_layout  = GridDefaultMpi(); | ||||||
|  |  | ||||||
|  |   int vol = 1; | ||||||
|  |   for(int d=0;d<latt_size.size();d++){ | ||||||
|  |     vol = vol * latt_size[d]; | ||||||
|  |   } | ||||||
|  |   GridCartesian         GRID(latt_size,simd_layout,mpi_layout); | ||||||
|  |   GridRedBlackCartesian RBGRID(&GRID); | ||||||
|  |  | ||||||
|  |   ComplexD ci(0.0,1.0); | ||||||
|  |  | ||||||
|  |   std::vector<int> seeds({1,2,3,4}); | ||||||
|  |   GridSerialRNG          sRNG;  sRNG.SeedFixedIntegers(seeds); // naughty seeding | ||||||
|  |   GridParallelRNG          pRNG(&GRID); | ||||||
|  |   pRNG.SeedFixedIntegers(seeds); | ||||||
|  |  | ||||||
|  |   LatticeGaugeFieldD Umu(&GRID); | ||||||
|  |  | ||||||
|  |   SU<Nc>::ColdConfiguration(pRNG,Umu); // Unit gauge | ||||||
|  |  | ||||||
|  |   //////////////////////////////////////////////////// | ||||||
|  |   // PF prop | ||||||
|  |   //////////////////////////////////////////////////// | ||||||
|  |   LatticeFermionD    src(&GRID); | ||||||
|  |  | ||||||
|  |   gaussian(pRNG,src); | ||||||
|  | #if 1 | ||||||
|  |     Coordinate point(4,0); | ||||||
|  |     src=Zero(); | ||||||
|  |     SpinColourVectorD ferm; gaussian(sRNG,ferm); | ||||||
|  |     pokeSite(ferm,src,point); | ||||||
|  | #endif | ||||||
|  |    | ||||||
|  |   { | ||||||
|  |     std::cout<<"****************************************"<<std::endl; | ||||||
|  |     std::cout << "Testing PartialFraction Hw kernel Mom space 4d propagator \n"; | ||||||
|  |     std::cout<<"****************************************"<<std::endl; | ||||||
|  |  | ||||||
|  |     //    LatticeFermionD    src(&GRID); gaussian(pRNG,src); | ||||||
|  |     LatticeFermionD    tmp(&GRID); | ||||||
|  |     LatticeFermionD    ref(&GRID); | ||||||
|  |     LatticeFermionD    diff(&GRID); | ||||||
|  |  | ||||||
|  |     const int Ls=48+1; | ||||||
|  |     GridCartesian         * FGrid   = SpaceTimeGrid::makeFiveDimGrid(Ls,&GRID); | ||||||
|  |     GridRedBlackCartesian * FrbGrid = SpaceTimeGrid::makeFiveDimRedBlackGrid(Ls,&GRID); | ||||||
|  |  | ||||||
|  |     RealD mass=0.1; | ||||||
|  |     RealD M5  =0.8; | ||||||
|  |     OverlapWilsonPartialFractionZolotarevFermionD Dov(Umu,*FGrid,*FrbGrid,GRID,RBGRID,mass,M5,0.001,8.0); | ||||||
|  |  | ||||||
|  |     // Momentum space prop | ||||||
|  |     std::cout << " Solving by FFT and Feynman rules" <<std::endl; | ||||||
|  |     bool fiveD = false; //calculate 4d free propagator | ||||||
|  |  | ||||||
|  |     std::cout << " Free propagator " <<std::endl; | ||||||
|  |     Dov.FreePropagator(src,ref,mass) ; | ||||||
|  |     std::cout << " Free propagator norm "<< norm2(ref) <<std::endl; | ||||||
|  |  | ||||||
|  |     Gamma G5(Gamma::Algebra::Gamma5); | ||||||
|  |  | ||||||
|  |     LatticeFermionD    src5(FGrid); src5=Zero(); | ||||||
|  |     LatticeFermionD    tmp5(FGrid);  | ||||||
|  |     LatticeFermionD    result5(FGrid); result5=Zero(); | ||||||
|  |     LatticeFermionD    result4(&GRID);  | ||||||
|  |     const int sdir=0; | ||||||
|  |  | ||||||
|  |     //////////////////////////////////////////////////////////////////////// | ||||||
|  |     // Import | ||||||
|  |     //////////////////////////////////////////////////////////////////////// | ||||||
|  |     std::cout << " Free propagator Import "<< norm2(src) <<std::endl; | ||||||
|  |     Dov.ImportPhysicalFermionSource  (src,src5); | ||||||
|  |     std::cout << " Free propagator Imported "<< norm2(src5) <<std::endl; | ||||||
|  |      | ||||||
|  |     //////////////////////////////////////////////////////////////////////// | ||||||
|  |     // Conjugate gradient on normal equations system | ||||||
|  |     //////////////////////////////////////////////////////////////////////// | ||||||
|  |     std::cout << " Solving by Conjugate Gradient (CGNE)" <<std::endl; | ||||||
|  |     Dov.Mdag(src5,tmp5); | ||||||
|  |     src5=tmp5; | ||||||
|  |     MdagMLinearOperator<OverlapWilsonPartialFractionZolotarevFermionD,LatticeFermionD> HermOp(Dov); | ||||||
|  |     ConjugateGradient<LatticeFermionD> CG(1.0e-8,10000); | ||||||
|  |     CG(HermOp,src5,result5); | ||||||
|  |     //////////////////////////////////////////////////////////////////////// | ||||||
|  |     // Domain wall physical field propagator | ||||||
|  |     //////////////////////////////////////////////////////////////////////// | ||||||
|  |     Dov.ExportPhysicalFermionSolution(result5,result4); | ||||||
|  |  | ||||||
|  |     // From DWF4d.pdf : | ||||||
|  |     // | ||||||
|  |     // Dov_pf = 2/(1-m) D_cayley_ovlap  [ Page 43 ] | ||||||
|  |     // Dinv_cayley_ovlap = 2/(1-m) Dinv_pf  | ||||||
|  |     // Dinv_cayley_surface =1/(1-m) ( Dinv_cayley_ovlap - 1 ) =>  2/(1-m)^2 Dinv_pf - 1/(1-m) * src   [ Eq.2.67 ] | ||||||
|  |  | ||||||
|  |     RealD scale = 2.0/(1.0-mass)/(1.0-mass); | ||||||
|  |     result4 = result4 * scale; | ||||||
|  |     result4 = result4 - src*(1.0/(1.0-mass)); // Subtract contact term | ||||||
|  |     DumpSliceNorm("Src",src); | ||||||
|  |     DumpSliceNorm("Grid",result4); | ||||||
|  |     DumpSliceNorm("Fourier",ref); | ||||||
|  |  | ||||||
|  |     std::cout << "Dov result4 "<<norm2(result4)<<std::endl; | ||||||
|  |     std::cout << "Dov ref     "<<norm2(ref)<<std::endl; | ||||||
|  |  | ||||||
|  |     diff = result4- ref; | ||||||
|  |     DumpSliceNorm("diff ",diff); | ||||||
|  |      | ||||||
|  |   } | ||||||
|  |    | ||||||
|  |   //////////////////////////////////////////////////// | ||||||
|  |   // Dwf prop | ||||||
|  |   //////////////////////////////////////////////////// | ||||||
|  |   { | ||||||
|  |     std::cout<<"****************************************"<<std::endl; | ||||||
|  |     std::cout << "Testing Dov(Hw) Mom space 4d propagator \n"; | ||||||
|  |     std::cout<<"****************************************"<<std::endl; | ||||||
|  |  | ||||||
|  |     LatticeFermionD    tmp(&GRID); | ||||||
|  |     LatticeFermionD    ref(&GRID); | ||||||
|  |     LatticeFermionD    diff(&GRID); | ||||||
|  |  | ||||||
|  |     const int Ls=48; | ||||||
|  |     GridCartesian         * FGrid   = SpaceTimeGrid::makeFiveDimGrid(Ls,&GRID); | ||||||
|  |     GridRedBlackCartesian * FrbGrid = SpaceTimeGrid::makeFiveDimRedBlackGrid(Ls,&GRID); | ||||||
|  |  | ||||||
|  |     RealD mass=0.1; | ||||||
|  |     RealD M5  =0.8; | ||||||
|  |  | ||||||
|  |     OverlapWilsonCayleyTanhFermionD Dov(Umu,*FGrid,*FrbGrid,GRID,RBGRID,mass,M5,1.0); | ||||||
|  |  | ||||||
|  |     // Momentum space prop | ||||||
|  |     std::cout << " Solving by FFT and Feynman rules" <<std::endl; | ||||||
|  |     Dov.FreePropagator(src,ref,mass) ; | ||||||
|  |  | ||||||
|  |     Gamma G5(Gamma::Algebra::Gamma5); | ||||||
|  |  | ||||||
|  |     LatticeFermionD    src5(FGrid); src5=Zero(); | ||||||
|  |     LatticeFermionD    tmp5(FGrid);  | ||||||
|  |     LatticeFermionD    result5(FGrid); result5=Zero(); | ||||||
|  |     LatticeFermionD    result4(&GRID);  | ||||||
|  |     const int sdir=0; | ||||||
|  |  | ||||||
|  |     //////////////////////////////////////////////////////////////////////// | ||||||
|  |     // Domain wall physical field source; need D_minus | ||||||
|  |     //////////////////////////////////////////////////////////////////////// | ||||||
|  |     /* | ||||||
|  | 	chi_5[0]   = chiralProjectPlus(chi); | ||||||
|  | 	chi_5[Ls-1]= chiralProjectMinus(chi); | ||||||
|  |     */       | ||||||
|  |     tmp =   (src + G5*src)*0.5;      InsertSlice(tmp,src5,   0,sdir); | ||||||
|  |     tmp =   (src - G5*src)*0.5;      InsertSlice(tmp,src5,Ls-1,sdir); | ||||||
|  |      | ||||||
|  |     //////////////////////////////////////////////////////////////////////// | ||||||
|  |     // Conjugate gradient on normal equations system | ||||||
|  |     //////////////////////////////////////////////////////////////////////// | ||||||
|  |     std::cout << " Solving by Conjugate Gradient (CGNE)" <<std::endl; | ||||||
|  |     Dov.Dminus(src5,tmp5); | ||||||
|  |     src5=tmp5; | ||||||
|  |     Dov.Mdag(src5,tmp5); | ||||||
|  |     src5=tmp5; | ||||||
|  |     MdagMLinearOperator<OverlapWilsonCayleyTanhFermionD,LatticeFermionD> HermOp(Dov); | ||||||
|  |     ConjugateGradient<LatticeFermionD> CG(1.0e-16,10000); | ||||||
|  |     CG(HermOp,src5,result5); | ||||||
|  |      | ||||||
|  |     //////////////////////////////////////////////////////////////////////// | ||||||
|  |     // Domain wall physical field propagator | ||||||
|  |     //////////////////////////////////////////////////////////////////////// | ||||||
|  |     /* | ||||||
|  |       psi  = chiralProjectMinus(psi_5[0]); | ||||||
|  |       psi += chiralProjectPlus(psi_5[Ls-1]); | ||||||
|  |     */ | ||||||
|  |     ExtractSlice(tmp,result5,0   ,sdir);   result4 =         (tmp-G5*tmp)*0.5; | ||||||
|  |     ExtractSlice(tmp,result5,Ls-1,sdir);   result4 = result4+(tmp+G5*tmp)*0.5; | ||||||
|  |      | ||||||
|  |     std::cout << " Taking difference" <<std::endl; | ||||||
|  |     std::cout << "Dov result4 "<<norm2(result4)<<std::endl; | ||||||
|  |     std::cout << "Dov ref     "<<norm2(ref)<<std::endl; | ||||||
|  |     DumpSliceNorm("Grid",result4); | ||||||
|  |     DumpSliceNorm("Fourier",ref); | ||||||
|  |     diff = ref - result4; | ||||||
|  |     std::cout << "result - ref     "<<norm2(diff)<<std::endl; | ||||||
|  |      | ||||||
|  |     DumpSliceNorm("diff",diff); | ||||||
|  |  | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |    | ||||||
|  |   { | ||||||
|  |     std::cout<<"****************************************"<<std::endl; | ||||||
|  |     std::cout << "Testing PartialFraction Hw kernel Mom space 4d propagator with q\n"; | ||||||
|  |     std::cout<<"****************************************"<<std::endl; | ||||||
|  |  | ||||||
|  |     //    LatticeFermionD    src(&GRID); gaussian(pRNG,src); | ||||||
|  |     LatticeFermionD    tmp(&GRID); | ||||||
|  |     LatticeFermionD    ref(&GRID); | ||||||
|  |     LatticeFermionD    diff(&GRID); | ||||||
|  |  | ||||||
|  |     const int Ls=48+1; | ||||||
|  |     GridCartesian         * FGrid   = SpaceTimeGrid::makeFiveDimGrid(Ls,&GRID); | ||||||
|  |     GridRedBlackCartesian * FrbGrid = SpaceTimeGrid::makeFiveDimRedBlackGrid(Ls,&GRID); | ||||||
|  |  | ||||||
|  |     RealD mass=0.1; | ||||||
|  |     RealD M5  =0.8; | ||||||
|  |     OverlapWilsonPartialFractionZolotarevFermionD Dov(Umu,*FGrid,*FrbGrid,GRID,RBGRID,mass,M5,0.001,8.0); | ||||||
|  |  | ||||||
|  |     // Momentum space prop | ||||||
|  |     std::cout << " Solving by FFT and Feynman rules" <<std::endl; | ||||||
|  |     bool fiveD = false; //calculate 4d free propagator | ||||||
|  |  | ||||||
|  |     std::cout << " Free propagator " <<std::endl; | ||||||
|  |     Dov.FreePropagator(src,ref,mass) ; | ||||||
|  |     std::cout << " Free propagator norm "<< norm2(ref) <<std::endl; | ||||||
|  |  | ||||||
|  |     Gamma G5(Gamma::Algebra::Gamma5); | ||||||
|  |  | ||||||
|  |     LatticeFermionD    src5(FGrid); src5=Zero(); | ||||||
|  |     LatticeFermionD    tmp5(FGrid);  | ||||||
|  |     LatticeFermionD    result5(FGrid); result5=Zero(); | ||||||
|  |     LatticeFermionD    result4(&GRID);  | ||||||
|  |     const int sdir=0; | ||||||
|  |  | ||||||
|  |     //////////////////////////////////////////////////////////////////////// | ||||||
|  |     // Import | ||||||
|  |     //////////////////////////////////////////////////////////////////////// | ||||||
|  |     std::cout << " Free propagator Import "<< norm2(src) <<std::endl; | ||||||
|  |     Dov.ImportPhysicalFermionSource  (src,src5); | ||||||
|  |     std::cout << " Free propagator Imported "<< norm2(src5) <<std::endl; | ||||||
|  |      | ||||||
|  |     //////////////////////////////////////////////////////////////////////// | ||||||
|  |     // Conjugate gradient on normal equations system | ||||||
|  |     //////////////////////////////////////////////////////////////////////// | ||||||
|  |     std::cout << " Solving by Conjugate Gradient (CGNE)" <<std::endl; | ||||||
|  |     Dov.Mdag(src5,tmp5); | ||||||
|  |     src5=tmp5; | ||||||
|  |     MdagMLinearOperator<OverlapWilsonPartialFractionZolotarevFermionD,LatticeFermionD> HermOp(Dov); | ||||||
|  |     ConjugateGradient<LatticeFermionD> CG(1.0e-8,10000); | ||||||
|  |     CG(HermOp,src5,result5); | ||||||
|  |     //////////////////////////////////////////////////////////////////////// | ||||||
|  |     // Domain wall physical field propagator | ||||||
|  |     //////////////////////////////////////////////////////////////////////// | ||||||
|  |     Dov.ExportPhysicalFermionSolution(result5,result4); | ||||||
|  |  | ||||||
|  |     // From DWF4d.pdf : | ||||||
|  |     // | ||||||
|  |     // Dov_pf = 2/(1-m) D_cayley_ovlap  [ Page 43 ] | ||||||
|  |     // Dinv_cayley_ovlap = 2/(1-m) Dinv_pf  | ||||||
|  |     // Dinv_cayley_surface =1/(1-m) ( Dinv_cayley_ovlap - 1 ) =>  2/(1-m)^2 Dinv_pf - 1/(1-m) * src   [ Eq.2.67 ] | ||||||
|  |  | ||||||
|  |     RealD scale = 2.0/(1.0-mass)/(1.0-mass); | ||||||
|  |     result4 = result4 * scale; | ||||||
|  |     result4 = result4 - src*(1.0/(1.0-mass)); // Subtract contact term | ||||||
|  |     DumpSliceNorm("Src",src); | ||||||
|  |     DumpSliceNorm("Grid",result4); | ||||||
|  |     DumpSliceNorm("Fourier",ref); | ||||||
|  |  | ||||||
|  |     std::cout << "Dov result4 "<<norm2(result4)<<std::endl; | ||||||
|  |     std::cout << "Dov ref     "<<norm2(ref)<<std::endl; | ||||||
|  |  | ||||||
|  |     diff = result4- ref; | ||||||
|  |     DumpSliceNorm("diff ",diff); | ||||||
|  |      | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |    | ||||||
|  |   Grid_finalize(); | ||||||
|  | } | ||||||
| @@ -1,7 +1,6 @@ | |||||||
|     /************************************************************************************* |     /************************************************************************************* | ||||||
|  |  | ||||||
|     Grid physics library, www.github.com/paboyle/Grid  |     Grid physics library, www.github.com/paboyle/Grid  | ||||||
|  |  | ||||||
|     Source file: ./tests/qdpxx/Test_qdpxx_munprec.cc |     Source file: ./tests/qdpxx/Test_qdpxx_munprec.cc | ||||||
|  |  | ||||||
|     Copyright (C) 2015 |     Copyright (C) 2015 | ||||||
| @@ -26,13 +25,17 @@ Author: paboyle <paboyle@ph.ed.ac.uk> | |||||||
|     See the full license in the file "LICENSE" in the top level distribution directory |     See the full license in the file "LICENSE" in the top level distribution directory | ||||||
|     *************************************************************************************/ |     *************************************************************************************/ | ||||||
|     /*  END LEGAL */ |     /*  END LEGAL */ | ||||||
|  | #include <chroma.h> | ||||||
|  | #include <actions/ferm/invert/syssolver_linop_cg_array.h> | ||||||
|  | #include <actions/ferm/invert/syssolver_linop_aggregate.h> | ||||||
|  |  | ||||||
| #include <Grid/Grid.h> | #include <Grid/Grid.h> | ||||||
|  |  | ||||||
| int    Ls=8; | int    Ls=8; | ||||||
| double M5=1.6; | double M5=1.6; | ||||||
| double mq=0.01; | double mq=0.01; | ||||||
| double zolo_lo = 0.1; | double zolo_lo = 0.01; | ||||||
| double zolo_hi = 2.0; | double zolo_hi = 7.0; | ||||||
| double mobius_scale=2.0; | double mobius_scale=2.0; | ||||||
|  |  | ||||||
| enum ChromaAction { | enum ChromaAction { | ||||||
| @@ -55,11 +58,6 @@ enum ChromaAction { | |||||||
| void calc_grid      (ChromaAction action,Grid::LatticeGaugeField & lat, Grid::LatticeFermion &src, Grid::LatticeFermion &res,int dag); | void calc_grid      (ChromaAction action,Grid::LatticeGaugeField & lat, Grid::LatticeFermion &src, Grid::LatticeFermion &res,int dag); | ||||||
| void calc_chroma    (ChromaAction action,Grid::LatticeGaugeField & lat, Grid::LatticeFermion &src, Grid::LatticeFermion &res,int dag); | void calc_chroma    (ChromaAction action,Grid::LatticeGaugeField & lat, Grid::LatticeFermion &src, Grid::LatticeFermion &res,int dag); | ||||||
|  |  | ||||||
| #include <chroma.h> |  | ||||||
| #include <actions/ferm/invert/syssolver_linop_cg_array.h> |  | ||||||
| #include <actions/ferm/invert/syssolver_linop_aggregate.h> |  | ||||||
|  |  | ||||||
|  |  | ||||||
|  |  | ||||||
| namespace Chroma {  | namespace Chroma {  | ||||||
|  |  | ||||||
| @@ -81,7 +79,7 @@ public: | |||||||
|  |  | ||||||
|     std::vector<int> x(4); |     std::vector<int> x(4); | ||||||
|     QDP::multi1d<int> cx(4); |     QDP::multi1d<int> cx(4); | ||||||
|     std::vector<int> gd= gr.Grid()->GlobalDimensions(); |     Grid::Coordinate gd = gr.Grid()->GlobalDimensions(); | ||||||
|  |  | ||||||
|     for (x[0]=0;x[0]<gd[0];x[0]++){ |     for (x[0]=0;x[0]<gd[0];x[0]++){ | ||||||
|     for (x[1]=0;x[1]<gd[1];x[1]++){ |     for (x[1]=0;x[1]<gd[1];x[1]++){ | ||||||
| @@ -124,7 +122,7 @@ public: | |||||||
|  |  | ||||||
|     std::vector<int> x(5); |     std::vector<int> x(5); | ||||||
|     QDP::multi1d<int> cx(4); |     QDP::multi1d<int> cx(4); | ||||||
|     std::vector<int> gd= gr.Grid()->GlobalDimensions(); |     Grid::Coordinate gd= gr.Grid()->GlobalDimensions(); | ||||||
|  |  | ||||||
|     for (x[0]=0;x[0]<gd[0];x[0]++){ |     for (x[0]=0;x[0]<gd[0];x[0]++){ | ||||||
|     for (x[1]=0;x[1]<gd[1];x[1]++){ |     for (x[1]=0;x[1]<gd[1];x[1]++){ | ||||||
| @@ -166,7 +164,7 @@ public: | |||||||
|  |  | ||||||
|     std::vector<int> x(5); |     std::vector<int> x(5); | ||||||
|     QDP::multi1d<int> cx(4); |     QDP::multi1d<int> cx(4); | ||||||
|     std::vector<int> gd= gr.Grid()->GlobalDimensions(); |     Grid::Coordinate gd= gr.Grid()->GlobalDimensions(); | ||||||
|  |  | ||||||
|     for (x[0]=0;x[0]<gd[0];x[0]++){ |     for (x[0]=0;x[0]<gd[0];x[0]++){ | ||||||
|     for (x[1]=0;x[1]<gd[1];x[1]++){ |     for (x[1]=0;x[1]<gd[1];x[1]++){ | ||||||
| @@ -304,7 +302,30 @@ public: | |||||||
|      //     param.approximation_type=COEFF_TYPE_TANH_UNSCALED; |      //     param.approximation_type=COEFF_TYPE_TANH_UNSCALED; | ||||||
|      //     param.approximation_type=COEFF_TYPE_TANH; |      //     param.approximation_type=COEFF_TYPE_TANH; | ||||||
|      param.tuning_strategy_xml= |      param.tuning_strategy_xml= | ||||||
| "<TuningStrategy><Name>OVEXT_CONSTANT_STRATEGY</Name></TuningStrategy>\n"; | "<TuningStrategy><Name>OVEXT_CONSTANT_STRATEGY</Name><TuningConstant>1.0</TuningConstant></TuningStrategy>\n"; | ||||||
|  |      UnprecOvExtFermActArray S_f(cfs,param); | ||||||
|  |      Handle< FermState<T4,U,U> > fs( S_f.createState(u) ); | ||||||
|  |      Handle< LinearOperatorArray<T4> > M(S_f.linOp(fs)); | ||||||
|  |      return M; | ||||||
|  |    } | ||||||
|  |    if ( parms == HwPartFracTanh ) { | ||||||
|  |      if ( Ls%2 == 0 ) {  | ||||||
|  |        printf("Ls is not odd\n"); | ||||||
|  |        exit(-1); | ||||||
|  |      } | ||||||
|  |      UnprecOvExtFermActArrayParams param; | ||||||
|  |      param.OverMass=M5;  | ||||||
|  |      param.Mass=_mq; | ||||||
|  |      param.RatPolyDeg = Ls; | ||||||
|  |      param.ApproxMin =eps_lo; | ||||||
|  |      param.ApproxMax =eps_hi; | ||||||
|  |      param.b5 =1.0; | ||||||
|  |      param.c5 =1.0; | ||||||
|  |      //     param.approximation_type=COEFF_TYPE_ZOLOTAREV; | ||||||
|  |      param.approximation_type=COEFF_TYPE_TANH_UNSCALED; | ||||||
|  |      //param.approximation_type=COEFF_TYPE_TANH; | ||||||
|  |      param.tuning_strategy_xml= | ||||||
|  |        "<TuningStrategy><Name>OVEXT_CONSTANT_STRATEGY</Name><TuningConstant>1.0</TuningConstant></TuningStrategy>\n"; | ||||||
|      UnprecOvExtFermActArray S_f(cfs,param); |      UnprecOvExtFermActArray S_f(cfs,param); | ||||||
|      Handle< FermState<T4,U,U> > fs( S_f.createState(u) ); |      Handle< FermState<T4,U,U> > fs( S_f.createState(u) ); | ||||||
|      Handle< LinearOperatorArray<T4> > M(S_f.linOp(fs)); |      Handle< LinearOperatorArray<T4> > M(S_f.linOp(fs)); | ||||||
| @@ -316,7 +337,35 @@ public: | |||||||
|      param.ApproxMin=eps_lo; |      param.ApproxMin=eps_lo; | ||||||
|      param.ApproxMax=eps_hi; |      param.ApproxMax=eps_hi; | ||||||
|      param.approximation_type=COEFF_TYPE_ZOLOTAREV; |      param.approximation_type=COEFF_TYPE_ZOLOTAREV; | ||||||
|      param.RatPolyDeg=Ls; |      param.RatPolyDeg=Ls-1; | ||||||
|  |      // The following is why I think Chroma made some directional errors: | ||||||
|  |      param.AuxFermAct= std::string( | ||||||
|  | "<AuxFermAct>\n" | ||||||
|  | "  <FermAct>UNPRECONDITIONED_WILSON</FermAct>\n" | ||||||
|  | "  <Mass>-1.8</Mass>\n" | ||||||
|  | "  <b5>1</b5>\n" | ||||||
|  | "  <c5>0</c5>\n" | ||||||
|  | "  <MaxCG>1000</MaxCG>\n" | ||||||
|  | "  <RsdCG>1.0e-9</RsdCG>\n" | ||||||
|  | "  <FermionBC>\n" | ||||||
|  | "      <FermBC>SIMPLE_FERMBC</FermBC>\n" | ||||||
|  | "      <boundary>1 1 1 1</boundary>\n" | ||||||
|  | "   </FermionBC> \n" | ||||||
|  | "</AuxFermAct>" | ||||||
|  | ); | ||||||
|  |      param.AuxFermActGrp= std::string(""); | ||||||
|  |      UnprecOvlapContFrac5DFermActArray S_f(fbc,param); | ||||||
|  |      Handle< FermState<T4,U,U> > fs( S_f.createState(u) ); | ||||||
|  |      Handle< LinearOperatorArray<T4> > M(S_f.linOp(fs)); | ||||||
|  |      return  M; | ||||||
|  |    } | ||||||
|  |    if ( parms == HwContFracTanh ) { | ||||||
|  |      UnprecOvlapContFrac5DFermActParams param; | ||||||
|  |      param.Mass=_mq; // How is M5 set? Wilson mass In AuxFermAct | ||||||
|  |      param.ApproxMin=eps_lo; | ||||||
|  |      param.ApproxMax=eps_hi; | ||||||
|  |      param.approximation_type=COEFF_TYPE_TANH_UNSCALED; | ||||||
|  |      param.RatPolyDeg=Ls-1; | ||||||
|      // The following is why I think Chroma made some directional errors: |      // The following is why I think Chroma made some directional errors: | ||||||
|      param.AuxFermAct= std::string( |      param.AuxFermAct= std::string( | ||||||
| "<AuxFermAct>\n" | "<AuxFermAct>\n" | ||||||
| @@ -378,7 +427,14 @@ int main (int argc,char **argv ) | |||||||
|    * Setup QDP |    * Setup QDP | ||||||
|    *********************************************************/ |    *********************************************************/ | ||||||
|   Chroma::initialize(&argc,&argv); |   Chroma::initialize(&argc,&argv); | ||||||
|   Chroma::WilsonTypeFermActs4DEnv::registerAll();  |   //  Chroma::WilsonTypeFermActs4DEnv::registerAll();  | ||||||
|  |   Chroma::WilsonTypeFermActsEnv::registerAll();  | ||||||
|  |   //bool linkageHack(void) | ||||||
|  |   //{ | ||||||
|  |   //  bool foo = true; | ||||||
|  |   // Inline Measurements | ||||||
|  |   //  InlineAggregateEnv::registerAll(); | ||||||
|  |   //  GaugeInitEnv::registerAll(); | ||||||
|  |  | ||||||
|   /******************************************************** |   /******************************************************** | ||||||
|    * Setup Grid |    * Setup Grid | ||||||
| @@ -388,26 +444,34 @@ int main (int argc,char **argv ) | |||||||
|                                                                        Grid::GridDefaultSimd(Grid::Nd,Grid::vComplex::Nsimd()), |                                                                        Grid::GridDefaultSimd(Grid::Nd,Grid::vComplex::Nsimd()), | ||||||
|                                                                        Grid::GridDefaultMpi()); |                                                                        Grid::GridDefaultMpi()); | ||||||
|    |    | ||||||
|   std::vector<int> gd = UGrid->GlobalDimensions(); |   Grid::Coordinate gd = UGrid->GlobalDimensions(); | ||||||
|   QDP::multi1d<int> nrow(QDP::Nd); |   QDP::multi1d<int> nrow(QDP::Nd); | ||||||
|   for(int mu=0;mu<4;mu++) nrow[mu] = gd[mu]; |   for(int mu=0;mu<4;mu++) nrow[mu] = gd[mu]; | ||||||
|  |  | ||||||
|   QDP::Layout::setLattSize(nrow); |   QDP::Layout::setLattSize(nrow); | ||||||
|   QDP::Layout::create(); |   QDP::Layout::create(); | ||||||
|  |  | ||||||
|   Grid::GridCartesian         * FGrid   = Grid::SpaceTimeGrid::makeFiveDimGrid(Ls,UGrid); |  | ||||||
|   Grid::LatticeGaugeField lat(UGrid); |  | ||||||
|   Grid::LatticeFermion    src(FGrid); |  | ||||||
|   Grid::LatticeFermion    res_chroma(FGrid); |  | ||||||
|   Grid::LatticeFermion    res_grid  (FGrid); |  | ||||||
|    |  | ||||||
|   std::vector<ChromaAction> ActionList({ |   std::vector<ChromaAction> ActionList({ | ||||||
| 		 HtCayleyTanh, // Plain old DWF. | 		 HtCayleyTanh, // Plain old DWF. | ||||||
| 		 HmCayleyTanh, | 		 HmCayleyTanh, | ||||||
| 		 HwCayleyTanh, | 		 HwCayleyTanh, | ||||||
| 		 HtCayleyZolo, // Plain old DWF. | 		 HtCayleyZolo, // Plain old DWF. | ||||||
| 		 HmCayleyZolo, | 		 HmCayleyZolo, | ||||||
| 		 HwCayleyZolo | 		 HwCayleyZolo, | ||||||
|  | 		 HwPartFracZolo, | ||||||
|  | 		 HwContFracZolo, | ||||||
|  | 		 HwContFracTanh | ||||||
|  |   }); | ||||||
|  |   std::vector<int> LsList({ | ||||||
|  |       8,//HtCayleyTanh, // Plain old DWF. | ||||||
|  |       8,//HmCayleyTanh, | ||||||
|  |       8,//HwCayleyTanh, | ||||||
|  |       8,//HtCayleyZolo, // Plain old DWF. | ||||||
|  |       8,//HmCayleyZolo, | ||||||
|  |       8,//HwCayleyZolo, | ||||||
|  |       9,//HwPartFracZolo | ||||||
|  |       9, //HwContFracZolo | ||||||
|  |       9 //HwContFracTanh | ||||||
|   }); |   }); | ||||||
|   std::vector<std::string> ActionName({ |   std::vector<std::string> ActionName({ | ||||||
|         "HtCayleyTanh", |         "HtCayleyTanh", | ||||||
| @@ -415,10 +479,19 @@ int main (int argc,char **argv ) | |||||||
| 	"HwCayleyTanh", | 	"HwCayleyTanh", | ||||||
| 	"HtCayleyZolo", | 	"HtCayleyZolo", | ||||||
| 	"HmCayleyZolo", | 	"HmCayleyZolo", | ||||||
|         "HwCayleyZolo" |         "HwCayleyZolo", | ||||||
|  | 	"HwPartFracZolo", | ||||||
|  | 	"HwContFracZolo", | ||||||
|  | 	"HwContFracTanh" | ||||||
|   }); |   }); | ||||||
|  |  | ||||||
|   for(int i=0;i<ActionList.size();i++) { |   for(int i=0;i<ActionList.size();i++) { | ||||||
|  |     Ls = LsList[i]; | ||||||
|  |     Grid::GridCartesian      * FGrid   = Grid::SpaceTimeGrid::makeFiveDimGrid(Ls,UGrid); | ||||||
|  |     Grid::LatticeGaugeField lat(UGrid); | ||||||
|  |     Grid::LatticeFermion    src(FGrid); | ||||||
|  |     Grid::LatticeFermion    res_chroma(FGrid); | ||||||
|  |     Grid::LatticeFermion    res_grid  (FGrid); | ||||||
|     std::cout << "*****************************"<<std::endl; |     std::cout << "*****************************"<<std::endl; | ||||||
|     std::cout << "Action "<<ActionName[i]<<std::endl; |     std::cout << "Action "<<ActionName[i]<<std::endl; | ||||||
|     std::cout << "*****************************"<<std::endl; |     std::cout << "*****************************"<<std::endl; | ||||||
| @@ -439,6 +512,7 @@ int main (int argc,char **argv ) | |||||||
|        |        | ||||||
|       std::cout << "Norm of difference "<<Grid::norm2(res_chroma)<<std::endl; |       std::cout << "Norm of difference "<<Grid::norm2(res_chroma)<<std::endl; | ||||||
|     } |     } | ||||||
|  |     delete FGrid; | ||||||
|   } |   } | ||||||
|  |  | ||||||
|   std::cout << "Finished test "<<std::endl; |   std::cout << "Finished test "<<std::endl; | ||||||
| @@ -502,7 +576,7 @@ void calc_grid(ChromaAction action,Grid::LatticeGaugeField & Umu, Grid::LatticeF | |||||||
|   Grid::gaussian(RNG5,src); |   Grid::gaussian(RNG5,src); | ||||||
|   Grid::gaussian(RNG5,res); |   Grid::gaussian(RNG5,res); | ||||||
|  |  | ||||||
|   Grid::SU<Nc>::HotConfiguration(RNG4,Umu); |   Grid::SU<Grid::Nc>::HotConfiguration(RNG4,Umu); | ||||||
|  |  | ||||||
|   /* |   /* | ||||||
|   Grid::LatticeColourMatrix U(UGrid); |   Grid::LatticeColourMatrix U(UGrid); | ||||||
| @@ -519,7 +593,7 @@ void calc_grid(ChromaAction action,Grid::LatticeGaugeField & Umu, Grid::LatticeF | |||||||
|  |  | ||||||
|   if ( action == HtCayleyTanh ) {  |   if ( action == HtCayleyTanh ) {  | ||||||
|  |  | ||||||
|     Grid::DomainWallFermionR Ddwf(Umu,*FGrid,*FrbGrid,*UGrid,*UrbGrid,_mass,_M5); |     Grid::DomainWallFermionD Ddwf(Umu,*FGrid,*FrbGrid,*UGrid,*UrbGrid,_mass,_M5); | ||||||
|  |  | ||||||
|     std::cout << Grid::GridLogMessage <<" Calling domain wall multiply "<<std::endl; |     std::cout << Grid::GridLogMessage <<" Calling domain wall multiply "<<std::endl; | ||||||
|  |  | ||||||
| @@ -535,7 +609,7 @@ void calc_grid(ChromaAction action,Grid::LatticeGaugeField & Umu, Grid::LatticeF | |||||||
|  |  | ||||||
|     Grid::Real _b = 0.5*(mobius_scale +1.0); |     Grid::Real _b = 0.5*(mobius_scale +1.0); | ||||||
|     Grid::Real _c = 0.5*(mobius_scale -1.0); |     Grid::Real _c = 0.5*(mobius_scale -1.0); | ||||||
|     Grid::MobiusZolotarevFermionR D(Umu,*FGrid,*FrbGrid,*UGrid,*UrbGrid,_mass,_M5,_b,_c,zolo_lo,zolo_hi); |     Grid::MobiusZolotarevFermionD D(Umu,*FGrid,*FrbGrid,*UGrid,*UrbGrid,_mass,_M5,_b,_c,zolo_lo,zolo_hi); | ||||||
|  |  | ||||||
|     std::cout << Grid::GridLogMessage <<" Calling mobius zolo multiply "<<std::endl; |     std::cout << Grid::GridLogMessage <<" Calling mobius zolo multiply "<<std::endl; | ||||||
|  |  | ||||||
| @@ -549,7 +623,7 @@ void calc_grid(ChromaAction action,Grid::LatticeGaugeField & Umu, Grid::LatticeF | |||||||
|  |  | ||||||
|   if ( action == HtCayleyZolo ) { |   if ( action == HtCayleyZolo ) { | ||||||
|  |  | ||||||
|     Grid::ShamirZolotarevFermionR D(Umu,*FGrid,*FrbGrid,*UGrid,*UrbGrid,_mass,_M5,zolo_lo,zolo_hi); |     Grid::ShamirZolotarevFermionD D(Umu,*FGrid,*FrbGrid,*UGrid,*UrbGrid,_mass,_M5,zolo_lo,zolo_hi); | ||||||
|  |  | ||||||
|     std::cout << Grid::GridLogMessage <<" Calling shamir zolo multiply "<<std::endl; |     std::cout << Grid::GridLogMessage <<" Calling shamir zolo multiply "<<std::endl; | ||||||
|  |  | ||||||
| @@ -561,6 +635,60 @@ void calc_grid(ChromaAction action,Grid::LatticeGaugeField & Umu, Grid::LatticeF | |||||||
|     return; |     return; | ||||||
|   } |   } | ||||||
|  |  | ||||||
|  |   if ( action == HwPartFracTanh ) { | ||||||
|  |  | ||||||
|  |     Grid::OverlapWilsonPartialFractionTanhFermionD Dov(Umu,*FGrid,*FrbGrid,*UGrid,*UrbGrid,_mass,_M5,1.0); | ||||||
|  |  | ||||||
|  |     std::cout << Grid::GridLogMessage <<" Calling part frac tanh multiply "<<std::endl; | ||||||
|  |  | ||||||
|  |     if ( dag )  | ||||||
|  |       Dov.Mdag(src,res);   | ||||||
|  |     else  | ||||||
|  |       Dov.M(src,res);   | ||||||
|  |  | ||||||
|  |     return; | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   if ( action == HwContFracTanh ) { | ||||||
|  |  | ||||||
|  |     Grid::OverlapWilsonContFracTanhFermionD Dov(Umu,*FGrid,*FrbGrid,*UGrid,*UrbGrid,_mass,_M5,1.0); | ||||||
|  |  | ||||||
|  |     std::cout << Grid::GridLogMessage <<" Calling cont frac tanh multiply "<<std::endl; | ||||||
|  |  | ||||||
|  |     if ( dag )  | ||||||
|  |       Dov.Mdag(src,res);   | ||||||
|  |     else  | ||||||
|  |       Dov.M(src,res);   | ||||||
|  |  | ||||||
|  |     return; | ||||||
|  |   } | ||||||
|  |   if ( action == HwContFracZolo ) { | ||||||
|  |  | ||||||
|  |     Grid::OverlapWilsonContFracZolotarevFermionD Dov(Umu,*FGrid,*FrbGrid,*UGrid,*UrbGrid,_mass,_M5,zolo_lo,zolo_hi); | ||||||
|  |  | ||||||
|  |     std::cout << Grid::GridLogMessage <<" Calling cont frac zolo multiply "<<std::endl; | ||||||
|  |  | ||||||
|  |     if ( dag )  | ||||||
|  |       Dov.Mdag(src,res);   | ||||||
|  |     else  | ||||||
|  |       Dov.M(src,res);   | ||||||
|  |  | ||||||
|  |     return; | ||||||
|  |   } | ||||||
|  |  | ||||||
|  |   if ( action == HwPartFracZolo ) { | ||||||
|  |  | ||||||
|  |     Grid::OverlapWilsonPartialFractionZolotarevFermionD Dov(Umu,*FGrid,*FrbGrid,*UGrid,*UrbGrid,_mass,_M5,zolo_lo,zolo_hi); | ||||||
|  |     std::cout << Grid::GridLogMessage <<" Calling part frac zolotarev multiply "<<std::endl; | ||||||
|  |  | ||||||
|  |     if ( dag )  | ||||||
|  |       Dov.Mdag(src,res);   | ||||||
|  |     else  | ||||||
|  |       Dov.M(src,res);   | ||||||
|  |  | ||||||
|  |     return; | ||||||
|  |   } | ||||||
|  |    | ||||||
|   /* |   /* | ||||||
|   if ( action == HmCayleyTanh ) { |   if ( action == HmCayleyTanh ) { | ||||||
|     Grid::Real _b = 0.5*(mobius_scale +1.0); |     Grid::Real _b = 0.5*(mobius_scale +1.0); | ||||||
| @@ -581,7 +709,7 @@ void calc_grid(ChromaAction action,Grid::LatticeGaugeField & Umu, Grid::LatticeF | |||||||
|  |  | ||||||
|   if ( action == HmCayleyTanh ) { |   if ( action == HmCayleyTanh ) { | ||||||
|  |  | ||||||
|     Grid::ScaledShamirFermionR D(Umu,*FGrid,*FrbGrid,*UGrid,*UrbGrid,_mass,_M5,mobius_scale); |     Grid::ScaledShamirFermionD D(Umu,*FGrid,*FrbGrid,*UGrid,*UrbGrid,_mass,_M5,mobius_scale); | ||||||
|  |  | ||||||
|     std::cout << Grid::GridLogMessage <<" Calling scaled shamir multiply "<<std::endl; |     std::cout << Grid::GridLogMessage <<" Calling scaled shamir multiply "<<std::endl; | ||||||
|  |  | ||||||
| @@ -595,7 +723,7 @@ void calc_grid(ChromaAction action,Grid::LatticeGaugeField & Umu, Grid::LatticeF | |||||||
|  |  | ||||||
|   if ( action == HwCayleyTanh ) { |   if ( action == HwCayleyTanh ) { | ||||||
|  |  | ||||||
|     Grid::OverlapWilsonCayleyTanhFermionR D(Umu,*FGrid,*FrbGrid,*UGrid,*UrbGrid,_mass,_M5,1.0); |     Grid::OverlapWilsonCayleyTanhFermionD D(Umu,*FGrid,*FrbGrid,*UGrid,*UrbGrid,_mass,_M5,1.0); | ||||||
|  |  | ||||||
|     if ( dag )  |     if ( dag )  | ||||||
|       D.Mdag(src,res);   |       D.Mdag(src,res);   | ||||||
| @@ -607,7 +735,7 @@ void calc_grid(ChromaAction action,Grid::LatticeGaugeField & Umu, Grid::LatticeF | |||||||
|  |  | ||||||
|   if ( action == HwCayleyZolo ) { |   if ( action == HwCayleyZolo ) { | ||||||
|  |  | ||||||
|     Grid::OverlapWilsonCayleyZolotarevFermionR D(Umu,*FGrid,*FrbGrid,*UGrid,*UrbGrid,_mass,_M5,zolo_lo,zolo_hi); |     Grid::OverlapWilsonCayleyZolotarevFermionD D(Umu,*FGrid,*FrbGrid,*UGrid,*UrbGrid,_mass,_M5,zolo_lo,zolo_hi); | ||||||
|  |  | ||||||
|     if ( dag )  |     if ( dag )  | ||||||
|       D.Mdag(src,res);   |       D.Mdag(src,res);   | ||||||
|   | |||||||
| @@ -1,4 +1,4 @@ | |||||||
| /************************************************************************************* | ************************************************************************************* | ||||||
|  |  | ||||||
| Grid physics library, www.github.com/paboyle/Grid | Grid physics library, www.github.com/paboyle/Grid | ||||||
|  |  | ||||||
| @@ -67,7 +67,13 @@ int main(int argc, char** argv) { | |||||||
|   result = Zero(); |   result = Zero(); | ||||||
|   LatticeGaugeField Umu(UGrid); |   LatticeGaugeField Umu(UGrid); | ||||||
|  |  | ||||||
|  | #if 0 | ||||||
|  |   FieldMetaData header; | ||||||
|  |   std::string file("ckpoint_lat.4000"); | ||||||
|  |   NerscIO::readConfiguration(Umu,header,file); | ||||||
|  | #else   | ||||||
|   SU<Nc>::HotConfiguration(RNG4, Umu); |   SU<Nc>::HotConfiguration(RNG4, Umu); | ||||||
|  | #endif | ||||||
|  |  | ||||||
|   std::cout << GridLogMessage << "Lattice dimensions: " << GridDefaultLatt() |   std::cout << GridLogMessage << "Lattice dimensions: " << GridDefaultLatt() | ||||||
|             << "   Ls: " << Ls << std::endl; |             << "   Ls: " << Ls << std::endl; | ||||||
|   | |||||||
| @@ -54,14 +54,29 @@ int main (int argc, char ** argv) | |||||||
|   GridCartesian         * FGrid   = SpaceTimeGrid::makeFiveDimGrid(Ls,UGrid); |   GridCartesian         * FGrid   = SpaceTimeGrid::makeFiveDimGrid(Ls,UGrid); | ||||||
|   GridRedBlackCartesian * FrbGrid = SpaceTimeGrid::makeFiveDimRedBlackGrid(Ls,UGrid); |   GridRedBlackCartesian * FrbGrid = SpaceTimeGrid::makeFiveDimRedBlackGrid(Ls,UGrid); | ||||||
|  |  | ||||||
|  |   std::vector<ComplexD> qmu; | ||||||
|  |   qmu.push_back(ComplexD(0.1,0.0)); | ||||||
|  |   qmu.push_back(ComplexD(0.0,0.0)); | ||||||
|  |   qmu.push_back(ComplexD(0.0,0.0)); | ||||||
|  |   qmu.push_back(ComplexD(0.0,0.01)); | ||||||
|  |    | ||||||
|  |  | ||||||
|   std::vector<int> seeds4({1,2,3,4}); |   std::vector<int> seeds4({1,2,3,4}); | ||||||
|   std::vector<int> seeds5({5,6,7,8}); |   std::vector<int> seeds5({5,6,7,8}); | ||||||
|   GridParallelRNG          RNG5(FGrid);  RNG5.SeedFixedIntegers(seeds5); |   GridParallelRNG          RNG5(FGrid);  RNG5.SeedFixedIntegers(seeds5); | ||||||
|   GridParallelRNG          RNG4(UGrid);  RNG4.SeedFixedIntegers(seeds4); |   GridParallelRNG          RNG4(UGrid);  RNG4.SeedFixedIntegers(seeds4); | ||||||
|  |  | ||||||
|  |   LatticeFermion    tmp(FGrid); | ||||||
|   LatticeFermion    src(FGrid); random(RNG5,src); |   LatticeFermion    src(FGrid); random(RNG5,src); | ||||||
|   LatticeFermion result(FGrid); result=Zero(); |   LatticeFermion result(FGrid); result=Zero(); | ||||||
|   LatticeGaugeField Umu(UGrid); SU<Nc>::HotConfiguration(RNG4,Umu); |   LatticeGaugeField Umu(UGrid);  | ||||||
|  | #if 0 | ||||||
|  |   FieldMetaData header; | ||||||
|  |   std::string file("ckpoint_lat.4000"); | ||||||
|  |   NerscIO::readConfiguration(Umu,header,file); | ||||||
|  | #else   | ||||||
|  |   SU<Nc>::HotConfiguration(RNG4,Umu); | ||||||
|  | #endif | ||||||
|    |    | ||||||
|   std::vector<LatticeColourMatrix> U(4,UGrid); |   std::vector<LatticeColourMatrix> U(4,UGrid); | ||||||
|   for(int mu=0;mu<Nd;mu++){ |   for(int mu=0;mu<Nd;mu++){ | ||||||
| @@ -71,8 +86,15 @@ int main (int argc, char ** argv) | |||||||
|   RealD mass=0.1; |   RealD mass=0.1; | ||||||
|   RealD M5=1.8; |   RealD M5=1.8; | ||||||
|   DomainWallFermionD Ddwf(Umu,*FGrid,*FrbGrid,*UGrid,*UrbGrid,mass,M5); |   DomainWallFermionD Ddwf(Umu,*FGrid,*FrbGrid,*UGrid,*UrbGrid,mass,M5); | ||||||
|  |   Ddwf.qmu = qmu; | ||||||
|  |  | ||||||
|  |   Ddwf.M(src,tmp); | ||||||
|  |   std::cout << " |M src|^2 "<<norm2(tmp)<<std::endl; | ||||||
|   MdagMLinearOperator<DomainWallFermionD,LatticeFermion> HermOp(Ddwf); |   MdagMLinearOperator<DomainWallFermionD,LatticeFermion> HermOp(Ddwf); | ||||||
|  |   HermOp.HermOp(src,tmp); | ||||||
|  |  | ||||||
|  |   std::cout << " <src|MdagM| src> "<<innerProduct(src,tmp)<<std::endl; | ||||||
|  |    | ||||||
|   ConjugateGradient<LatticeFermion> CG(1.0e-6,10000); |   ConjugateGradient<LatticeFermion> CG(1.0e-6,10000); | ||||||
|   CG(HermOp,src,result); |   CG(HermOp,src,result); | ||||||
|  |  | ||||||
|   | |||||||
		Reference in New Issue
	
	Block a user