mirror of
				https://github.com/paboyle/Grid.git
				synced 2025-10-31 03:54:33 +00:00 
			
		
		
		
	Compare commits
	
		
			1 Commits
		
	
	
		
			feature/cp
			...
			feature/ad
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
|  | b284d50863 | 
							
								
								
									
										1
									
								
								.gitignore
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										1
									
								
								.gitignore
									
									
									
									
										vendored
									
									
								
							| @@ -88,7 +88,6 @@ Thumbs.db | ||||
| # build directory # | ||||
| ################### | ||||
| build*/* | ||||
| Documentation/_build | ||||
|  | ||||
| # IDE related files # | ||||
| ##################### | ||||
|   | ||||
| @@ -358,7 +358,7 @@ public: | ||||
|     autoView( in_v , in, AcceleratorRead); | ||||
|     autoView( out_v , out, AcceleratorWrite); | ||||
|     autoView( Stencil_v  , Stencil, AcceleratorRead); | ||||
|     int npoint = geom.npoint; | ||||
|     auto& geom_v = geom; | ||||
|     typedef LatticeView<Cobj> Aview; | ||||
|        | ||||
|     Vector<Aview> AcceleratorViewContainer; | ||||
| @@ -380,7 +380,7 @@ public: | ||||
|       int ptype; | ||||
|       StencilEntry *SE; | ||||
|  | ||||
|       for(int point=0;point<npoint;point++){ | ||||
|       for(int point=0;point<geom_v.npoint;point++){ | ||||
|  | ||||
| 	SE=Stencil_v.GetEntry(ptype,point,ss); | ||||
| 	   | ||||
| @@ -424,7 +424,7 @@ public: | ||||
|     autoView( in_v , in, AcceleratorRead); | ||||
|     autoView( out_v , out, AcceleratorWrite); | ||||
|     autoView( Stencil_v  , Stencil, AcceleratorRead); | ||||
|     int npoint = geom.npoint; | ||||
|     auto& geom_v = geom; | ||||
|     typedef LatticeView<Cobj> Aview; | ||||
|  | ||||
|     Vector<Aview> AcceleratorViewContainer; | ||||
| @@ -442,8 +442,6 @@ public: | ||||
|     for(int p=0; p<geom.npoint; p++) | ||||
|       points[p] = geom.points_dagger[p]; | ||||
|  | ||||
|     auto points_p = &points[0]; | ||||
|  | ||||
|     RealD* dag_factor_p = &dag_factor[0]; | ||||
|  | ||||
|     accelerator_for(sss, Grid()->oSites()*nbasis, Nsimd, { | ||||
| @@ -454,8 +452,8 @@ public: | ||||
|       int ptype; | ||||
|       StencilEntry *SE; | ||||
|  | ||||
|       for(int p=0;p<npoint;p++){ | ||||
|         int point = points_p[p]; | ||||
|       for(int p=0;p<geom_v.npoint;p++){ | ||||
|         int point = points[p]; | ||||
|  | ||||
| 	SE=Stencil_v.GetEntry(ptype,point,ss); | ||||
|  | ||||
| @@ -710,8 +708,6 @@ public: | ||||
|     for(int p=0; p<npoint; p++) | ||||
|       points[p] = (dag && !hermitian) ? geom.points_dagger[p] : p; | ||||
|  | ||||
|     auto points_p = &points[0]; | ||||
|  | ||||
|     Vector<Aview> AcceleratorViewContainer; | ||||
|     for(int p=0;p<npoint;p++) AcceleratorViewContainer.push_back(a[p].View(AcceleratorRead)); | ||||
|     Aview *Aview_p = & AcceleratorViewContainer[0]; | ||||
| @@ -732,7 +728,7 @@ public: | ||||
|         StencilEntry *SE; | ||||
|  | ||||
|         for(int p=0;p<npoint;p++){ | ||||
|           int point = points_p[p]; | ||||
|           int point = points[p]; | ||||
|           SE=st_v.GetEntry(ptype,point,ss); | ||||
|  | ||||
|           if(SE->_is_local) { | ||||
| @@ -758,7 +754,7 @@ public: | ||||
|         StencilEntry *SE; | ||||
|  | ||||
|         for(int p=0;p<npoint;p++){ | ||||
|           int point = points_p[p]; | ||||
|           int point = points[p]; | ||||
|           SE=st_v.GetEntry(ptype,point,ss); | ||||
|  | ||||
|           if(SE->_is_local) { | ||||
|   | ||||
| @@ -136,7 +136,7 @@ public: | ||||
|     flops=0; | ||||
|     usec =0; | ||||
|     Coordinate layout(Nd,1); | ||||
|     sgrid = new GridCartesian(dimensions,layout,processors,*grid); | ||||
|     sgrid = new GridCartesian(dimensions,layout,processors); | ||||
|   }; | ||||
|      | ||||
|   ~FFT ( void)  { | ||||
| @@ -182,7 +182,7 @@ public: | ||||
|     pencil_gd[dim] = G*processors[dim]; | ||||
|        | ||||
|     // Pencil global vol LxLxGxLxL per node | ||||
|     GridCartesian pencil_g(pencil_gd,layout,processors,*vgrid); | ||||
|     GridCartesian pencil_g(pencil_gd,layout,processors); | ||||
|        | ||||
|     // Construct pencils | ||||
|     typedef typename vobj::scalar_object sobj; | ||||
|   | ||||
| @@ -52,7 +52,6 @@ public: | ||||
|   virtual void AdjOp  (const Field &in, Field &out) = 0; // Abstract base | ||||
|   virtual void HermOpAndNorm(const Field &in, Field &out,RealD &n1,RealD &n2)=0; | ||||
|   virtual void HermOp(const Field &in, Field &out)=0; | ||||
|   virtual ~LinearOperatorBase(){}; | ||||
| }; | ||||
|  | ||||
|  | ||||
| @@ -508,7 +507,7 @@ class SchurStaggeredOperator :  public SchurOperatorBase<Field> { | ||||
|   virtual  void MpcDag   (const Field &in, Field &out){ | ||||
|     Mpc(in,out); | ||||
|   } | ||||
|   virtual void MpcDagMpc(const Field &in, Field &out) { | ||||
|   virtual void MpcDagMpc(const Field &in, Field &out,RealD &ni,RealD &no) { | ||||
|     assert(0);// Never need with staggered | ||||
|   } | ||||
| }; | ||||
| @@ -531,16 +530,6 @@ public: | ||||
| template<class Field> class LinearFunction { | ||||
| public: | ||||
|   virtual void operator() (const Field &in, Field &out) = 0; | ||||
|  | ||||
|   virtual void operator() (const std::vector<Field> &in, std::vector<Field> &out) | ||||
|   { | ||||
|     assert(in.size() == out.size()); | ||||
|  | ||||
|     for (unsigned int i = 0; i < in.size(); ++i) | ||||
|     { | ||||
|       (*this)(in[i], out[i]); | ||||
|     } | ||||
|   } | ||||
| }; | ||||
|  | ||||
| template<class Field> class IdentityLinearFunction : public LinearFunction<Field> { | ||||
| @@ -586,7 +575,6 @@ class HermOpOperatorFunction : public OperatorFunction<Field> { | ||||
| template<typename Field> | ||||
| class PlainHermOp : public LinearFunction<Field> { | ||||
| public: | ||||
|   using LinearFunction<Field>::operator(); | ||||
|   LinearOperatorBase<Field> &_Linop; | ||||
|        | ||||
|   PlainHermOp(LinearOperatorBase<Field>& linop) : _Linop(linop)  | ||||
| @@ -600,7 +588,6 @@ public: | ||||
| template<typename Field> | ||||
| class FunctionHermOp : public LinearFunction<Field> { | ||||
| public: | ||||
|   using LinearFunction<Field>::operator();  | ||||
|   OperatorFunction<Field>   & _poly; | ||||
|   LinearOperatorBase<Field> &_Linop; | ||||
|        | ||||
|   | ||||
| @@ -30,19 +30,13 @@ Author: Azusa Yamaguchi <ayamaguc@staffmail.ed.ac.uk> | ||||
|  | ||||
| NAMESPACE_BEGIN(Grid); | ||||
|  | ||||
| template<class Field> using Preconditioner =  LinearFunction<Field> ; | ||||
|  | ||||
| /* | ||||
| template<class Field> class Preconditioner :  public LinearFunction<Field> { | ||||
|   using LinearFunction<Field>::operator(); | ||||
| template<class Field> class Preconditioner :  public LinearFunction<Field> {  | ||||
|   virtual void operator()(const Field &src, Field & psi)=0; | ||||
| }; | ||||
| */ | ||||
|  | ||||
| template<class Field> class TrivialPrecon :  public Preconditioner<Field> {  | ||||
| public: | ||||
|   using Preconditioner<Field>::operator(); | ||||
|   virtual void operator()(const Field &src, Field & psi){ | ||||
|   void operator()(const Field &src, Field & psi){ | ||||
|     psi = src; | ||||
|   } | ||||
|   TrivialPrecon(void){}; | ||||
|   | ||||
| @@ -48,7 +48,6 @@ public: | ||||
|   virtual  void Mdiag    (const Field &in, Field &out)=0; | ||||
|   virtual  void Mdir     (const Field &in, Field &out,int dir, int disp)=0; | ||||
|   virtual  void MdirAll  (const Field &in, std::vector<Field> &out)=0; | ||||
|   virtual ~SparseMatrixBase() {}; | ||||
| }; | ||||
|  | ||||
| ///////////////////////////////////////////////////////////////////////////////////////////// | ||||
| @@ -73,7 +72,7 @@ public: | ||||
|   virtual  void MeooeDag    (const Field &in, Field &out)=0; | ||||
|   virtual  void MooeeDag    (const Field &in, Field &out)=0; | ||||
|   virtual  void MooeeInvDag (const Field &in, Field &out)=0; | ||||
|   virtual ~CheckerBoardedSparseMatrixBase() {}; | ||||
|  | ||||
| }; | ||||
|  | ||||
| NAMESPACE_END(Grid); | ||||
|   | ||||
| @@ -36,8 +36,7 @@ NAMESPACE_BEGIN(Grid); | ||||
| template<class FieldD, class FieldF, typename std::enable_if< getPrecision<FieldD>::value == 2, int>::type = 0, typename std::enable_if< getPrecision<FieldF>::value == 1, int>::type = 0>  | ||||
| class MixedPrecisionBiCGSTAB : public LinearFunction<FieldD>  | ||||
| { | ||||
|   public: | ||||
|     using LinearFunction<FieldD>::operator(); | ||||
|   public:                                                 | ||||
|     RealD   Tolerance; | ||||
|     RealD   InnerTolerance; // Initial tolerance for inner CG. Defaults to Tolerance but can be changed | ||||
|     Integer MaxInnerIterations; | ||||
|   | ||||
| @@ -35,8 +35,7 @@ NAMESPACE_BEGIN(Grid); | ||||
|     typename std::enable_if< getPrecision<FieldD>::value == 2, int>::type = 0, | ||||
|     typename std::enable_if< getPrecision<FieldF>::value == 1, int>::type = 0>  | ||||
|   class MixedPrecisionConjugateGradient : public LinearFunction<FieldD> { | ||||
|   public: | ||||
|     using LinearFunction<FieldD>::operator(); | ||||
|   public:                                                 | ||||
|     RealD   Tolerance; | ||||
|     RealD   InnerTolerance; //Initial tolerance for inner CG. Defaults to Tolerance but can be changed | ||||
|     Integer MaxInnerIterations; | ||||
|   | ||||
| @@ -33,19 +33,16 @@ namespace Grid { | ||||
| template<class Field> | ||||
| class ZeroGuesser: public LinearFunction<Field> { | ||||
| public: | ||||
|   using LinearFunction<Field>::operator(); | ||||
|     virtual void operator()(const Field &src, Field &guess) { guess = Zero(); }; | ||||
| }; | ||||
| template<class Field> | ||||
| class DoNothingGuesser: public LinearFunction<Field> { | ||||
| public: | ||||
|   using LinearFunction<Field>::operator(); | ||||
|   virtual void operator()(const Field &src, Field &guess) {  }; | ||||
| }; | ||||
| template<class Field> | ||||
| class SourceGuesser: public LinearFunction<Field> { | ||||
| public: | ||||
|   using LinearFunction<Field>::operator(); | ||||
|   virtual void operator()(const Field &src, Field &guess) { guess = src; }; | ||||
| }; | ||||
|  | ||||
| @@ -57,24 +54,15 @@ class DeflatedGuesser: public LinearFunction<Field> { | ||||
| private: | ||||
|   const std::vector<Field> &evec; | ||||
|   const std::vector<RealD> &eval; | ||||
|   const unsigned int       N; | ||||
|  | ||||
| public: | ||||
|   using LinearFunction<Field>::operator(); | ||||
|  | ||||
|   DeflatedGuesser(const std::vector<Field> & _evec,const std::vector<RealD> & _eval) | ||||
|   : DeflatedGuesser(_evec, _eval, _evec.size()) | ||||
|   {} | ||||
|  | ||||
|   DeflatedGuesser(const std::vector<Field> & _evec, const std::vector<RealD> & _eval, const unsigned int _N) | ||||
|   : evec(_evec), eval(_eval), N(_N) | ||||
|   { | ||||
|     assert(evec.size()==eval.size()); | ||||
|     assert(N <= evec.size()); | ||||
|   }  | ||||
|   DeflatedGuesser(const std::vector<Field> & _evec,const std::vector<RealD> & _eval) : evec(_evec), eval(_eval) {}; | ||||
|  | ||||
|   virtual void operator()(const Field &src,Field &guess) { | ||||
|     guess = Zero(); | ||||
|     assert(evec.size()==eval.size()); | ||||
|     auto N = evec.size(); | ||||
|     for (int i=0;i<N;i++) { | ||||
|       const Field& tmp = evec[i]; | ||||
|       axpy(guess,TensorRemove(innerProduct(tmp,src)) / eval[i],tmp,guess); | ||||
| @@ -91,7 +79,6 @@ private: | ||||
|   const std::vector<RealD>       &eval_coarse; | ||||
| public: | ||||
|    | ||||
|   using LinearFunction<FineField>::operator(); | ||||
|   LocalCoherenceDeflatedGuesser(const std::vector<FineField>   &_subspace, | ||||
| 				const std::vector<CoarseField> &_evec_coarse, | ||||
| 				const std::vector<RealD>       &_eval_coarse) | ||||
|   | ||||
| @@ -67,7 +67,6 @@ public: | ||||
| template<class Fobj,class CComplex,int nbasis> | ||||
| class ProjectedHermOp : public LinearFunction<Lattice<iVector<CComplex,nbasis > > > { | ||||
| public: | ||||
|   using LinearFunction<Lattice<iVector<CComplex,nbasis > > >::operator(); | ||||
|   typedef iVector<CComplex,nbasis >           CoarseSiteVector; | ||||
|   typedef Lattice<CoarseSiteVector>           CoarseField; | ||||
|   typedef Lattice<CComplex>   CoarseScalar; // used for inner products on fine field | ||||
| @@ -98,7 +97,6 @@ public: | ||||
| template<class Fobj,class CComplex,int nbasis> | ||||
| class ProjectedFunctionHermOp : public LinearFunction<Lattice<iVector<CComplex,nbasis > > > { | ||||
| public: | ||||
|   using LinearFunction<Lattice<iVector<CComplex,nbasis > > >::operator(); | ||||
|   typedef iVector<CComplex,nbasis >           CoarseSiteVector; | ||||
|   typedef Lattice<CoarseSiteVector>           CoarseField; | ||||
|   typedef Lattice<CComplex>   CoarseScalar; // used for inner products on fine field | ||||
|   | ||||
| @@ -43,7 +43,7 @@ NAMESPACE_BEGIN(Grid); | ||||
| template<class Field> | ||||
| class PrecGeneralisedConjugateResidual : public LinearFunction<Field> { | ||||
| public:                                                 | ||||
|   using LinearFunction<Field>::operator(); | ||||
|  | ||||
|   RealD   Tolerance; | ||||
|   Integer MaxIterations; | ||||
|   int verbose; | ||||
|   | ||||
| @@ -43,7 +43,7 @@ NAMESPACE_BEGIN(Grid); | ||||
| template<class Field> | ||||
| class PrecGeneralisedConjugateResidualNonHermitian : public LinearFunction<Field> { | ||||
| public:                                                 | ||||
|   using LinearFunction<Field>::operator(); | ||||
|  | ||||
|   RealD   Tolerance; | ||||
|   Integer MaxIterations; | ||||
|   int verbose; | ||||
| @@ -119,8 +119,7 @@ public: | ||||
|   RealD GCRnStep(const Field &src, Field &psi,RealD rsq){ | ||||
|  | ||||
|     RealD cp; | ||||
|     ComplexD a, b; | ||||
|     //    ComplexD zAz; | ||||
|     ComplexD a, b, zAz; | ||||
|     RealD zAAz; | ||||
|     ComplexD rq; | ||||
|  | ||||
| @@ -147,7 +146,7 @@ public: | ||||
|     ////////////////////////////////// | ||||
|     MatTimer.Start(); | ||||
|     Linop.Op(psi,Az); | ||||
|     //    zAz = innerProduct(Az,psi); | ||||
|     zAz = innerProduct(Az,psi); | ||||
|     zAAz= norm2(Az); | ||||
|     MatTimer.Stop(); | ||||
|      | ||||
| @@ -171,7 +170,7 @@ public: | ||||
|  | ||||
|     LinalgTimer.Start(); | ||||
|  | ||||
|     //    zAz = innerProduct(Az,psi); | ||||
|     zAz = innerProduct(Az,psi); | ||||
|     zAAz= norm2(Az); | ||||
|  | ||||
|     //p[0],q[0],qq[0]  | ||||
| @@ -213,7 +212,7 @@ public: | ||||
|       MatTimer.Start(); | ||||
|       Linop.Op(z,Az); | ||||
|       MatTimer.Stop(); | ||||
|       //      zAz = innerProduct(Az,psi); | ||||
|       zAz = innerProduct(Az,psi); | ||||
|       zAAz= norm2(Az); | ||||
|  | ||||
|       LinalgTimer.Start(); | ||||
|   | ||||
| @@ -132,31 +132,6 @@ namespace Grid { | ||||
|       (*this)(_Matrix,in,out,guess); | ||||
|     } | ||||
|  | ||||
|     void RedBlackSource(Matrix &_Matrix, const std::vector<Field> &in, std::vector<Field> &src_o)  | ||||
|     { | ||||
|       GridBase *grid = _Matrix.RedBlackGrid(); | ||||
|       Field tmp(grid); | ||||
|       int nblock = in.size(); | ||||
|       for(int b=0;b<nblock;b++){ | ||||
| 	RedBlackSource(_Matrix,in[b],tmp,src_o[b]); | ||||
|       } | ||||
|     } | ||||
|     // James can write his own deflated guesser | ||||
|     // with optimised code for the inner products | ||||
|     //    RedBlackSolveSplitGrid(); | ||||
|     //    RedBlackSolve(_Matrix,src_o,sol_o);  | ||||
|  | ||||
|     void RedBlackSolution(Matrix &_Matrix, const std::vector<Field> &in, const std::vector<Field> &sol_o, std::vector<Field> &out) | ||||
|     { | ||||
|       GridBase *grid = _Matrix.RedBlackGrid(); | ||||
|       Field tmp(grid); | ||||
|       int nblock = in.size(); | ||||
|       for(int b=0;b<nblock;b++) { | ||||
| 	pickCheckerboard(Even,tmp,in[b]); | ||||
| 	RedBlackSolution(_Matrix,sol_o[b],tmp,out[b]); | ||||
|       } | ||||
|     } | ||||
|  | ||||
|     template<class Guesser> | ||||
|     void operator()(Matrix &_Matrix, const std::vector<Field> &in, std::vector<Field> &out,Guesser &guess)  | ||||
|     { | ||||
| @@ -175,29 +150,24 @@ namespace Grid { | ||||
|       //////////////////////////////////////////////// | ||||
|       // Prepare RedBlack source | ||||
|       //////////////////////////////////////////////// | ||||
|       RedBlackSource(_Matrix,in,src_o); | ||||
| 	//      for(int b=0;b<nblock;b++){ | ||||
| 	//	RedBlackSource(_Matrix,in[b],tmp,src_o[b]); | ||||
| 	//      } | ||||
|        | ||||
|       for(int b=0;b<nblock;b++){ | ||||
| 	RedBlackSource(_Matrix,in[b],tmp,src_o[b]); | ||||
|       } | ||||
|       //////////////////////////////////////////////// | ||||
|       // Make the guesses | ||||
|       //////////////////////////////////////////////// | ||||
|       if ( subGuess ) guess_save.resize(nblock,grid); | ||||
|  | ||||
|        | ||||
|       if(useSolnAsInitGuess) { | ||||
|         for(int b=0;b<nblock;b++){ | ||||
|       for(int b=0;b<nblock;b++){ | ||||
|         if(useSolnAsInitGuess) { | ||||
|           pickCheckerboard(Odd, sol_o[b], out[b]); | ||||
|         } else { | ||||
|           guess(src_o[b],sol_o[b]);  | ||||
|         } | ||||
|       } else { | ||||
|         guess(src_o, sol_o);  | ||||
|       } | ||||
|  | ||||
| 	    if ( subGuess ) {  | ||||
|         for(int b=0;b<nblock;b++){ | ||||
|           guess_save[b] = sol_o[b]; | ||||
|         } | ||||
| 	if ( subGuess ) {  | ||||
| 	  guess_save[b] = sol_o[b]; | ||||
| 	} | ||||
|       } | ||||
|       ////////////////////////////////////////////////////////////// | ||||
|       // Call the block solver | ||||
|   | ||||
| @@ -9,30 +9,14 @@ NAMESPACE_BEGIN(Grid); | ||||
| #define AccSmall (3) | ||||
| #define Shared   (4) | ||||
| #define SharedSmall (5) | ||||
| #undef GRID_MM_VERBOSE  | ||||
| uint64_t total_shared; | ||||
| uint64_t total_device; | ||||
| uint64_t total_host;; | ||||
| void MemoryManager::PrintBytes(void) | ||||
| { | ||||
|   std::cout << " MemoryManager : ------------------------------------ "<<std::endl; | ||||
|   std::cout << " MemoryManager : PrintBytes "<<std::endl; | ||||
|   std::cout << " MemoryManager : ------------------------------------ "<<std::endl; | ||||
|   std::cout << " MemoryManager : "<<(total_shared>>20)<<" shared      Mbytes "<<std::endl; | ||||
|   std::cout << " MemoryManager : "<<(total_device>>20)<<" accelerator Mbytes "<<std::endl; | ||||
|   std::cout << " MemoryManager : "<<(total_host>>20)  <<" cpu         Mbytes "<<std::endl; | ||||
|   uint64_t cacheBytes; | ||||
|   cacheBytes = CacheBytes[Cpu]; | ||||
|   std::cout << " MemoryManager : "<<(cacheBytes>>20) <<" cpu cache Mbytes "<<std::endl; | ||||
|   cacheBytes = CacheBytes[Acc]; | ||||
|   std::cout << " MemoryManager : "<<(cacheBytes>>20) <<" acc cache Mbytes "<<std::endl; | ||||
|   cacheBytes = CacheBytes[Shared]; | ||||
|   std::cout << " MemoryManager : "<<(cacheBytes>>20) <<" shared cache Mbytes "<<std::endl; | ||||
|    | ||||
| #ifdef GRID_CUDA | ||||
|   cuda_mem(); | ||||
| #endif | ||||
|    | ||||
|   std::cout << " MemoryManager : "<<total_shared<<" shared      bytes "<<std::endl; | ||||
|   std::cout << " MemoryManager : "<<total_device<<" accelerator bytes "<<std::endl; | ||||
|   std::cout << " MemoryManager : "<<total_host  <<" cpu         bytes "<<std::endl; | ||||
| } | ||||
|  | ||||
| ////////////////////////////////////////////////////////////////////// | ||||
| @@ -40,114 +24,86 @@ void MemoryManager::PrintBytes(void) | ||||
| ////////////////////////////////////////////////////////////////////// | ||||
| MemoryManager::AllocationCacheEntry MemoryManager::Entries[MemoryManager::NallocType][MemoryManager::NallocCacheMax]; | ||||
| int MemoryManager::Victim[MemoryManager::NallocType]; | ||||
| int MemoryManager::Ncache[MemoryManager::NallocType] = { 2, 8, 2, 8, 2, 8 }; | ||||
| uint64_t MemoryManager::CacheBytes[MemoryManager::NallocType]; | ||||
| int MemoryManager::Ncache[MemoryManager::NallocType] = { 8, 32, 8, 32, 8, 32 }; | ||||
|  | ||||
| ////////////////////////////////////////////////////////////////////// | ||||
| // Actual allocation and deallocation utils | ||||
| ////////////////////////////////////////////////////////////////////// | ||||
| void *MemoryManager::AcceleratorAllocate(size_t bytes) | ||||
| { | ||||
|   total_device+=bytes; | ||||
|   void *ptr = (void *) Lookup(bytes,Acc); | ||||
|   if ( ptr == (void *) NULL ) { | ||||
|     ptr = (void *) acceleratorAllocDevice(bytes); | ||||
|     total_device+=bytes; | ||||
|   } | ||||
| #ifdef GRID_MM_VERBOSE | ||||
|   std::cout <<"AcceleratorAllocate "<<std::endl; | ||||
|   PrintBytes(); | ||||
| #endif | ||||
|   return ptr; | ||||
| } | ||||
| void  MemoryManager::AcceleratorFree    (void *ptr,size_t bytes) | ||||
| { | ||||
|   total_device-=bytes; | ||||
|   void *__freeme = Insert(ptr,bytes,Acc); | ||||
|   if ( __freeme ) { | ||||
|     acceleratorFreeDevice(__freeme); | ||||
|     total_device-=bytes; | ||||
|     //    PrintBytes(); | ||||
|   } | ||||
| #ifdef GRID_MM_VERBOSE | ||||
|   std::cout <<"AcceleratorFree "<<std::endl; | ||||
|   PrintBytes(); | ||||
| #endif | ||||
| } | ||||
| void *MemoryManager::SharedAllocate(size_t bytes) | ||||
| { | ||||
|   total_shared+=bytes; | ||||
|   void *ptr = (void *) Lookup(bytes,Shared); | ||||
|   if ( ptr == (void *) NULL ) { | ||||
|     ptr = (void *) acceleratorAllocShared(bytes); | ||||
|     total_shared+=bytes; | ||||
|     //    std::cout <<"AcceleratorAllocate: allocated Shared pointer "<<std::hex<<ptr<<std::dec<<std::endl; | ||||
|     //    PrintBytes(); | ||||
|   } | ||||
| #ifdef GRID_MM_VERBOSE | ||||
|   std::cout <<"SharedAllocate "<<std::endl; | ||||
|   PrintBytes(); | ||||
| #endif | ||||
|   return ptr; | ||||
| } | ||||
| void  MemoryManager::SharedFree    (void *ptr,size_t bytes) | ||||
| { | ||||
|   total_shared-=bytes; | ||||
|   void *__freeme = Insert(ptr,bytes,Shared); | ||||
|   if ( __freeme ) { | ||||
|     acceleratorFreeShared(__freeme); | ||||
|     total_shared-=bytes; | ||||
|     //    PrintBytes(); | ||||
|   } | ||||
| #ifdef GRID_MM_VERBOSE | ||||
|   std::cout <<"SharedFree "<<std::endl; | ||||
|   PrintBytes(); | ||||
| #endif | ||||
| } | ||||
| #ifdef GRID_UVM | ||||
| void *MemoryManager::CpuAllocate(size_t bytes) | ||||
| { | ||||
|   total_host+=bytes; | ||||
|   void *ptr = (void *) Lookup(bytes,Cpu); | ||||
|   if ( ptr == (void *) NULL ) { | ||||
|     ptr = (void *) acceleratorAllocShared(bytes); | ||||
|     total_host+=bytes; | ||||
|   } | ||||
| #ifdef GRID_MM_VERBOSE | ||||
|   std::cout <<"CpuAllocate "<<std::endl; | ||||
|   PrintBytes(); | ||||
| #endif | ||||
|   return ptr; | ||||
| } | ||||
| void  MemoryManager::CpuFree    (void *_ptr,size_t bytes) | ||||
| { | ||||
|   total_host-=bytes; | ||||
|   NotifyDeletion(_ptr); | ||||
|   void *__freeme = Insert(_ptr,bytes,Cpu); | ||||
|   if ( __freeme ) {  | ||||
|     acceleratorFreeShared(__freeme); | ||||
|     total_host-=bytes; | ||||
|   } | ||||
| #ifdef GRID_MM_VERBOSE | ||||
|   std::cout <<"CpuFree "<<std::endl; | ||||
|   PrintBytes(); | ||||
| #endif | ||||
| } | ||||
| #else | ||||
| void *MemoryManager::CpuAllocate(size_t bytes) | ||||
| { | ||||
|   total_host+=bytes; | ||||
|   void *ptr = (void *) Lookup(bytes,Cpu); | ||||
|   if ( ptr == (void *) NULL ) { | ||||
|     ptr = (void *) acceleratorAllocCpu(bytes); | ||||
|     total_host+=bytes; | ||||
|   } | ||||
| #ifdef GRID_MM_VERBOSE | ||||
|   std::cout <<"CpuAllocate "<<std::endl; | ||||
|   PrintBytes(); | ||||
| #endif | ||||
|   return ptr; | ||||
| } | ||||
| void  MemoryManager::CpuFree    (void *_ptr,size_t bytes) | ||||
| { | ||||
|   total_host-=bytes; | ||||
|   NotifyDeletion(_ptr); | ||||
|   void *__freeme = Insert(_ptr,bytes,Cpu); | ||||
|   if ( __freeme ) {  | ||||
|     acceleratorFreeCpu(__freeme); | ||||
|     total_host-=bytes; | ||||
|   } | ||||
| #ifdef GRID_MM_VERBOSE | ||||
|   std::cout <<"CpuFree "<<std::endl; | ||||
|   PrintBytes(); | ||||
| #endif | ||||
| } | ||||
| #endif | ||||
|  | ||||
| @@ -159,6 +115,7 @@ void MemoryManager::Init(void) | ||||
|  | ||||
|   char * str; | ||||
|   int Nc; | ||||
|   int NcS; | ||||
|    | ||||
|   str= getenv("GRID_ALLOC_NCACHE_LARGE"); | ||||
|   if ( str ) { | ||||
| @@ -224,13 +181,13 @@ void *MemoryManager::Insert(void *ptr,size_t bytes,int type) | ||||
| #ifdef ALLOCATION_CACHE | ||||
|   bool small = (bytes < GRID_ALLOC_SMALL_LIMIT); | ||||
|   int cache = type + small; | ||||
|   return Insert(ptr,bytes,Entries[cache],Ncache[cache],Victim[cache],CacheBytes[cache]);   | ||||
|   return Insert(ptr,bytes,Entries[cache],Ncache[cache],Victim[cache]);   | ||||
| #else | ||||
|   return ptr; | ||||
| #endif | ||||
| } | ||||
|  | ||||
| void *MemoryManager::Insert(void *ptr,size_t bytes,AllocationCacheEntry *entries,int ncache,int &victim, uint64_t &cacheBytes)  | ||||
| void *MemoryManager::Insert(void *ptr,size_t bytes,AllocationCacheEntry *entries,int ncache,int &victim)  | ||||
| { | ||||
|   assert(ncache>0); | ||||
| #ifdef GRID_OMP | ||||
| @@ -254,7 +211,6 @@ void *MemoryManager::Insert(void *ptr,size_t bytes,AllocationCacheEntry *entries | ||||
|  | ||||
|   if ( entries[v].valid ) { | ||||
|     ret = entries[v].address; | ||||
|     cacheBytes -= entries[v].bytes; | ||||
|     entries[v].valid = 0; | ||||
|     entries[v].address = NULL; | ||||
|     entries[v].bytes = 0; | ||||
| @@ -263,7 +219,6 @@ void *MemoryManager::Insert(void *ptr,size_t bytes,AllocationCacheEntry *entries | ||||
|   entries[v].address=ptr; | ||||
|   entries[v].bytes  =bytes; | ||||
|   entries[v].valid  =1; | ||||
|   cacheBytes += bytes; | ||||
|  | ||||
|   return ret; | ||||
| } | ||||
| @@ -273,13 +228,13 @@ void *MemoryManager::Lookup(size_t bytes,int type) | ||||
| #ifdef ALLOCATION_CACHE | ||||
|   bool small = (bytes < GRID_ALLOC_SMALL_LIMIT); | ||||
|   int cache = type+small; | ||||
|   return Lookup(bytes,Entries[cache],Ncache[cache],CacheBytes[cache]); | ||||
|   return Lookup(bytes,Entries[cache],Ncache[cache]); | ||||
| #else | ||||
|   return NULL; | ||||
| #endif | ||||
| } | ||||
|  | ||||
| void *MemoryManager::Lookup(size_t bytes,AllocationCacheEntry *entries,int ncache,uint64_t & cacheBytes)  | ||||
| void *MemoryManager::Lookup(size_t bytes,AllocationCacheEntry *entries,int ncache)  | ||||
| { | ||||
|   assert(ncache>0); | ||||
| #ifdef GRID_OMP | ||||
| @@ -288,7 +243,6 @@ void *MemoryManager::Lookup(size_t bytes,AllocationCacheEntry *entries,int ncach | ||||
|   for(int e=0;e<ncache;e++){ | ||||
|     if ( entries[e].valid && ( entries[e].bytes == bytes ) ) { | ||||
|       entries[e].valid = 0; | ||||
|       cacheBytes -= entries[e].bytes; | ||||
|       return entries[e].address; | ||||
|     } | ||||
|   } | ||||
|   | ||||
| @@ -82,15 +82,14 @@ private: | ||||
|   static AllocationCacheEntry Entries[NallocType][NallocCacheMax]; | ||||
|   static int Victim[NallocType]; | ||||
|   static int Ncache[NallocType]; | ||||
|   static uint64_t CacheBytes[NallocType]; | ||||
|  | ||||
|   ///////////////////////////////////////////////// | ||||
|   // Free pool | ||||
|   ///////////////////////////////////////////////// | ||||
|   static void *Insert(void *ptr,size_t bytes,int type) ; | ||||
|   static void *Lookup(size_t bytes,int type) ; | ||||
|   static void *Insert(void *ptr,size_t bytes,AllocationCacheEntry *entries,int ncache,int &victim,uint64_t &cbytes) ; | ||||
|   static void *Lookup(size_t bytes,AllocationCacheEntry *entries,int ncache,uint64_t &cbytes) ; | ||||
|   static void *Insert(void *ptr,size_t bytes,AllocationCacheEntry *entries,int ncache,int &victim) ; | ||||
|   static void *Lookup(size_t bytes,AllocationCacheEntry *entries,int ncache) ; | ||||
|  | ||||
|   static void PrintBytes(void); | ||||
|  public: | ||||
| @@ -170,7 +169,6 @@ private: | ||||
|  | ||||
|  public: | ||||
|   static void Print(void); | ||||
|   static void PrintState( void* CpuPtr); | ||||
|   static int   isOpen   (void* CpuPtr); | ||||
|   static void  ViewClose(void* CpuPtr,ViewMode mode); | ||||
|   static void *ViewOpen (void* CpuPtr,size_t bytes,ViewMode mode,ViewAdvise hint); | ||||
|   | ||||
| @@ -3,7 +3,7 @@ | ||||
|  | ||||
| #warning "Using explicit device memory copies" | ||||
| NAMESPACE_BEGIN(Grid); | ||||
| //#define dprintf(...) printf ( __VA_ARGS__ ); fflush(stdout); | ||||
| //define dprintf(...) printf ( __VA_ARGS__ ); fflush(stdout); | ||||
| #define dprintf(...) | ||||
|  | ||||
|  | ||||
| @@ -429,7 +429,6 @@ void  MemoryManager::NotifyDeletion(void *_ptr) | ||||
| } | ||||
| void  MemoryManager::Print(void) | ||||
| { | ||||
|   PrintBytes(); | ||||
|   std::cout << GridLogDebug << "--------------------------------------------" << std::endl; | ||||
|   std::cout << GridLogDebug << "Memory Manager                             " << std::endl; | ||||
|   std::cout << GridLogDebug << "--------------------------------------------" << std::endl; | ||||
| @@ -474,32 +473,6 @@ int   MemoryManager::isOpen   (void* _CpuPtr) | ||||
|   } | ||||
| } | ||||
|  | ||||
| void MemoryManager::PrintState(void* _CpuPtr) | ||||
| { | ||||
|   uint64_t CpuPtr = (uint64_t)_CpuPtr; | ||||
|  | ||||
|   if ( EntryPresent(CpuPtr) ){ | ||||
|     auto AccCacheIterator = EntryLookup(CpuPtr); | ||||
|     auto & AccCache = AccCacheIterator->second; | ||||
|     std::string str; | ||||
|     if ( AccCache.state==Empty    ) str = std::string("Empty"); | ||||
|     if ( AccCache.state==CpuDirty ) str = std::string("CpuDirty"); | ||||
|     if ( AccCache.state==AccDirty ) str = std::string("AccDirty"); | ||||
|     if ( AccCache.state==Consistent)str = std::string("Consistent"); | ||||
|     if ( AccCache.state==EvictNext) str = std::string("EvictNext"); | ||||
|  | ||||
|     std::cout << GridLogMessage << "CpuAddr\t\tAccAddr\t\tState\t\tcpuLock\taccLock\tLRU_valid "<<std::endl; | ||||
|     std::cout << GridLogMessage << "0x"<<std::hex<<AccCache.CpuPtr<<std::dec | ||||
|     << "\t0x"<<std::hex<<AccCache.AccPtr<<std::dec<<"\t" <<str | ||||
|     << "\t" << AccCache.cpuLock | ||||
|     << "\t" << AccCache.accLock | ||||
|     << "\t" << AccCache.LRU_valid<<std::endl; | ||||
|  | ||||
|   } else { | ||||
|     std::cout << GridLogMessage << "No Entry in AccCache table." << std::endl;  | ||||
|   } | ||||
| } | ||||
|  | ||||
| NAMESPACE_END(Grid); | ||||
|  | ||||
| #endif | ||||
|   | ||||
| @@ -16,10 +16,6 @@ uint64_t  MemoryManager::DeviceToHostXfer; | ||||
| void  MemoryManager::ViewClose(void* AccPtr,ViewMode mode){}; | ||||
| void *MemoryManager::ViewOpen(void* CpuPtr,size_t bytes,ViewMode mode,ViewAdvise hint){ return CpuPtr; }; | ||||
| int   MemoryManager::isOpen   (void* CpuPtr) { return 0;} | ||||
| void  MemoryManager::PrintState(void* CpuPtr) | ||||
| { | ||||
| std::cout << GridLogMessage << "Host<->Device memory movement not currently managed by Grid." << std::endl; | ||||
| }; | ||||
| void  MemoryManager::Print(void){}; | ||||
| void  MemoryManager::NotifyDeletion(void *ptr){}; | ||||
|  | ||||
|   | ||||
| @@ -33,8 +33,6 @@ Author: Peter Boyle <paboyle@ph.ed.ac.uk> | ||||
|  | ||||
| NAMESPACE_BEGIN(Grid); | ||||
|  | ||||
| bool Stencil_force_mpi = true; | ||||
|  | ||||
| /////////////////////////////////////////////////////////////// | ||||
| // Info that is setup once and indept of cartesian layout | ||||
| /////////////////////////////////////////////////////////////// | ||||
|   | ||||
| @@ -35,8 +35,6 @@ Author: Peter Boyle <paboyle@ph.ed.ac.uk> | ||||
|  | ||||
| NAMESPACE_BEGIN(Grid); | ||||
|  | ||||
| extern bool Stencil_force_mpi ; | ||||
|  | ||||
| class CartesianCommunicator : public SharedMemory { | ||||
|  | ||||
| public:     | ||||
|   | ||||
| @@ -370,7 +370,7 @@ double CartesianCommunicator::StencilSendToRecvFromBegin(std::vector<CommsReques | ||||
|   double off_node_bytes=0.0; | ||||
|   int tag; | ||||
|  | ||||
|   if ( (gfrom ==MPI_UNDEFINED) || Stencil_force_mpi ) { | ||||
|   if ( gfrom ==MPI_UNDEFINED) { | ||||
|     tag= dir+from*32; | ||||
|     ierr=MPI_Irecv(recv, bytes, MPI_CHAR,from,tag,communicator_halo[commdir],&rrq); | ||||
|     assert(ierr==0); | ||||
| @@ -378,18 +378,12 @@ double CartesianCommunicator::StencilSendToRecvFromBegin(std::vector<CommsReques | ||||
|     off_node_bytes+=bytes; | ||||
|   } | ||||
|  | ||||
|   if ( (gdest == MPI_UNDEFINED) || Stencil_force_mpi ) { | ||||
|   if ( gdest == MPI_UNDEFINED ) { | ||||
|     tag= dir+_processor*32; | ||||
|     ierr =MPI_Isend(xmit, bytes, MPI_CHAR,dest,tag,communicator_halo[commdir],&xrq); | ||||
|     assert(ierr==0); | ||||
|     list.push_back(xrq); | ||||
|     off_node_bytes+=bytes; | ||||
|   } else { | ||||
|     // TODO : make a OMP loop on CPU, call threaded bcopy | ||||
|     void *shm = (void *) this->ShmBufferTranslate(dest,recv); | ||||
|     assert(shm!=NULL); | ||||
|     //    std::cout <<"acceleratorCopyDeviceToDeviceAsynch"<< std::endl; | ||||
|     acceleratorCopyDeviceToDeviceAsynch(xmit,shm,bytes); | ||||
|   } | ||||
|  | ||||
|   if ( CommunicatorPolicy == CommunicatorPolicySequential ) { | ||||
| @@ -400,9 +394,6 @@ double CartesianCommunicator::StencilSendToRecvFromBegin(std::vector<CommsReques | ||||
| } | ||||
| void CartesianCommunicator::StencilSendToRecvFromComplete(std::vector<CommsRequest_t> &list,int dir) | ||||
| { | ||||
|   //   std::cout << "Copy Synchronised\n"<<std::endl; | ||||
|   acceleratorCopySynchronise(); | ||||
|  | ||||
|   int nreq=list.size(); | ||||
|  | ||||
|   if (nreq==0) return; | ||||
|   | ||||
| @@ -35,9 +35,6 @@ Author: Christoph Lehner <christoph@lhnr.de> | ||||
| #endif | ||||
| #ifdef GRID_HIP | ||||
| #include <hip/hip_runtime_api.h> | ||||
| #endif | ||||
| #ifdef GRID_SYCl | ||||
|  | ||||
| #endif | ||||
|  | ||||
| NAMESPACE_BEGIN(Grid);  | ||||
| @@ -73,7 +70,6 @@ void GlobalSharedMemory::Init(Grid_MPI_Comm comm) | ||||
|   WorldNodes = WorldSize/WorldShmSize; | ||||
|   assert( (WorldNodes * WorldShmSize) == WorldSize ); | ||||
|  | ||||
|  | ||||
|   // FIXME: Check all WorldShmSize are the same ? | ||||
|  | ||||
|   ///////////////////////////////////////////////////////////////////// | ||||
| @@ -450,47 +446,7 @@ void GlobalSharedMemory::SharedMemoryAllocate(uint64_t bytes, int flags) | ||||
| //////////////////////////////////////////////////////////////////////////////////////////// | ||||
| // Hugetlbfs mapping intended | ||||
| //////////////////////////////////////////////////////////////////////////////////////////// | ||||
| #if defined(GRID_CUDA) ||defined(GRID_HIP)  || defined(GRID_SYCL) | ||||
|  | ||||
| //if defined(GRID_SYCL) | ||||
| #if 0 | ||||
| void GlobalSharedMemory::SharedMemoryAllocate(uint64_t bytes, int flags) | ||||
| { | ||||
|   void * ShmCommBuf ;  | ||||
|   assert(_ShmSetup==1); | ||||
|   assert(_ShmAlloc==0); | ||||
|  | ||||
|   ////////////////////////////////////////////////////////////////////////////////////////////////////////// | ||||
|   // allocate the pointer array for shared windows for our group | ||||
|   ////////////////////////////////////////////////////////////////////////////////////////////////////////// | ||||
|   MPI_Barrier(WorldShmComm); | ||||
|   WorldShmCommBufs.resize(WorldShmSize); | ||||
|  | ||||
|   /////////////////////////////////////////////////////////////////////////////////////////////////////////// | ||||
|   // Each MPI rank should allocate our own buffer | ||||
|   /////////////////////////////////////////////////////////////////////////////////////////////////////////// | ||||
|   ShmCommBuf = acceleratorAllocDevice(bytes); | ||||
|  | ||||
|   if (ShmCommBuf == (void *)NULL ) { | ||||
|     std::cerr << " SharedMemoryMPI.cc acceleratorAllocDevice failed NULL pointer for " << bytes<<" bytes " << std::endl; | ||||
|     exit(EXIT_FAILURE);   | ||||
|   } | ||||
|  | ||||
|   std::cout << WorldRank << header " SharedMemoryMPI.cc acceleratorAllocDevice "<< bytes  | ||||
| 	    << "bytes at "<< std::hex<< ShmCommBuf <<std::dec<<" for comms buffers " <<std::endl; | ||||
|  | ||||
|   SharedMemoryZero(ShmCommBuf,bytes); | ||||
|  | ||||
|   assert(WorldShmSize == 1); | ||||
|   for(int r=0;r<WorldShmSize;r++){ | ||||
|     WorldShmCommBufs[r] = ShmCommBuf; | ||||
|   } | ||||
|   _ShmAllocBytes=bytes; | ||||
|   _ShmAlloc=1; | ||||
| } | ||||
| #endif | ||||
|  | ||||
| #if defined(GRID_CUDA) ||defined(GRID_HIP) ||defined(GRID_SYCL)   | ||||
| #if defined(GRID_CUDA) ||defined(GRID_HIP) | ||||
| void GlobalSharedMemory::SharedMemoryAllocate(uint64_t bytes, int flags) | ||||
| { | ||||
|   void * ShmCommBuf ;  | ||||
| @@ -514,16 +470,18 @@ void GlobalSharedMemory::SharedMemoryAllocate(uint64_t bytes, int flags) | ||||
|   // Each MPI rank should allocate our own buffer | ||||
|   /////////////////////////////////////////////////////////////////////////////////////////////////////////// | ||||
|   ShmCommBuf = acceleratorAllocDevice(bytes); | ||||
|  | ||||
|   if (ShmCommBuf == (void *)NULL ) { | ||||
|     std::cerr << " SharedMemoryMPI.cc acceleratorAllocDevice failed NULL pointer for " << bytes<<" bytes " << std::endl; | ||||
|     exit(EXIT_FAILURE);   | ||||
|   } | ||||
|   if ( WorldRank == 0 ){ | ||||
|   //  if ( WorldRank == 0 ){ | ||||
|   if ( 1 ){ | ||||
|     std::cout << WorldRank << header " SharedMemoryMPI.cc acceleratorAllocDevice "<< bytes  | ||||
| 	      << "bytes at "<< std::hex<< ShmCommBuf <<std::dec<<" for comms buffers " <<std::endl; | ||||
|   } | ||||
|   SharedMemoryZero(ShmCommBuf,bytes); | ||||
|   std::cout<< "Setting up IPC"<<std::endl; | ||||
|  | ||||
|   /////////////////////////////////////////////////////////////////////////////////////////////////////////// | ||||
|   // Loop over ranks/gpu's on our node | ||||
|   /////////////////////////////////////////////////////////////////////////////////////////////////////////// | ||||
| @@ -533,29 +491,6 @@ void GlobalSharedMemory::SharedMemoryAllocate(uint64_t bytes, int flags) | ||||
|     ////////////////////////////////////////////////// | ||||
|     // If it is me, pass around the IPC access key | ||||
|     ////////////////////////////////////////////////// | ||||
|     void * thisBuf = ShmCommBuf; | ||||
|     if(!Stencil_force_mpi) { | ||||
| #ifdef GRID_SYCL_LEVEL_ZERO_IPC | ||||
|     typedef struct { int fd; pid_t pid ; } clone_mem_t; | ||||
|  | ||||
|     auto zeDevice    = cl::sycl::get_native<cl::sycl::backend::level_zero>(theGridAccelerator->get_device()); | ||||
|     auto zeContext   = cl::sycl::get_native<cl::sycl::backend::level_zero>(theGridAccelerator->get_context()); | ||||
|        | ||||
|     ze_ipc_mem_handle_t ihandle; | ||||
|     clone_mem_t handle; | ||||
|  | ||||
|     if ( r==WorldShmRank ) {  | ||||
|       auto err = zeMemGetIpcHandle(zeContext,ShmCommBuf,&ihandle); | ||||
|       if ( err != ZE_RESULT_SUCCESS ) { | ||||
| 	std::cout << "SharedMemoryMPI.cc zeMemGetIpcHandle failed for rank "<<r<<" "<<std::hex<<err<<std::dec<<std::endl; | ||||
| 	exit(EXIT_FAILURE); | ||||
|       } else { | ||||
| 	std::cout << "SharedMemoryMPI.cc zeMemGetIpcHandle succeeded for rank "<<r<<" "<<std::hex<<err<<std::dec<<std::endl; | ||||
|       } | ||||
|       memcpy((void *)&handle.fd,(void *)&ihandle,sizeof(int)); | ||||
|       handle.pid = getpid(); | ||||
|     } | ||||
| #endif | ||||
| #ifdef GRID_CUDA | ||||
|     cudaIpcMemHandle_t handle; | ||||
|     if ( r==WorldShmRank ) {  | ||||
| @@ -576,7 +511,6 @@ void GlobalSharedMemory::SharedMemoryAllocate(uint64_t bytes, int flags) | ||||
|       } | ||||
|     } | ||||
| #endif | ||||
|  | ||||
|     ////////////////////////////////////////////////// | ||||
|     // Share this IPC handle across the Shm Comm | ||||
|     ////////////////////////////////////////////////// | ||||
| @@ -592,35 +526,7 @@ void GlobalSharedMemory::SharedMemoryAllocate(uint64_t bytes, int flags) | ||||
|     /////////////////////////////////////////////////////////////// | ||||
|     // If I am not the source, overwrite thisBuf with remote buffer | ||||
|     /////////////////////////////////////////////////////////////// | ||||
|  | ||||
| #ifdef GRID_SYCL_LEVEL_ZERO_IPC | ||||
|     if ( r!=WorldShmRank ) { | ||||
|       thisBuf = nullptr; | ||||
|       std::cout<<"mapping seeking remote pid/fd " | ||||
| 	       <<handle.pid<<"/" | ||||
| 	       <<handle.fd<<std::endl; | ||||
|  | ||||
|       int pidfd = syscall(SYS_pidfd_open,handle.pid,0); | ||||
|       std::cout<<"Using IpcHandle pidfd "<<pidfd<<"\n"; | ||||
|       //      int myfd  = syscall(SYS_pidfd_getfd,pidfd,handle.fd,0); | ||||
|       int myfd  = syscall(438,pidfd,handle.fd,0); | ||||
|  | ||||
|       std::cout<<"Using IpcHandle myfd "<<myfd<<"\n"; | ||||
|        | ||||
|       memcpy((void *)&ihandle,(void *)&myfd,sizeof(int)); | ||||
|  | ||||
|       auto err = zeMemOpenIpcHandle(zeContext,zeDevice,ihandle,0,&thisBuf); | ||||
|       if ( err != ZE_RESULT_SUCCESS ) { | ||||
| 	std::cout << "SharedMemoryMPI.cc "<<zeContext<<" "<<zeDevice<<std::endl; | ||||
| 	std::cout << "SharedMemoryMPI.cc zeMemOpenIpcHandle failed for rank "<<r<<" "<<std::hex<<err<<std::dec<<std::endl;  | ||||
| 	exit(EXIT_FAILURE); | ||||
|       } else { | ||||
| 	std::cout << "SharedMemoryMPI.cc zeMemOpenIpcHandle succeeded for rank "<<r<<std::endl; | ||||
| 	std::cout << "SharedMemoryMPI.cc zeMemOpenIpcHandle pointer is "<<std::hex<<thisBuf<<std::dec<<std::endl; | ||||
|       } | ||||
|       assert(thisBuf!=nullptr); | ||||
|     } | ||||
| #endif | ||||
|     void * thisBuf = ShmCommBuf; | ||||
| #ifdef GRID_CUDA | ||||
|     if ( r!=WorldShmRank ) {  | ||||
|       auto err = cudaIpcOpenMemHandle(&thisBuf,handle,cudaIpcMemLazyEnablePeerAccess); | ||||
| @@ -642,7 +548,6 @@ void GlobalSharedMemory::SharedMemoryAllocate(uint64_t bytes, int flags) | ||||
|     /////////////////////////////////////////////////////////////// | ||||
|     // Save a copy of the device buffers | ||||
|     /////////////////////////////////////////////////////////////// | ||||
|     } | ||||
|     WorldShmCommBufs[r] = thisBuf; | ||||
| #else | ||||
|     WorldShmCommBufs[r] = ShmCommBuf; | ||||
| @@ -652,8 +557,6 @@ void GlobalSharedMemory::SharedMemoryAllocate(uint64_t bytes, int flags) | ||||
|   _ShmAllocBytes=bytes; | ||||
|   _ShmAlloc=1; | ||||
| } | ||||
| #endif | ||||
|  | ||||
| #else  | ||||
| #ifdef GRID_MPI3_SHMMMAP | ||||
| void GlobalSharedMemory::SharedMemoryAllocate(uint64_t bytes, int flags) | ||||
| @@ -824,16 +727,16 @@ void GlobalSharedMemory::SharedMemoryAllocate(uint64_t bytes, int flags) | ||||
| ///////////////////////////////////////////////////////////////////////// | ||||
| void GlobalSharedMemory::SharedMemoryZero(void *dest,size_t bytes) | ||||
| { | ||||
| #if defined(GRID_CUDA) || defined(GRID_HIP) || defined(GRID_SYCL) | ||||
|   acceleratorMemSet(dest,0,bytes); | ||||
| #ifdef GRID_CUDA | ||||
|   cudaMemset(dest,0,bytes); | ||||
| #else | ||||
|   bzero(dest,bytes); | ||||
| #endif | ||||
| } | ||||
| void GlobalSharedMemory::SharedMemoryCopy(void *dest,void *src,size_t bytes) | ||||
| { | ||||
| #if defined(GRID_CUDA) || defined(GRID_HIP) || defined(GRID_SYCL) | ||||
|   acceleratorCopyToDevice(src,dest,bytes); | ||||
| #ifdef GRID_CUDA | ||||
|   cudaMemcpy(dest,src,bytes,cudaMemcpyDefault); | ||||
| #else    | ||||
|   bcopy(src,dest,bytes); | ||||
| #endif | ||||
| @@ -897,7 +800,7 @@ void SharedMemory::SetCommunicator(Grid_MPI_Comm comm) | ||||
|   } | ||||
| #endif | ||||
|  | ||||
|   //SharedMemoryTest(); | ||||
|   SharedMemoryTest(); | ||||
| } | ||||
| ////////////////////////////////////////////////////////////////// | ||||
| // On node barrier | ||||
|   | ||||
| @@ -225,7 +225,7 @@ void axpy(Lattice<vobj> &ret,sobj a,const Lattice<vobj> &x,const Lattice<vobj> & | ||||
|   autoView( x_v , x, AcceleratorRead); | ||||
|   autoView( y_v , y, AcceleratorRead); | ||||
|   accelerator_for(ss,x_v.size(),vobj::Nsimd(),{ | ||||
|     auto tmp = a*coalescedRead(x_v[ss])+coalescedRead(y_v[ss]); | ||||
|     auto tmp = a*x_v(ss)+y_v(ss); | ||||
|     coalescedWrite(ret_v[ss],tmp); | ||||
|   }); | ||||
| } | ||||
|   | ||||
| @@ -88,13 +88,6 @@ public: | ||||
|     LatticeView<vobj> accessor(*( (LatticeAccelerator<vobj> *) this),mode); | ||||
|     accessor.ViewClose(); | ||||
|   } | ||||
|  | ||||
|   // Helper function to print the state of this object in the AccCache | ||||
|   void PrintCacheState(void) | ||||
|   { | ||||
|     MemoryManager::PrintState(this->_odata); | ||||
|   } | ||||
|  | ||||
|   ///////////////////////////////////////////////////////////////////////////////// | ||||
|   // Return a view object that may be dereferenced in site loops. | ||||
|   // The view is trivially copy constructible and may be copied to an accelerator device | ||||
|   | ||||
| @@ -125,7 +125,7 @@ void basisRotate(VField &basis,Matrix& Qt,int j0, int j1, int k0,int k1,int Nm) | ||||
|  | ||||
| 	for(int k=k0; k<k1; ++k){ | ||||
| 	  auto tmp = coalescedRead(Bp[ss*nrot+j]); | ||||
| 	  coalescedWrite(Bp[ss*nrot+j],tmp+ Qt_p[jj*Nm+k] * coalescedRead(basis_vp[k][sss])); | ||||
| 	  coalescedWrite(Bp[ss*nrot+j],tmp+ Qt_p[jj*Nm+k] * coalescedRead(basis_v[k][sss])); | ||||
| 	} | ||||
|       }); | ||||
|  | ||||
| @@ -134,7 +134,7 @@ void basisRotate(VField &basis,Matrix& Qt,int j0, int j1, int k0,int k1,int Nm) | ||||
| 	int jj  =j0+j; | ||||
| 	int ss =sj/nrot; | ||||
| 	int sss=ss+s; | ||||
| 	coalescedWrite(basis_vp[jj][sss],coalescedRead(Bp[ss*nrot+j])); | ||||
| 	coalescedWrite(basis_v[jj][sss],coalescedRead(Bp[ss*nrot+j])); | ||||
|       }); | ||||
|   } | ||||
| #endif | ||||
|   | ||||
| @@ -361,7 +361,6 @@ template<class vobj> inline void sliceSum(const Lattice<vobj> &Data,std::vector< | ||||
|   // But easily avoided by using double precision fields | ||||
|   /////////////////////////////////////////////////////// | ||||
|   typedef typename vobj::scalar_object sobj; | ||||
|   typedef typename vobj::scalar_object::scalar_type scalar_type; | ||||
|   GridBase  *grid = Data.Grid(); | ||||
|   assert(grid!=NULL); | ||||
|  | ||||
| @@ -420,19 +419,20 @@ template<class vobj> inline void sliceSum(const Lattice<vobj> &Data,std::vector< | ||||
|   } | ||||
|    | ||||
|   // sum over nodes. | ||||
|   sobj gsum; | ||||
|   for(int t=0;t<fd;t++){ | ||||
|     int pt = t/ld; // processor plane | ||||
|     int lt = t%ld; | ||||
|     if ( pt == grid->_processor_coor[orthogdim] ) { | ||||
|       result[t]=lsSum[lt]; | ||||
|       gsum=lsSum[lt]; | ||||
|     } else { | ||||
|       result[t]=Zero(); | ||||
|       gsum=Zero(); | ||||
|     } | ||||
|  | ||||
|     grid->GlobalSum(gsum); | ||||
|  | ||||
|     result[t]=gsum; | ||||
|   } | ||||
|   scalar_type * ptr = (scalar_type *) &result[0]; | ||||
|   int words = fd*sizeof(sobj)/sizeof(scalar_type); | ||||
|   grid->GlobalSumVector(ptr, words); | ||||
| } | ||||
|  | ||||
| template<class vobj> | ||||
|   | ||||
| @@ -42,6 +42,7 @@ void getNumBlocksAndThreads(const Iterator n, const size_t sizeofsobj, Iterator | ||||
|   std::cout << GridLogDebug << "\twarpSize            = " << warpSize << std::endl; | ||||
|   std::cout << GridLogDebug << "\tsharedMemPerBlock   = " << sharedMemPerBlock << std::endl; | ||||
|   std::cout << GridLogDebug << "\tmaxThreadsPerBlock  = " << maxThreadsPerBlock << std::endl; | ||||
|   std::cout << GridLogDebug << "\tmaxThreadsPerBlock  = " << warpSize << std::endl; | ||||
|   std::cout << GridLogDebug << "\tmultiProcessorCount = " << multiProcessorCount << std::endl; | ||||
|    | ||||
|   if (warpSize != WARP_SIZE) { | ||||
| @@ -51,10 +52,6 @@ void getNumBlocksAndThreads(const Iterator n, const size_t sizeofsobj, Iterator | ||||
|    | ||||
|   // let the number of threads in a block be a multiple of 2, starting from warpSize | ||||
|   threads = warpSize; | ||||
|   if ( threads*sizeofsobj > sharedMemPerBlock ) { | ||||
|     std::cout << GridLogError << "The object is too large for the shared memory." << std::endl; | ||||
|     exit(EXIT_FAILURE); | ||||
|   } | ||||
|   while( 2*threads*sizeofsobj < sharedMemPerBlock && 2*threads <= maxThreadsPerBlock ) threads *= 2; | ||||
|   // keep all the streaming multiprocessors busy | ||||
|   blocks = nextPow2(multiProcessorCount); | ||||
|   | ||||
| @@ -85,76 +85,6 @@ template<class vobj> inline void setCheckerboard(Lattice<vobj> &full,const Latti | ||||
|   }); | ||||
| } | ||||
|  | ||||
| template<class vobj> inline void acceleratorPickCheckerboard(int cb,Lattice<vobj> &half,const Lattice<vobj> &full, int checker_dim_half=0) | ||||
| { | ||||
|   half.Checkerboard() = cb; | ||||
|   autoView(half_v, half, AcceleratorWrite); | ||||
|   autoView(full_v, full, AcceleratorRead); | ||||
|   Coordinate rdim_full             = full.Grid()->_rdimensions; | ||||
|   Coordinate rdim_half             = half.Grid()->_rdimensions; | ||||
|   unsigned long ndim_half          = half.Grid()->_ndimension; | ||||
|   Coordinate checker_dim_mask_half = half.Grid()->_checker_dim_mask; | ||||
|   Coordinate ostride_half          = half.Grid()->_ostride; | ||||
|   accelerator_for(ss, full.Grid()->oSites(),full.Grid()->Nsimd(),{ | ||||
|      | ||||
|     Coordinate coor; | ||||
|     int cbos; | ||||
|     int linear=0; | ||||
|  | ||||
|     Lexicographic::CoorFromIndex(coor,ss,rdim_full); | ||||
|     assert(coor.size()==ndim_half); | ||||
|  | ||||
|     for(int d=0;d<ndim_half;d++){  | ||||
|       if(checker_dim_mask_half[d]) linear += coor[d]; | ||||
|     } | ||||
|     cbos = (linear&0x1); | ||||
|  | ||||
|     if (cbos==cb) { | ||||
|       int ssh=0; | ||||
|       for(int d=0;d<ndim_half;d++) { | ||||
|         if (d == checker_dim_half) ssh += ostride_half[d] * ((coor[d] / 2) % rdim_half[d]); | ||||
|         else ssh += ostride_half[d] * (coor[d] % rdim_half[d]); | ||||
|       } | ||||
|       coalescedWrite(half_v[ssh],full_v(ss)); | ||||
|     } | ||||
|   }); | ||||
| } | ||||
| template<class vobj> inline void acceleratorSetCheckerboard(Lattice<vobj> &full,const Lattice<vobj> &half, int checker_dim_half=0) | ||||
| { | ||||
|   int cb = half.Checkerboard(); | ||||
|   autoView(half_v , half, AcceleratorRead); | ||||
|   autoView(full_v , full, AcceleratorWrite); | ||||
|   Coordinate rdim_full             = full.Grid()->_rdimensions; | ||||
|   Coordinate rdim_half             = half.Grid()->_rdimensions; | ||||
|   unsigned long ndim_half          = half.Grid()->_ndimension; | ||||
|   Coordinate checker_dim_mask_half = half.Grid()->_checker_dim_mask; | ||||
|   Coordinate ostride_half          = half.Grid()->_ostride; | ||||
|   accelerator_for(ss,full.Grid()->oSites(),full.Grid()->Nsimd(),{ | ||||
|  | ||||
|     Coordinate coor; | ||||
|     int cbos; | ||||
|     int linear=0; | ||||
|    | ||||
|     Lexicographic::CoorFromIndex(coor,ss,rdim_full); | ||||
|     assert(coor.size()==ndim_half); | ||||
|  | ||||
|     for(int d=0;d<ndim_half;d++){  | ||||
|       if(checker_dim_mask_half[d]) linear += coor[d]; | ||||
|     } | ||||
|     cbos = (linear&0x1); | ||||
|  | ||||
|     if (cbos==cb) { | ||||
|       int ssh=0; | ||||
|       for(int d=0;d<ndim_half;d++){ | ||||
|         if (d == checker_dim_half) ssh += ostride_half[d] * ((coor[d] / 2) % rdim_half[d]); | ||||
|         else ssh += ostride_half[d] * (coor[d] % rdim_half[d]); | ||||
|       } | ||||
|       coalescedWrite(full_v[ss],half_v(ssh)); | ||||
|     } | ||||
|  | ||||
|   }); | ||||
| } | ||||
|  | ||||
| //////////////////////////////////////////////////////////////////////////////////////////// | ||||
| // Flexible Type Conversion for internal promotion to double as well as graceful | ||||
| // treatment of scalar-compatible types | ||||
| @@ -434,22 +364,16 @@ inline void blockSum(Lattice<vobj> &coarseData,const Lattice<vobj> &fineData) | ||||
|   autoView( coarseData_ , coarseData, AcceleratorWrite); | ||||
|   autoView( fineData_   , fineData, AcceleratorRead); | ||||
|  | ||||
|   auto coarseData_p = &coarseData_[0]; | ||||
|   auto fineData_p = &fineData_[0]; | ||||
|    | ||||
|   Coordinate fine_rdimensions = fine->_rdimensions; | ||||
|   Coordinate coarse_rdimensions = coarse->_rdimensions; | ||||
|  | ||||
|   vobj zz = Zero(); | ||||
|    | ||||
|   accelerator_for(sc,coarse->oSites(),1,{ | ||||
|  | ||||
|       // One thread per sub block | ||||
|       Coordinate coor_c(_ndimension); | ||||
|       Lexicographic::CoorFromIndex(coor_c,sc,coarse_rdimensions);  // Block coordinate | ||||
|       coarseData_[sc]=Zero(); | ||||
|  | ||||
|       vobj cd = zz; | ||||
|        | ||||
|       for(int sb=0;sb<blockVol;sb++){ | ||||
|  | ||||
| 	int sf; | ||||
| @@ -459,11 +383,9 @@ inline void blockSum(Lattice<vobj> &coarseData,const Lattice<vobj> &fineData) | ||||
| 	for(int d=0;d<_ndimension;d++) coor_f[d]=coor_c[d]*block_r[d] + coor_b[d]; | ||||
| 	Lexicographic::IndexFromCoor(coor_f,sf,fine_rdimensions); | ||||
|  | ||||
| 	cd=cd+fineData_p[sf]; | ||||
| 	coarseData_[sc]=coarseData_[sc]+fineData_[sf]; | ||||
|       } | ||||
|  | ||||
|       coarseData_p[sc] = cd; | ||||
|  | ||||
|     }); | ||||
|   return; | ||||
| } | ||||
|   | ||||
| @@ -576,8 +576,6 @@ class ScidacReader : public GridLimeReader { | ||||
|     std::string rec_name(ILDG_BINARY_DATA); | ||||
|     while ( limeReaderNextRecord(LimeR) == LIME_SUCCESS ) {  | ||||
|       if ( !strncmp(limeReaderType(LimeR), rec_name.c_str(),strlen(rec_name.c_str()) )  ) { | ||||
|   // in principle should do the line below, but that breaks backard compatibility with old data | ||||
|   // skipPastObjectRecord(std::string(GRID_FIELD_NORM)); | ||||
| 	skipPastObjectRecord(std::string(SCIDAC_CHECKSUM)); | ||||
| 	return; | ||||
|       } | ||||
|   | ||||
| @@ -1,240 +0,0 @@ | ||||
| /************************************************************************************* | ||||
|  | ||||
|     Grid physics library, www.github.com/paboyle/Grid | ||||
|  | ||||
|     Source file: ./lib/qcd/action/fermion/CompactWilsonCloverFermion.h | ||||
|  | ||||
|     Copyright (C) 2020 - 2022 | ||||
|  | ||||
|     Author: Daniel Richtmann <daniel.richtmann@gmail.com> | ||||
|     Author: Nils Meyer <nils.meyer@ur.de> | ||||
|  | ||||
|     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 */ | ||||
|  | ||||
| #pragma once | ||||
|  | ||||
| #include <Grid/qcd/action/fermion/WilsonCloverTypes.h> | ||||
| #include <Grid/qcd/action/fermion/WilsonCloverHelpers.h> | ||||
|  | ||||
| NAMESPACE_BEGIN(Grid); | ||||
|  | ||||
| // see Grid/qcd/action/fermion/WilsonCloverFermion.h for description | ||||
| // | ||||
| // Modifications done here: | ||||
| // | ||||
| // Original: clover term = 12x12 matrix per site | ||||
| // | ||||
| // But: Only two diagonal 6x6 hermitian blocks are non-zero (also true for original, verified by running) | ||||
| // Sufficient to store/transfer only the real parts of the diagonal and one triangular part | ||||
| // 2 * (6 + 15 * 2) = 72 real or 36 complex words to be stored/transfered | ||||
| // | ||||
| // Here: Above but diagonal as complex numbers, i.e., need to store/transfer | ||||
| // 2 * (6 * 2 + 15 * 2) = 84 real or 42 complex words | ||||
| // | ||||
| // Words per site and improvement compared to original (combined with the input and output spinors): | ||||
| // | ||||
| // - Original: 2*12 + 12*12 = 168 words -> 1.00 x less | ||||
| // - Minimal:  2*12 + 36    =  60 words -> 2.80 x less | ||||
| // - Here:     2*12 + 42    =  66 words -> 2.55 x less | ||||
| // | ||||
| // These improvements directly translate to wall-clock time | ||||
| // | ||||
| // Data layout: | ||||
| // | ||||
| // - diagonal and triangle part as separate lattice fields, | ||||
| //   this was faster than as 1 combined field on all tested machines | ||||
| // - diagonal: as expected | ||||
| // - triangle: store upper right triangle in row major order | ||||
| // - graphical: | ||||
| //        0  1  2  3  4 | ||||
| //           5  6  7  8 | ||||
| //              9 10 11 = upper right triangle indices | ||||
| //                12 13 | ||||
| //                   14 | ||||
| //     0 | ||||
| //        1 | ||||
| //           2 | ||||
| //              3       = diagonal indices | ||||
| //                 4 | ||||
| //                    5 | ||||
| //     0 | ||||
| //     1  5 | ||||
| //     2  6  9          = lower left triangle indices | ||||
| //     3  7 10 12 | ||||
| //     4  8 11 13 14 | ||||
| // | ||||
| // Impact on total memory consumption: | ||||
| // - Original: (2 * 1 + 8 * 1/2) 12x12 matrices = 6 12x12 matrices = 864 complex words per site | ||||
| // - Here:     (2 * 1 + 4 * 1/2) diagonal parts = 4 diagonal parts =  24 complex words per site | ||||
| //           + (2 * 1 + 4 * 1/2) triangle parts = 4 triangle parts =  60 complex words per site | ||||
| //                                                                 =  84 complex words per site | ||||
|  | ||||
| template<class Impl> | ||||
| class CompactWilsonCloverFermion : public WilsonFermion<Impl>, | ||||
|                                    public WilsonCloverHelpers<Impl>, | ||||
|                                    public CompactWilsonCloverHelpers<Impl> { | ||||
|   ///////////////////////////////////////////// | ||||
|   // Sizes | ||||
|   ///////////////////////////////////////////// | ||||
|  | ||||
| public: | ||||
|  | ||||
|   INHERIT_COMPACT_CLOVER_SIZES(Impl); | ||||
|  | ||||
|   ///////////////////////////////////////////// | ||||
|   // Type definitions | ||||
|   ///////////////////////////////////////////// | ||||
|  | ||||
| public: | ||||
|  | ||||
|   INHERIT_IMPL_TYPES(Impl); | ||||
|   INHERIT_CLOVER_TYPES(Impl); | ||||
|   INHERIT_COMPACT_CLOVER_TYPES(Impl); | ||||
|  | ||||
|   typedef WilsonFermion<Impl>              WilsonBase; | ||||
|   typedef WilsonCloverHelpers<Impl>        Helpers; | ||||
|   typedef CompactWilsonCloverHelpers<Impl> CompactHelpers; | ||||
|  | ||||
|   ///////////////////////////////////////////// | ||||
|   // Constructors | ||||
|   ///////////////////////////////////////////// | ||||
|  | ||||
| public: | ||||
|  | ||||
|   CompactWilsonCloverFermion(GaugeField& _Umu, | ||||
| 			    GridCartesian& Fgrid, | ||||
| 			    GridRedBlackCartesian& Hgrid, | ||||
| 			    const RealD _mass, | ||||
| 			    const RealD _csw_r = 0.0, | ||||
| 			    const RealD _csw_t = 0.0, | ||||
| 			    const RealD _cF = 1.0, | ||||
| 			    const WilsonAnisotropyCoefficients& clover_anisotropy = WilsonAnisotropyCoefficients(), | ||||
| 			    const ImplParams& impl_p = ImplParams()); | ||||
|  | ||||
|   ///////////////////////////////////////////// | ||||
|   // Member functions (implementing interface) | ||||
|   ///////////////////////////////////////////// | ||||
|  | ||||
| public: | ||||
|  | ||||
|   virtual void Instantiatable() {}; | ||||
|   int          ConstEE()     override { return 0; }; | ||||
|   int          isTrivialEE() override { return 0; }; | ||||
|  | ||||
|   void Dhop(const FermionField& in, FermionField& out, int dag) override; | ||||
|  | ||||
|   void DhopOE(const FermionField& in, FermionField& out, int dag) override; | ||||
|  | ||||
|   void DhopEO(const FermionField& in, FermionField& out, int dag) override; | ||||
|  | ||||
|   void DhopDir(const FermionField& in, FermionField& out, int dir, int disp) override; | ||||
|  | ||||
|   void DhopDirAll(const FermionField& in, std::vector<FermionField>& out) /* override */; | ||||
|  | ||||
|   void M(const FermionField& in, FermionField& out) override; | ||||
|  | ||||
|   void Mdag(const FermionField& in, FermionField& out) override; | ||||
|  | ||||
|   void Meooe(const FermionField& in, FermionField& out) override; | ||||
|  | ||||
|   void MeooeDag(const FermionField& in, FermionField& out) override; | ||||
|  | ||||
|   void Mooee(const FermionField& in, FermionField& out) override; | ||||
|  | ||||
|   void MooeeDag(const FermionField& in, FermionField& out) override; | ||||
|  | ||||
|   void MooeeInv(const FermionField& in, FermionField& out) override; | ||||
|  | ||||
|   void MooeeInvDag(const FermionField& in, FermionField& out) override; | ||||
|  | ||||
|   void Mdir(const FermionField& in, FermionField& out, int dir, int disp) override; | ||||
|  | ||||
|   void MdirAll(const FermionField& in, std::vector<FermionField>& out) override; | ||||
|  | ||||
|   void MDeriv(GaugeField& force, const FermionField& X, const FermionField& Y, int dag) override; | ||||
|  | ||||
|   void MooDeriv(GaugeField& mat, const FermionField& U, const FermionField& V, int dag) override; | ||||
|  | ||||
|   void MeeDeriv(GaugeField& mat, const FermionField& U, const FermionField& V, int dag) override; | ||||
|  | ||||
|   ///////////////////////////////////////////// | ||||
|   // Member functions (internals) | ||||
|   ///////////////////////////////////////////// | ||||
|  | ||||
|   void MooeeInternal(const FermionField&        in, | ||||
|                      FermionField&              out, | ||||
|                      const CloverDiagonalField& diagonal, | ||||
|                      const CloverTriangleField& triangle); | ||||
|  | ||||
|   ///////////////////////////////////////////// | ||||
|   // Helpers | ||||
|   ///////////////////////////////////////////// | ||||
|  | ||||
|   void ImportGauge(const GaugeField& _Umu) override; | ||||
|  | ||||
|   ///////////////////////////////////////////// | ||||
|   // Helpers | ||||
|   ///////////////////////////////////////////// | ||||
|  | ||||
| private: | ||||
|  | ||||
|   template<class Field> | ||||
|   const MaskField* getCorrectMaskField(const Field &in) const { | ||||
|     if(in.Grid()->_isCheckerBoarded) { | ||||
|       if(in.Checkerboard() == Odd) { | ||||
|         return &this->BoundaryMaskOdd; | ||||
|       } else { | ||||
|         return &this->BoundaryMaskEven; | ||||
|       } | ||||
|     } else { | ||||
|       return &this->BoundaryMask; | ||||
|     } | ||||
|   } | ||||
|  | ||||
|   template<class Field> | ||||
|   void ApplyBoundaryMask(Field& f) { | ||||
|     const MaskField* m = getCorrectMaskField(f); assert(m != nullptr); | ||||
|     assert(m != nullptr); | ||||
|     CompactHelpers::ApplyBoundaryMask(f, *m); | ||||
|   } | ||||
|  | ||||
|   ///////////////////////////////////////////// | ||||
|   // Member Data | ||||
|   ///////////////////////////////////////////// | ||||
|  | ||||
| public: | ||||
|  | ||||
|   RealD csw_r; | ||||
|   RealD csw_t; | ||||
|   RealD cF; | ||||
|  | ||||
|   bool open_boundaries; | ||||
|  | ||||
|   CloverDiagonalField Diagonal,    DiagonalEven,    DiagonalOdd; | ||||
|   CloverDiagonalField DiagonalInv, DiagonalInvEven, DiagonalInvOdd; | ||||
|  | ||||
|   CloverTriangleField Triangle,    TriangleEven,    TriangleOdd; | ||||
|   CloverTriangleField TriangleInv, TriangleInvEven, TriangleInvOdd; | ||||
|  | ||||
|   FermionField Tmp; | ||||
|  | ||||
|   MaskField BoundaryMask, BoundaryMaskEven, BoundaryMaskOdd; | ||||
| }; | ||||
|  | ||||
| NAMESPACE_END(Grid); | ||||
| @@ -53,7 +53,6 @@ NAMESPACE_CHECK(Wilson); | ||||
| #include <Grid/qcd/action/fermion/WilsonTMFermion.h>       // 4d wilson like | ||||
| NAMESPACE_CHECK(WilsonTM); | ||||
| #include <Grid/qcd/action/fermion/WilsonCloverFermion.h> // 4d wilson clover fermions | ||||
| #include <Grid/qcd/action/fermion/CompactWilsonCloverFermion.h> // 4d compact wilson clover fermions | ||||
| NAMESPACE_CHECK(WilsonClover); | ||||
| #include <Grid/qcd/action/fermion/WilsonFermion5D.h>     // 5d base used by all 5d overlap types | ||||
| NAMESPACE_CHECK(Wilson5D); | ||||
| @@ -116,9 +115,9 @@ typedef WilsonFermion<WilsonImplR> WilsonFermionR; | ||||
| typedef WilsonFermion<WilsonImplF> WilsonFermionF; | ||||
| typedef WilsonFermion<WilsonImplD> WilsonFermionD; | ||||
|  | ||||
| //typedef WilsonFermion<WilsonImplRL> WilsonFermionRL; | ||||
| //typedef WilsonFermion<WilsonImplFH> WilsonFermionFH; | ||||
| //typedef WilsonFermion<WilsonImplDF> WilsonFermionDF; | ||||
| typedef WilsonFermion<WilsonImplRL> WilsonFermionRL; | ||||
| typedef WilsonFermion<WilsonImplFH> WilsonFermionFH; | ||||
| typedef WilsonFermion<WilsonImplDF> WilsonFermionDF; | ||||
|  | ||||
| typedef WilsonFermion<WilsonAdjImplR> WilsonAdjFermionR; | ||||
| typedef WilsonFermion<WilsonAdjImplF> WilsonAdjFermionF; | ||||
| @@ -154,63 +153,46 @@ typedef WilsonCloverFermion<WilsonTwoIndexAntiSymmetricImplR> WilsonCloverTwoInd | ||||
| typedef WilsonCloverFermion<WilsonTwoIndexAntiSymmetricImplF> WilsonCloverTwoIndexAntiSymmetricFermionF; | ||||
| typedef WilsonCloverFermion<WilsonTwoIndexAntiSymmetricImplD> WilsonCloverTwoIndexAntiSymmetricFermionD; | ||||
|  | ||||
| // Compact Clover fermions | ||||
| typedef CompactWilsonCloverFermion<WilsonImplR> CompactWilsonCloverFermionR; | ||||
| typedef CompactWilsonCloverFermion<WilsonImplF> CompactWilsonCloverFermionF; | ||||
| typedef CompactWilsonCloverFermion<WilsonImplD> CompactWilsonCloverFermionD; | ||||
|  | ||||
| typedef CompactWilsonCloverFermion<WilsonAdjImplR> CompactWilsonCloverAdjFermionR; | ||||
| typedef CompactWilsonCloverFermion<WilsonAdjImplF> CompactWilsonCloverAdjFermionF; | ||||
| typedef CompactWilsonCloverFermion<WilsonAdjImplD> CompactWilsonCloverAdjFermionD; | ||||
|  | ||||
| typedef CompactWilsonCloverFermion<WilsonTwoIndexSymmetricImplR> CompactWilsonCloverTwoIndexSymmetricFermionR; | ||||
| typedef CompactWilsonCloverFermion<WilsonTwoIndexSymmetricImplF> CompactWilsonCloverTwoIndexSymmetricFermionF; | ||||
| typedef CompactWilsonCloverFermion<WilsonTwoIndexSymmetricImplD> CompactWilsonCloverTwoIndexSymmetricFermionD; | ||||
|  | ||||
| typedef CompactWilsonCloverFermion<WilsonTwoIndexAntiSymmetricImplR> CompactWilsonCloverTwoIndexAntiSymmetricFermionR; | ||||
| typedef CompactWilsonCloverFermion<WilsonTwoIndexAntiSymmetricImplF> CompactWilsonCloverTwoIndexAntiSymmetricFermionF; | ||||
| typedef CompactWilsonCloverFermion<WilsonTwoIndexAntiSymmetricImplD> CompactWilsonCloverTwoIndexAntiSymmetricFermionD; | ||||
|  | ||||
| // Domain Wall fermions | ||||
| typedef DomainWallFermion<WilsonImplR> DomainWallFermionR; | ||||
| typedef DomainWallFermion<WilsonImplF> DomainWallFermionF; | ||||
| typedef DomainWallFermion<WilsonImplD> DomainWallFermionD; | ||||
|  | ||||
| //typedef DomainWallFermion<WilsonImplRL> DomainWallFermionRL; | ||||
| //typedef DomainWallFermion<WilsonImplFH> DomainWallFermionFH; | ||||
| //typedef DomainWallFermion<WilsonImplDF> DomainWallFermionDF; | ||||
| typedef DomainWallFermion<WilsonImplRL> DomainWallFermionRL; | ||||
| typedef DomainWallFermion<WilsonImplFH> DomainWallFermionFH; | ||||
| typedef DomainWallFermion<WilsonImplDF> DomainWallFermionDF; | ||||
|  | ||||
| typedef DomainWallEOFAFermion<WilsonImplR> DomainWallEOFAFermionR; | ||||
| typedef DomainWallEOFAFermion<WilsonImplF> DomainWallEOFAFermionF; | ||||
| typedef DomainWallEOFAFermion<WilsonImplD> DomainWallEOFAFermionD; | ||||
|  | ||||
| //typedef DomainWallEOFAFermion<WilsonImplRL> DomainWallEOFAFermionRL; | ||||
| //typedef DomainWallEOFAFermion<WilsonImplFH> DomainWallEOFAFermionFH; | ||||
| //typedef DomainWallEOFAFermion<WilsonImplDF> DomainWallEOFAFermionDF; | ||||
| typedef DomainWallEOFAFermion<WilsonImplRL> DomainWallEOFAFermionRL; | ||||
| typedef DomainWallEOFAFermion<WilsonImplFH> DomainWallEOFAFermionFH; | ||||
| typedef DomainWallEOFAFermion<WilsonImplDF> DomainWallEOFAFermionDF; | ||||
|  | ||||
| typedef MobiusFermion<WilsonImplR> MobiusFermionR; | ||||
| typedef MobiusFermion<WilsonImplF> MobiusFermionF; | ||||
| typedef MobiusFermion<WilsonImplD> MobiusFermionD; | ||||
|  | ||||
| //typedef MobiusFermion<WilsonImplRL> MobiusFermionRL; | ||||
| //typedef MobiusFermion<WilsonImplFH> MobiusFermionFH; | ||||
| //typedef MobiusFermion<WilsonImplDF> MobiusFermionDF; | ||||
| typedef MobiusFermion<WilsonImplRL> MobiusFermionRL; | ||||
| typedef MobiusFermion<WilsonImplFH> MobiusFermionFH; | ||||
| typedef MobiusFermion<WilsonImplDF> MobiusFermionDF; | ||||
|  | ||||
| typedef MobiusEOFAFermion<WilsonImplR> MobiusEOFAFermionR; | ||||
| typedef MobiusEOFAFermion<WilsonImplF> MobiusEOFAFermionF; | ||||
| typedef MobiusEOFAFermion<WilsonImplD> MobiusEOFAFermionD; | ||||
|  | ||||
| //typedef MobiusEOFAFermion<WilsonImplRL> MobiusEOFAFermionRL; | ||||
| //typedef MobiusEOFAFermion<WilsonImplFH> MobiusEOFAFermionFH; | ||||
| //typedef MobiusEOFAFermion<WilsonImplDF> MobiusEOFAFermionDF; | ||||
| typedef MobiusEOFAFermion<WilsonImplRL> MobiusEOFAFermionRL; | ||||
| typedef MobiusEOFAFermion<WilsonImplFH> MobiusEOFAFermionFH; | ||||
| typedef MobiusEOFAFermion<WilsonImplDF> MobiusEOFAFermionDF; | ||||
|  | ||||
| typedef ZMobiusFermion<ZWilsonImplR> ZMobiusFermionR; | ||||
| typedef ZMobiusFermion<ZWilsonImplF> ZMobiusFermionF; | ||||
| typedef ZMobiusFermion<ZWilsonImplD> ZMobiusFermionD; | ||||
|  | ||||
| //typedef ZMobiusFermion<ZWilsonImplRL> ZMobiusFermionRL; | ||||
| //typedef ZMobiusFermion<ZWilsonImplFH> ZMobiusFermionFH; | ||||
| //typedef ZMobiusFermion<ZWilsonImplDF> ZMobiusFermionDF; | ||||
| typedef ZMobiusFermion<ZWilsonImplRL> ZMobiusFermionRL; | ||||
| typedef ZMobiusFermion<ZWilsonImplFH> ZMobiusFermionFH; | ||||
| typedef ZMobiusFermion<ZWilsonImplDF> ZMobiusFermionDF; | ||||
|  | ||||
| // Ls vectorised | ||||
| typedef ScaledShamirFermion<WilsonImplR> ScaledShamirFermionR; | ||||
| @@ -253,49 +235,49 @@ typedef WilsonFermion<GparityWilsonImplR>     GparityWilsonFermionR; | ||||
| typedef WilsonFermion<GparityWilsonImplF>     GparityWilsonFermionF; | ||||
| typedef WilsonFermion<GparityWilsonImplD>     GparityWilsonFermionD; | ||||
|  | ||||
| //typedef WilsonFermion<GparityWilsonImplRL>     GparityWilsonFermionRL; | ||||
| //typedef WilsonFermion<GparityWilsonImplFH>     GparityWilsonFermionFH; | ||||
| //typedef WilsonFermion<GparityWilsonImplDF>     GparityWilsonFermionDF; | ||||
| typedef WilsonFermion<GparityWilsonImplRL>     GparityWilsonFermionRL; | ||||
| typedef WilsonFermion<GparityWilsonImplFH>     GparityWilsonFermionFH; | ||||
| typedef WilsonFermion<GparityWilsonImplDF>     GparityWilsonFermionDF; | ||||
|  | ||||
| typedef DomainWallFermion<GparityWilsonImplR> GparityDomainWallFermionR; | ||||
| typedef DomainWallFermion<GparityWilsonImplF> GparityDomainWallFermionF; | ||||
| typedef DomainWallFermion<GparityWilsonImplD> GparityDomainWallFermionD; | ||||
|  | ||||
| //typedef DomainWallFermion<GparityWilsonImplRL> GparityDomainWallFermionRL; | ||||
| //typedef DomainWallFermion<GparityWilsonImplFH> GparityDomainWallFermionFH; | ||||
| //typedef DomainWallFermion<GparityWilsonImplDF> GparityDomainWallFermionDF; | ||||
| typedef DomainWallFermion<GparityWilsonImplRL> GparityDomainWallFermionRL; | ||||
| typedef DomainWallFermion<GparityWilsonImplFH> GparityDomainWallFermionFH; | ||||
| typedef DomainWallFermion<GparityWilsonImplDF> GparityDomainWallFermionDF; | ||||
|  | ||||
| typedef DomainWallEOFAFermion<GparityWilsonImplR> GparityDomainWallEOFAFermionR; | ||||
| typedef DomainWallEOFAFermion<GparityWilsonImplF> GparityDomainWallEOFAFermionF; | ||||
| typedef DomainWallEOFAFermion<GparityWilsonImplD> GparityDomainWallEOFAFermionD; | ||||
|  | ||||
| //typedef DomainWallEOFAFermion<GparityWilsonImplRL> GparityDomainWallEOFAFermionRL; | ||||
| //typedef DomainWallEOFAFermion<GparityWilsonImplFH> GparityDomainWallEOFAFermionFH; | ||||
| //typedef DomainWallEOFAFermion<GparityWilsonImplDF> GparityDomainWallEOFAFermionDF; | ||||
| typedef DomainWallEOFAFermion<GparityWilsonImplRL> GparityDomainWallEOFAFermionRL; | ||||
| typedef DomainWallEOFAFermion<GparityWilsonImplFH> GparityDomainWallEOFAFermionFH; | ||||
| typedef DomainWallEOFAFermion<GparityWilsonImplDF> GparityDomainWallEOFAFermionDF; | ||||
|  | ||||
| typedef WilsonTMFermion<GparityWilsonImplR> GparityWilsonTMFermionR; | ||||
| typedef WilsonTMFermion<GparityWilsonImplF> GparityWilsonTMFermionF; | ||||
| typedef WilsonTMFermion<GparityWilsonImplD> GparityWilsonTMFermionD; | ||||
|  | ||||
| //typedef WilsonTMFermion<GparityWilsonImplRL> GparityWilsonTMFermionRL; | ||||
| //typedef WilsonTMFermion<GparityWilsonImplFH> GparityWilsonTMFermionFH; | ||||
| //typedef WilsonTMFermion<GparityWilsonImplDF> GparityWilsonTMFermionDF; | ||||
| typedef WilsonTMFermion<GparityWilsonImplRL> GparityWilsonTMFermionRL; | ||||
| typedef WilsonTMFermion<GparityWilsonImplFH> GparityWilsonTMFermionFH; | ||||
| typedef WilsonTMFermion<GparityWilsonImplDF> GparityWilsonTMFermionDF; | ||||
|  | ||||
| typedef MobiusFermion<GparityWilsonImplR> GparityMobiusFermionR; | ||||
| typedef MobiusFermion<GparityWilsonImplF> GparityMobiusFermionF; | ||||
| typedef MobiusFermion<GparityWilsonImplD> GparityMobiusFermionD; | ||||
|  | ||||
| //typedef MobiusFermion<GparityWilsonImplRL> GparityMobiusFermionRL; | ||||
| //typedef MobiusFermion<GparityWilsonImplFH> GparityMobiusFermionFH; | ||||
| //typedef MobiusFermion<GparityWilsonImplDF> GparityMobiusFermionDF; | ||||
| typedef MobiusFermion<GparityWilsonImplRL> GparityMobiusFermionRL; | ||||
| typedef MobiusFermion<GparityWilsonImplFH> GparityMobiusFermionFH; | ||||
| typedef MobiusFermion<GparityWilsonImplDF> GparityMobiusFermionDF; | ||||
|  | ||||
| typedef MobiusEOFAFermion<GparityWilsonImplR> GparityMobiusEOFAFermionR; | ||||
| typedef MobiusEOFAFermion<GparityWilsonImplF> GparityMobiusEOFAFermionF; | ||||
| typedef MobiusEOFAFermion<GparityWilsonImplD> GparityMobiusEOFAFermionD; | ||||
|  | ||||
| //typedef MobiusEOFAFermion<GparityWilsonImplRL> GparityMobiusEOFAFermionRL; | ||||
| //typedef MobiusEOFAFermion<GparityWilsonImplFH> GparityMobiusEOFAFermionFH; | ||||
| //typedef MobiusEOFAFermion<GparityWilsonImplDF> GparityMobiusEOFAFermionDF; | ||||
| typedef MobiusEOFAFermion<GparityWilsonImplRL> GparityMobiusEOFAFermionRL; | ||||
| typedef MobiusEOFAFermion<GparityWilsonImplFH> GparityMobiusEOFAFermionFH; | ||||
| typedef MobiusEOFAFermion<GparityWilsonImplDF> GparityMobiusEOFAFermionDF; | ||||
|  | ||||
| typedef ImprovedStaggeredFermion<StaggeredImplR> ImprovedStaggeredFermionR; | ||||
| typedef ImprovedStaggeredFermion<StaggeredImplF> ImprovedStaggeredFermionF; | ||||
|   | ||||
| @@ -327,8 +327,8 @@ typedef GparityWilsonImpl<vComplex , FundamentalRepresentation,CoeffReal> Gparit | ||||
| typedef GparityWilsonImpl<vComplexF, FundamentalRepresentation,CoeffReal> GparityWilsonImplF;  // Float | ||||
| typedef GparityWilsonImpl<vComplexD, FundamentalRepresentation,CoeffReal> GparityWilsonImplD;  // Double | ||||
|   | ||||
| //typedef GparityWilsonImpl<vComplex , FundamentalRepresentation,CoeffRealHalfComms> GparityWilsonImplRL;  // Real.. whichever prec | ||||
| //typedef GparityWilsonImpl<vComplexF, FundamentalRepresentation,CoeffRealHalfComms> GparityWilsonImplFH;  // Float | ||||
| //typedef GparityWilsonImpl<vComplexD, FundamentalRepresentation,CoeffRealHalfComms> GparityWilsonImplDF;  // Double | ||||
| typedef GparityWilsonImpl<vComplex , FundamentalRepresentation,CoeffRealHalfComms> GparityWilsonImplRL;  // Real.. whichever prec | ||||
| typedef GparityWilsonImpl<vComplexF, FundamentalRepresentation,CoeffRealHalfComms> GparityWilsonImplFH;  // Float | ||||
| typedef GparityWilsonImpl<vComplexD, FundamentalRepresentation,CoeffRealHalfComms> GparityWilsonImplDF;  // Double | ||||
|  | ||||
| NAMESPACE_END(Grid); | ||||
|   | ||||
| @@ -4,11 +4,10 @@ | ||||
|  | ||||
|     Source file: ./lib/qcd/action/fermion/WilsonCloverFermion.h | ||||
|  | ||||
|     Copyright (C) 2017 - 2022 | ||||
|     Copyright (C) 2017 | ||||
|  | ||||
|     Author: Guido Cossu <guido.cossu@ed.ac.uk> | ||||
|     Author: David Preti <> | ||||
|     Author: Daniel Richtmann <daniel.richtmann@gmail.com> | ||||
|  | ||||
|     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 | ||||
| @@ -30,8 +29,7 @@ | ||||
|  | ||||
| #pragma once | ||||
|  | ||||
| #include <Grid/qcd/action/fermion/WilsonCloverTypes.h> | ||||
| #include <Grid/qcd/action/fermion/WilsonCloverHelpers.h> | ||||
| #include <Grid/Grid.h> | ||||
|  | ||||
| NAMESPACE_BEGIN(Grid); | ||||
|  | ||||
| @@ -52,15 +50,18 @@ NAMESPACE_BEGIN(Grid); | ||||
| ////////////////////////////////////////////////////////////////// | ||||
|  | ||||
| template <class Impl> | ||||
| class WilsonCloverFermion : public WilsonFermion<Impl>, | ||||
|                             public WilsonCloverHelpers<Impl> | ||||
| class WilsonCloverFermion : public WilsonFermion<Impl> | ||||
| { | ||||
| public: | ||||
|   // Types definitions | ||||
|   INHERIT_IMPL_TYPES(Impl); | ||||
|   INHERIT_CLOVER_TYPES(Impl); | ||||
|   template <typename vtype> | ||||
|   using iImplClover = iScalar<iMatrix<iMatrix<vtype, Impl::Dimension>, Ns>>; | ||||
|   typedef iImplClover<Simd> SiteCloverType; | ||||
|   typedef Lattice<SiteCloverType> CloverFieldType; | ||||
|  | ||||
|   typedef WilsonFermion<Impl>       WilsonBase; | ||||
|   typedef WilsonCloverHelpers<Impl> Helpers; | ||||
| public: | ||||
|   typedef WilsonFermion<Impl> WilsonBase; | ||||
|  | ||||
|   virtual int    ConstEE(void)     { return 0; }; | ||||
|   virtual void Instantiatable(void){}; | ||||
| @@ -71,7 +72,42 @@ public: | ||||
|                       const RealD _csw_r = 0.0, | ||||
|                       const RealD _csw_t = 0.0, | ||||
|                       const WilsonAnisotropyCoefficients &clover_anisotropy = WilsonAnisotropyCoefficients(), | ||||
|                       const ImplParams &impl_p = ImplParams()); | ||||
|                       const ImplParams &impl_p = ImplParams()) : WilsonFermion<Impl>(_Umu, | ||||
|                                                                                      Fgrid, | ||||
|                                                                                      Hgrid, | ||||
|                                                                                      _mass, impl_p, clover_anisotropy), | ||||
|                                                                  CloverTerm(&Fgrid), | ||||
|                                                                  CloverTermInv(&Fgrid), | ||||
|                                                                  CloverTermEven(&Hgrid), | ||||
|                                                                  CloverTermOdd(&Hgrid), | ||||
|                                                                  CloverTermInvEven(&Hgrid), | ||||
|                                                                  CloverTermInvOdd(&Hgrid), | ||||
|                                                                  CloverTermDagEven(&Hgrid), | ||||
|                                                                  CloverTermDagOdd(&Hgrid), | ||||
|                                                                  CloverTermInvDagEven(&Hgrid), | ||||
|                                                                  CloverTermInvDagOdd(&Hgrid) | ||||
|   { | ||||
|     assert(Nd == 4); // require 4 dimensions | ||||
|  | ||||
|     if (clover_anisotropy.isAnisotropic) | ||||
|     { | ||||
|       csw_r = _csw_r * 0.5 / clover_anisotropy.xi_0; | ||||
|       diag_mass = _mass + 1.0 + (Nd - 1) * (clover_anisotropy.nu / clover_anisotropy.xi_0); | ||||
|     } | ||||
|     else | ||||
|     { | ||||
|       csw_r = _csw_r * 0.5; | ||||
|       diag_mass = 4.0 + _mass; | ||||
|     } | ||||
|     csw_t = _csw_t * 0.5; | ||||
|  | ||||
|     if (csw_r == 0) | ||||
|       std::cout << GridLogWarning << "Initializing WilsonCloverFermion with csw_r = 0" << std::endl; | ||||
|     if (csw_t == 0) | ||||
|       std::cout << GridLogWarning << "Initializing WilsonCloverFermion with csw_t = 0" << std::endl; | ||||
|  | ||||
|     ImportGauge(_Umu); | ||||
|   } | ||||
|  | ||||
|   virtual void M(const FermionField &in, FermionField &out); | ||||
|   virtual void Mdag(const FermionField &in, FermionField &out); | ||||
| @@ -88,21 +124,250 @@ public: | ||||
|   void ImportGauge(const GaugeField &_Umu); | ||||
|  | ||||
|   // Derivative parts unpreconditioned pseudofermions | ||||
|   void MDeriv(GaugeField &force, const FermionField &X, const FermionField &Y, int dag); | ||||
|   void MDeriv(GaugeField &force, const FermionField &X, const FermionField &Y, int dag) | ||||
|   { | ||||
|     conformable(X.Grid(), Y.Grid()); | ||||
|     conformable(X.Grid(), force.Grid()); | ||||
|     GaugeLinkField force_mu(force.Grid()), lambda(force.Grid()); | ||||
|     GaugeField clover_force(force.Grid()); | ||||
|     PropagatorField Lambda(force.Grid()); | ||||
|  | ||||
| public: | ||||
|     // Guido: Here we are hitting some performance issues: | ||||
|     // need to extract the components of the DoubledGaugeField | ||||
|     // for each call | ||||
|     // Possible solution | ||||
|     // Create a vector object to store them? (cons: wasting space) | ||||
|     std::vector<GaugeLinkField> U(Nd, this->Umu.Grid()); | ||||
|  | ||||
|     Impl::extractLinkField(U, this->Umu); | ||||
|  | ||||
|     force = Zero(); | ||||
|     // Derivative of the Wilson hopping term | ||||
|     this->DhopDeriv(force, X, Y, dag); | ||||
|  | ||||
|     /////////////////////////////////////////////////////////// | ||||
|     // Clover term derivative | ||||
|     /////////////////////////////////////////////////////////// | ||||
|     Impl::outerProductImpl(Lambda, X, Y); | ||||
|     //std::cout << "Lambda:" << Lambda << std::endl; | ||||
|  | ||||
|     Gamma::Algebra sigma[] = { | ||||
|         Gamma::Algebra::SigmaXY, | ||||
|         Gamma::Algebra::SigmaXZ, | ||||
|         Gamma::Algebra::SigmaXT, | ||||
|         Gamma::Algebra::MinusSigmaXY, | ||||
|         Gamma::Algebra::SigmaYZ, | ||||
|         Gamma::Algebra::SigmaYT, | ||||
|         Gamma::Algebra::MinusSigmaXZ, | ||||
|         Gamma::Algebra::MinusSigmaYZ, | ||||
|         Gamma::Algebra::SigmaZT, | ||||
|         Gamma::Algebra::MinusSigmaXT, | ||||
|         Gamma::Algebra::MinusSigmaYT, | ||||
|         Gamma::Algebra::MinusSigmaZT}; | ||||
|  | ||||
|     /* | ||||
|       sigma_{\mu \nu}= | ||||
|       | 0         sigma[0]  sigma[1]  sigma[2] | | ||||
|       | sigma[3]    0       sigma[4]  sigma[5] | | ||||
|       | sigma[6]  sigma[7]     0      sigma[8] | | ||||
|       | sigma[9]  sigma[10] sigma[11]   0      | | ||||
|     */ | ||||
|  | ||||
|     int count = 0; | ||||
|     clover_force = Zero(); | ||||
|     for (int mu = 0; mu < 4; mu++) | ||||
|     { | ||||
|       force_mu = Zero(); | ||||
|       for (int nu = 0; nu < 4; nu++) | ||||
|       { | ||||
|         if (mu == nu) | ||||
|         continue; | ||||
|          | ||||
|         RealD factor; | ||||
|         if (nu == 4 || mu == 4) | ||||
|         { | ||||
|           factor = 2.0 * csw_t; | ||||
|         } | ||||
|         else | ||||
|         { | ||||
|           factor = 2.0 * csw_r; | ||||
|         } | ||||
|         PropagatorField Slambda = Gamma(sigma[count]) * Lambda; // sigma checked | ||||
|         Impl::TraceSpinImpl(lambda, Slambda);                   // traceSpin ok | ||||
|         force_mu -= factor*Cmunu(U, lambda, mu, nu);                   // checked | ||||
|         count++; | ||||
|       } | ||||
|  | ||||
|       pokeLorentz(clover_force, U[mu] * force_mu, mu); | ||||
|     } | ||||
|     //clover_force *= csw; | ||||
|     force += clover_force; | ||||
|   } | ||||
|  | ||||
|   // Computing C_{\mu \nu}(x) as in Eq.(B.39) in Zbigniew Sroczynski's PhD thesis | ||||
|   GaugeLinkField Cmunu(std::vector<GaugeLinkField> &U, GaugeLinkField &lambda, int mu, int nu) | ||||
|   { | ||||
|     conformable(lambda.Grid(), U[0].Grid()); | ||||
|     GaugeLinkField out(lambda.Grid()), tmp(lambda.Grid()); | ||||
|     // insertion in upper staple | ||||
|     // please check redundancy of shift operations | ||||
|  | ||||
|     // C1+ | ||||
|     tmp = lambda * U[nu]; | ||||
|     out = Impl::ShiftStaple(Impl::CovShiftForward(tmp, nu, Impl::CovShiftBackward(U[mu], mu, Impl::CovShiftIdentityBackward(U[nu], nu))), mu); | ||||
|  | ||||
|     // C2+ | ||||
|     tmp = U[mu] * Impl::ShiftStaple(adj(lambda), mu); | ||||
|     out += Impl::ShiftStaple(Impl::CovShiftForward(U[nu], nu, Impl::CovShiftBackward(tmp, mu, Impl::CovShiftIdentityBackward(U[nu], nu))), mu); | ||||
|  | ||||
|     // C3+ | ||||
|     tmp = U[nu] * Impl::ShiftStaple(adj(lambda), nu); | ||||
|     out += Impl::ShiftStaple(Impl::CovShiftForward(U[nu], nu, Impl::CovShiftBackward(U[mu], mu, Impl::CovShiftIdentityBackward(tmp, nu))), mu); | ||||
|  | ||||
|     // C4+ | ||||
|     out += Impl::ShiftStaple(Impl::CovShiftForward(U[nu], nu, Impl::CovShiftBackward(U[mu], mu, Impl::CovShiftIdentityBackward(U[nu], nu))), mu) * lambda; | ||||
|  | ||||
|     // insertion in lower staple | ||||
|     // C1- | ||||
|     out -= Impl::ShiftStaple(lambda, mu) * Impl::ShiftStaple(Impl::CovShiftBackward(U[nu], nu, Impl::CovShiftBackward(U[mu], mu, U[nu])), mu); | ||||
|  | ||||
|     // C2- | ||||
|     tmp = adj(lambda) * U[nu]; | ||||
|     out -= Impl::ShiftStaple(Impl::CovShiftBackward(tmp, nu, Impl::CovShiftBackward(U[mu], mu, U[nu])), mu); | ||||
|  | ||||
|     // C3- | ||||
|     tmp = lambda * U[nu]; | ||||
|     out -= Impl::ShiftStaple(Impl::CovShiftBackward(U[nu], nu, Impl::CovShiftBackward(U[mu], mu, tmp)), mu); | ||||
|  | ||||
|     // C4- | ||||
|     out -= Impl::ShiftStaple(Impl::CovShiftBackward(U[nu], nu, Impl::CovShiftBackward(U[mu], mu, U[nu])), mu) * lambda; | ||||
|  | ||||
|     return out; | ||||
|   } | ||||
|  | ||||
| protected: | ||||
|   // here fixing the 4 dimensions, make it more general? | ||||
|  | ||||
|   RealD csw_r;                                               // Clover coefficient - spatial | ||||
|   RealD csw_t;                                               // Clover coefficient - temporal | ||||
|   RealD diag_mass;                                           // Mass term | ||||
|   CloverField CloverTerm, CloverTermInv;                     // Clover term | ||||
|   CloverField CloverTermEven, CloverTermOdd;                 // Clover term EO | ||||
|   CloverField CloverTermInvEven, CloverTermInvOdd;           // Clover term Inv EO | ||||
|   CloverField CloverTermDagEven, CloverTermDagOdd;           // Clover term Dag EO | ||||
|   CloverField CloverTermInvDagEven, CloverTermInvDagOdd;     // Clover term Inv Dag EO | ||||
| }; | ||||
|   CloverFieldType CloverTerm, CloverTermInv;                 // Clover term | ||||
|   CloverFieldType CloverTermEven, CloverTermOdd;             // Clover term EO | ||||
|   CloverFieldType CloverTermInvEven, CloverTermInvOdd;       // Clover term Inv EO | ||||
|   CloverFieldType CloverTermDagEven, CloverTermDagOdd;       // Clover term Dag EO | ||||
|   CloverFieldType CloverTermInvDagEven, CloverTermInvDagOdd; // Clover term Inv Dag EO | ||||
|  | ||||
|  public: | ||||
|   // eventually these can be compressed into 6x6 blocks instead of the 12x12 | ||||
|   // using the DeGrand-Rossi basis for the gamma matrices | ||||
|   CloverFieldType fillCloverYZ(const GaugeLinkField &F) | ||||
|   { | ||||
|     CloverFieldType T(F.Grid()); | ||||
|     T = Zero(); | ||||
|     autoView(T_v,T,AcceleratorWrite); | ||||
|     autoView(F_v,F,AcceleratorRead); | ||||
|     accelerator_for(i, CloverTerm.Grid()->oSites(),1, | ||||
|     { | ||||
|       T_v[i]()(0, 1) = timesMinusI(F_v[i]()()); | ||||
|       T_v[i]()(1, 0) = timesMinusI(F_v[i]()()); | ||||
|       T_v[i]()(2, 3) = timesMinusI(F_v[i]()()); | ||||
|       T_v[i]()(3, 2) = timesMinusI(F_v[i]()()); | ||||
|     }); | ||||
|  | ||||
|     return T; | ||||
|   } | ||||
|  | ||||
|   CloverFieldType fillCloverXZ(const GaugeLinkField &F) | ||||
|   { | ||||
|     CloverFieldType T(F.Grid()); | ||||
|     T = Zero(); | ||||
|      | ||||
|     autoView(T_v, T,AcceleratorWrite); | ||||
|     autoView(F_v, F,AcceleratorRead); | ||||
|     accelerator_for(i, CloverTerm.Grid()->oSites(),1, | ||||
|     { | ||||
|       T_v[i]()(0, 1) = -F_v[i]()(); | ||||
|       T_v[i]()(1, 0) = F_v[i]()(); | ||||
|       T_v[i]()(2, 3) = -F_v[i]()(); | ||||
|       T_v[i]()(3, 2) = F_v[i]()(); | ||||
|     }); | ||||
|  | ||||
|     return T; | ||||
|   } | ||||
|  | ||||
|   CloverFieldType fillCloverXY(const GaugeLinkField &F) | ||||
|   { | ||||
|     CloverFieldType T(F.Grid()); | ||||
|     T = Zero(); | ||||
|  | ||||
|     autoView(T_v,T,AcceleratorWrite); | ||||
|     autoView(F_v,F,AcceleratorRead); | ||||
|     accelerator_for(i, CloverTerm.Grid()->oSites(),1, | ||||
|     { | ||||
|       T_v[i]()(0, 0) = timesMinusI(F_v[i]()()); | ||||
|       T_v[i]()(1, 1) = timesI(F_v[i]()()); | ||||
|       T_v[i]()(2, 2) = timesMinusI(F_v[i]()()); | ||||
|       T_v[i]()(3, 3) = timesI(F_v[i]()()); | ||||
|     }); | ||||
|  | ||||
|     return T; | ||||
|   } | ||||
|  | ||||
|   CloverFieldType fillCloverXT(const GaugeLinkField &F) | ||||
|   { | ||||
|     CloverFieldType T(F.Grid()); | ||||
|     T = Zero(); | ||||
|  | ||||
|     autoView( T_v , T, AcceleratorWrite); | ||||
|     autoView( F_v , F, AcceleratorRead); | ||||
|     accelerator_for(i, CloverTerm.Grid()->oSites(),1, | ||||
|     { | ||||
|       T_v[i]()(0, 1) = timesI(F_v[i]()()); | ||||
|       T_v[i]()(1, 0) = timesI(F_v[i]()()); | ||||
|       T_v[i]()(2, 3) = timesMinusI(F_v[i]()()); | ||||
|       T_v[i]()(3, 2) = timesMinusI(F_v[i]()()); | ||||
|     }); | ||||
|  | ||||
|     return T; | ||||
|   } | ||||
|  | ||||
|   CloverFieldType fillCloverYT(const GaugeLinkField &F) | ||||
|   { | ||||
|     CloverFieldType T(F.Grid()); | ||||
|     T = Zero(); | ||||
|      | ||||
|     autoView( T_v ,T,AcceleratorWrite); | ||||
|     autoView( F_v ,F,AcceleratorRead); | ||||
|     accelerator_for(i, CloverTerm.Grid()->oSites(),1, | ||||
|     { | ||||
|       T_v[i]()(0, 1) = -(F_v[i]()()); | ||||
|       T_v[i]()(1, 0) = (F_v[i]()()); | ||||
|       T_v[i]()(2, 3) = (F_v[i]()()); | ||||
|       T_v[i]()(3, 2) = -(F_v[i]()()); | ||||
|     }); | ||||
|  | ||||
|     return T; | ||||
|   } | ||||
|  | ||||
|   CloverFieldType fillCloverZT(const GaugeLinkField &F) | ||||
|   { | ||||
|     CloverFieldType T(F.Grid()); | ||||
|  | ||||
|     T = Zero(); | ||||
|  | ||||
|     autoView( T_v , T,AcceleratorWrite); | ||||
|     autoView( F_v , F,AcceleratorRead); | ||||
|     accelerator_for(i, CloverTerm.Grid()->oSites(),1, | ||||
|     { | ||||
|       T_v[i]()(0, 0) = timesI(F_v[i]()()); | ||||
|       T_v[i]()(1, 1) = timesMinusI(F_v[i]()()); | ||||
|       T_v[i]()(2, 2) = timesMinusI(F_v[i]()()); | ||||
|       T_v[i]()(3, 3) = timesI(F_v[i]()()); | ||||
|     }); | ||||
|  | ||||
|     return T; | ||||
|   } | ||||
| }; | ||||
| NAMESPACE_END(Grid); | ||||
|  | ||||
|  | ||||
|   | ||||
| @@ -1,761 +0,0 @@ | ||||
| /************************************************************************************* | ||||
|  | ||||
|     Grid physics library, www.github.com/paboyle/Grid | ||||
|  | ||||
|     Source file: ./lib/qcd/action/fermion/WilsonCloverHelpers.h | ||||
|  | ||||
|     Copyright (C) 2021 - 2022 | ||||
|  | ||||
|     Author: Daniel Richtmann <daniel.richtmann@gmail.com> | ||||
|  | ||||
|     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 */ | ||||
|  | ||||
| #pragma once | ||||
|  | ||||
| // Helper routines that implement common clover functionality | ||||
|  | ||||
| NAMESPACE_BEGIN(Grid); | ||||
|  | ||||
| template<class Impl> class WilsonCloverHelpers { | ||||
| public: | ||||
|  | ||||
|   INHERIT_IMPL_TYPES(Impl); | ||||
|   INHERIT_CLOVER_TYPES(Impl); | ||||
|  | ||||
|   // Computing C_{\mu \nu}(x) as in Eq.(B.39) in Zbigniew Sroczynski's PhD thesis | ||||
|   static GaugeLinkField Cmunu(std::vector<GaugeLinkField> &U, GaugeLinkField &lambda, int mu, int nu) | ||||
|   { | ||||
|     conformable(lambda.Grid(), U[0].Grid()); | ||||
|     GaugeLinkField out(lambda.Grid()), tmp(lambda.Grid()); | ||||
|     // insertion in upper staple | ||||
|     // please check redundancy of shift operations | ||||
|  | ||||
|     // C1+ | ||||
|     tmp = lambda * U[nu]; | ||||
|     out = Impl::ShiftStaple(Impl::CovShiftForward(tmp, nu, Impl::CovShiftBackward(U[mu], mu, Impl::CovShiftIdentityBackward(U[nu], nu))), mu); | ||||
|  | ||||
|     // C2+ | ||||
|     tmp = U[mu] * Impl::ShiftStaple(adj(lambda), mu); | ||||
|     out += Impl::ShiftStaple(Impl::CovShiftForward(U[nu], nu, Impl::CovShiftBackward(tmp, mu, Impl::CovShiftIdentityBackward(U[nu], nu))), mu); | ||||
|  | ||||
|     // C3+ | ||||
|     tmp = U[nu] * Impl::ShiftStaple(adj(lambda), nu); | ||||
|     out += Impl::ShiftStaple(Impl::CovShiftForward(U[nu], nu, Impl::CovShiftBackward(U[mu], mu, Impl::CovShiftIdentityBackward(tmp, nu))), mu); | ||||
|  | ||||
|     // C4+ | ||||
|     out += Impl::ShiftStaple(Impl::CovShiftForward(U[nu], nu, Impl::CovShiftBackward(U[mu], mu, Impl::CovShiftIdentityBackward(U[nu], nu))), mu) * lambda; | ||||
|  | ||||
|     // insertion in lower staple | ||||
|     // C1- | ||||
|     out -= Impl::ShiftStaple(lambda, mu) * Impl::ShiftStaple(Impl::CovShiftBackward(U[nu], nu, Impl::CovShiftBackward(U[mu], mu, U[nu])), mu); | ||||
|  | ||||
|     // C2- | ||||
|     tmp = adj(lambda) * U[nu]; | ||||
|     out -= Impl::ShiftStaple(Impl::CovShiftBackward(tmp, nu, Impl::CovShiftBackward(U[mu], mu, U[nu])), mu); | ||||
|  | ||||
|     // C3- | ||||
|     tmp = lambda * U[nu]; | ||||
|     out -= Impl::ShiftStaple(Impl::CovShiftBackward(U[nu], nu, Impl::CovShiftBackward(U[mu], mu, tmp)), mu); | ||||
|  | ||||
|     // C4- | ||||
|     out -= Impl::ShiftStaple(Impl::CovShiftBackward(U[nu], nu, Impl::CovShiftBackward(U[mu], mu, U[nu])), mu) * lambda; | ||||
|  | ||||
|     return out; | ||||
|   } | ||||
|  | ||||
|   static CloverField fillCloverYZ(const GaugeLinkField &F) | ||||
|   { | ||||
|     CloverField T(F.Grid()); | ||||
|     T = Zero(); | ||||
|     autoView(T_v,T,AcceleratorWrite); | ||||
|     autoView(F_v,F,AcceleratorRead); | ||||
|     accelerator_for(i, T.Grid()->oSites(),CloverField::vector_type::Nsimd(), | ||||
|     { | ||||
|       coalescedWrite(T_v[i]()(0, 1), coalescedRead(timesMinusI(F_v[i]()()))); | ||||
|       coalescedWrite(T_v[i]()(1, 0), coalescedRead(timesMinusI(F_v[i]()()))); | ||||
|       coalescedWrite(T_v[i]()(2, 3), coalescedRead(timesMinusI(F_v[i]()()))); | ||||
|       coalescedWrite(T_v[i]()(3, 2), coalescedRead(timesMinusI(F_v[i]()()))); | ||||
|     }); | ||||
|  | ||||
|     return T; | ||||
|   } | ||||
|  | ||||
|   static CloverField fillCloverXZ(const GaugeLinkField &F) | ||||
|   { | ||||
|     CloverField T(F.Grid()); | ||||
|     T = Zero(); | ||||
|      | ||||
|     autoView(T_v, T,AcceleratorWrite); | ||||
|     autoView(F_v, F,AcceleratorRead); | ||||
|     accelerator_for(i, T.Grid()->oSites(),CloverField::vector_type::Nsimd(), | ||||
|     { | ||||
|       coalescedWrite(T_v[i]()(0, 1), coalescedRead(-F_v[i]()())); | ||||
|       coalescedWrite(T_v[i]()(1, 0), coalescedRead(F_v[i]()())); | ||||
|       coalescedWrite(T_v[i]()(2, 3), coalescedRead(-F_v[i]()())); | ||||
|       coalescedWrite(T_v[i]()(3, 2), coalescedRead(F_v[i]()())); | ||||
|     }); | ||||
|  | ||||
|     return T; | ||||
|   } | ||||
|  | ||||
|   static CloverField fillCloverXY(const GaugeLinkField &F) | ||||
|   { | ||||
|     CloverField T(F.Grid()); | ||||
|     T = Zero(); | ||||
|  | ||||
|     autoView(T_v,T,AcceleratorWrite); | ||||
|     autoView(F_v,F,AcceleratorRead); | ||||
|     accelerator_for(i, T.Grid()->oSites(),CloverField::vector_type::Nsimd(), | ||||
|     { | ||||
|       coalescedWrite(T_v[i]()(0, 0), coalescedRead(timesMinusI(F_v[i]()()))); | ||||
|       coalescedWrite(T_v[i]()(1, 1), coalescedRead(timesI(F_v[i]()()))); | ||||
|       coalescedWrite(T_v[i]()(2, 2), coalescedRead(timesMinusI(F_v[i]()()))); | ||||
|       coalescedWrite(T_v[i]()(3, 3), coalescedRead(timesI(F_v[i]()()))); | ||||
|     }); | ||||
|  | ||||
|     return T; | ||||
|   } | ||||
|  | ||||
|   static CloverField fillCloverXT(const GaugeLinkField &F) | ||||
|   { | ||||
|     CloverField T(F.Grid()); | ||||
|     T = Zero(); | ||||
|  | ||||
|     autoView( T_v , T, AcceleratorWrite); | ||||
|     autoView( F_v , F, AcceleratorRead); | ||||
|     accelerator_for(i, T.Grid()->oSites(),CloverField::vector_type::Nsimd(), | ||||
|     { | ||||
|       coalescedWrite(T_v[i]()(0, 1), coalescedRead(timesI(F_v[i]()()))); | ||||
|       coalescedWrite(T_v[i]()(1, 0), coalescedRead(timesI(F_v[i]()()))); | ||||
|       coalescedWrite(T_v[i]()(2, 3), coalescedRead(timesMinusI(F_v[i]()()))); | ||||
|       coalescedWrite(T_v[i]()(3, 2), coalescedRead(timesMinusI(F_v[i]()()))); | ||||
|     }); | ||||
|  | ||||
|     return T; | ||||
|   } | ||||
|  | ||||
|   static CloverField fillCloverYT(const GaugeLinkField &F) | ||||
|   { | ||||
|     CloverField T(F.Grid()); | ||||
|     T = Zero(); | ||||
|      | ||||
|     autoView( T_v ,T,AcceleratorWrite); | ||||
|     autoView( F_v ,F,AcceleratorRead); | ||||
|     accelerator_for(i, T.Grid()->oSites(),CloverField::vector_type::Nsimd(), | ||||
|     { | ||||
|       coalescedWrite(T_v[i]()(0, 1), coalescedRead(-(F_v[i]()()))); | ||||
|       coalescedWrite(T_v[i]()(1, 0), coalescedRead((F_v[i]()()))); | ||||
|       coalescedWrite(T_v[i]()(2, 3), coalescedRead((F_v[i]()()))); | ||||
|       coalescedWrite(T_v[i]()(3, 2), coalescedRead(-(F_v[i]()()))); | ||||
|     }); | ||||
|  | ||||
|     return T; | ||||
|   } | ||||
|  | ||||
|   static CloverField fillCloverZT(const GaugeLinkField &F) | ||||
|   { | ||||
|     CloverField T(F.Grid()); | ||||
|  | ||||
|     T = Zero(); | ||||
|  | ||||
|     autoView( T_v , T,AcceleratorWrite); | ||||
|     autoView( F_v , F,AcceleratorRead); | ||||
|     accelerator_for(i, T.Grid()->oSites(),CloverField::vector_type::Nsimd(), | ||||
|     { | ||||
|       coalescedWrite(T_v[i]()(0, 0), coalescedRead(timesI(F_v[i]()()))); | ||||
|       coalescedWrite(T_v[i]()(1, 1), coalescedRead(timesMinusI(F_v[i]()()))); | ||||
|       coalescedWrite(T_v[i]()(2, 2), coalescedRead(timesMinusI(F_v[i]()()))); | ||||
|       coalescedWrite(T_v[i]()(3, 3), coalescedRead(timesI(F_v[i]()()))); | ||||
|     }); | ||||
|  | ||||
|     return T; | ||||
|   } | ||||
|  | ||||
|   template<class _Spinor> | ||||
|   static accelerator_inline void multClover(_Spinor& phi, const SiteClover& C, const _Spinor& chi) { | ||||
|     auto CC = coalescedRead(C); | ||||
|     mult(&phi, &CC, &chi); | ||||
|   } | ||||
|  | ||||
|   template<class _SpinorField> | ||||
|   inline void multCloverField(_SpinorField& out, const CloverField& C, const _SpinorField& phi) { | ||||
|     const int Nsimd = SiteSpinor::Nsimd(); | ||||
|     autoView(out_v, out, AcceleratorWrite); | ||||
|     autoView(phi_v, phi, AcceleratorRead); | ||||
|     autoView(C_v,   C,   AcceleratorRead); | ||||
|     typedef decltype(coalescedRead(out_v[0])) calcSpinor; | ||||
|     accelerator_for(sss,out.Grid()->oSites(),Nsimd,{ | ||||
|       calcSpinor tmp; | ||||
|       multClover(tmp,C_v[sss],phi_v(sss)); | ||||
|       coalescedWrite(out_v[sss],tmp); | ||||
|     }); | ||||
|   } | ||||
| }; | ||||
|  | ||||
|  | ||||
| template<class Impl> class CompactWilsonCloverHelpers { | ||||
| public: | ||||
|  | ||||
|   INHERIT_COMPACT_CLOVER_SIZES(Impl); | ||||
|  | ||||
|   INHERIT_IMPL_TYPES(Impl); | ||||
|   INHERIT_CLOVER_TYPES(Impl); | ||||
|   INHERIT_COMPACT_CLOVER_TYPES(Impl); | ||||
|  | ||||
|   #if 0 | ||||
|   static accelerator_inline typename SiteCloverTriangle::vector_type triangle_elem(const SiteCloverTriangle& triangle, int block, int i, int j) { | ||||
|     assert(i != j); | ||||
|     if(i < j) { | ||||
|       return triangle()(block)(triangle_index(i, j)); | ||||
|     } else { // i > j | ||||
|       return conjugate(triangle()(block)(triangle_index(i, j))); | ||||
|     } | ||||
|   } | ||||
|   #else | ||||
|   template<typename vobj> | ||||
|   static accelerator_inline vobj triangle_elem(const iImplCloverTriangle<vobj>& triangle, int block, int i, int j) { | ||||
|     assert(i != j); | ||||
|     if(i < j) { | ||||
|       return triangle()(block)(triangle_index(i, j)); | ||||
|     } else { // i > j | ||||
|       return conjugate(triangle()(block)(triangle_index(i, j))); | ||||
|     } | ||||
|   } | ||||
|   #endif | ||||
|  | ||||
|   static accelerator_inline int triangle_index(int i, int j) { | ||||
|     if(i == j) | ||||
|       return 0; | ||||
|     else if(i < j) | ||||
|       return Nred * (Nred - 1) / 2 - (Nred - i) * (Nred - i - 1) / 2 + j - i - 1; | ||||
|     else // i > j | ||||
|       return Nred * (Nred - 1) / 2 - (Nred - j) * (Nred - j - 1) / 2 + i - j - 1; | ||||
|   } | ||||
|  | ||||
|   static void MooeeKernel_gpu(int                        Nsite, | ||||
|                               int                        Ls, | ||||
|                               const FermionField&        in, | ||||
|                               FermionField&              out, | ||||
|                               const CloverDiagonalField& diagonal, | ||||
|                               const CloverTriangleField& triangle) { | ||||
|     autoView(diagonal_v, diagonal, AcceleratorRead); | ||||
|     autoView(triangle_v, triangle, AcceleratorRead); | ||||
|     autoView(in_v,       in,       AcceleratorRead); | ||||
|     autoView(out_v,      out,      AcceleratorWrite); | ||||
|  | ||||
|     typedef decltype(coalescedRead(out_v[0])) CalcSpinor; | ||||
|  | ||||
|     const uint64_t NN = Nsite * Ls; | ||||
|  | ||||
|     accelerator_for(ss, NN, Simd::Nsimd(), { | ||||
|       int sF = ss; | ||||
|       int sU = ss/Ls; | ||||
|       CalcSpinor res; | ||||
|       CalcSpinor in_t = in_v(sF); | ||||
|       auto diagonal_t = diagonal_v(sU); | ||||
|       auto triangle_t = triangle_v(sU); | ||||
|       for(int block=0; block<Nhs; block++) { | ||||
|         int s_start = block*Nhs; | ||||
|         for(int i=0; i<Nred; i++) { | ||||
|           int si = s_start + i/Nc, ci = i%Nc; | ||||
|           res()(si)(ci) = diagonal_t()(block)(i) * in_t()(si)(ci); | ||||
|           for(int j=0; j<Nred; j++) { | ||||
|             if (j == i) continue; | ||||
|             int sj = s_start + j/Nc, cj = j%Nc; | ||||
|             res()(si)(ci) = res()(si)(ci) + triangle_elem(triangle_t, block, i, j) * in_t()(sj)(cj); | ||||
|           }; | ||||
|         }; | ||||
|       }; | ||||
|       coalescedWrite(out_v[sF], res); | ||||
|     }); | ||||
|   } | ||||
|  | ||||
|   static void MooeeKernel_cpu(int                        Nsite, | ||||
|                               int                        Ls, | ||||
|                               const FermionField&        in, | ||||
|                               FermionField&              out, | ||||
|                               const CloverDiagonalField& diagonal, | ||||
|                               const CloverTriangleField& triangle) { | ||||
|     autoView(diagonal_v, diagonal, CpuRead); | ||||
|     autoView(triangle_v, triangle, CpuRead); | ||||
|     autoView(in_v,       in,       CpuRead); | ||||
|     autoView(out_v,      out,      CpuWrite); | ||||
|  | ||||
|     typedef SiteSpinor CalcSpinor; | ||||
|  | ||||
| #if defined(A64FX) || defined(A64FXFIXEDSIZE) | ||||
| #define PREFETCH_CLOVER(BASE) {                                     \ | ||||
|     uint64_t base;                                                  \ | ||||
|     int pf_dist_L1 = 1;                                             \ | ||||
|     int pf_dist_L2 = -5; /* -> penalty -> disable */                \ | ||||
|                                                                     \ | ||||
|     if ((pf_dist_L1 >= 0) && (sU + pf_dist_L1 < Nsite)) {           \ | ||||
|       base = (uint64_t)&diag_t()(pf_dist_L1+BASE)(0);               \ | ||||
|       svprfd(svptrue_b64(), (int64_t*)(base +    0), SV_PLDL1STRM); \ | ||||
|       svprfd(svptrue_b64(), (int64_t*)(base +  256), SV_PLDL1STRM); \ | ||||
|       svprfd(svptrue_b64(), (int64_t*)(base +  512), SV_PLDL1STRM); \ | ||||
|       svprfd(svptrue_b64(), (int64_t*)(base +  768), SV_PLDL1STRM); \ | ||||
|       svprfd(svptrue_b64(), (int64_t*)(base + 1024), SV_PLDL1STRM); \ | ||||
|       svprfd(svptrue_b64(), (int64_t*)(base + 1280), SV_PLDL1STRM); \ | ||||
|     }                                                               \ | ||||
|                                                                     \ | ||||
|     if ((pf_dist_L2 >= 0) && (sU + pf_dist_L2 < Nsite)) {           \ | ||||
|       base = (uint64_t)&diag_t()(pf_dist_L2+BASE)(0);               \ | ||||
|       svprfd(svptrue_b64(), (int64_t*)(base +    0), SV_PLDL2STRM); \ | ||||
|       svprfd(svptrue_b64(), (int64_t*)(base +  256), SV_PLDL2STRM); \ | ||||
|       svprfd(svptrue_b64(), (int64_t*)(base +  512), SV_PLDL2STRM); \ | ||||
|       svprfd(svptrue_b64(), (int64_t*)(base +  768), SV_PLDL2STRM); \ | ||||
|       svprfd(svptrue_b64(), (int64_t*)(base + 1024), SV_PLDL2STRM); \ | ||||
|       svprfd(svptrue_b64(), (int64_t*)(base + 1280), SV_PLDL2STRM); \ | ||||
|     }                                                               \ | ||||
|   } | ||||
| // TODO: Implement/generalize this for other architectures | ||||
| // I played around a bit on KNL (see below) but didn't bring anything | ||||
| // #elif defined(AVX512) | ||||
| // #define PREFETCH_CLOVER(BASE) {                              \ | ||||
| //     uint64_t base;                                           \ | ||||
| //     int pf_dist_L1 = 1;                                      \ | ||||
| //     int pf_dist_L2 = +4;                                     \ | ||||
| //                                                              \ | ||||
| //     if ((pf_dist_L1 >= 0) && (sU + pf_dist_L1 < Nsite)) {    \ | ||||
| //       base = (uint64_t)&diag_t()(pf_dist_L1+BASE)(0);        \ | ||||
| //       _mm_prefetch((const char*)(base +    0), _MM_HINT_T0); \ | ||||
| //       _mm_prefetch((const char*)(base +   64), _MM_HINT_T0); \ | ||||
| //       _mm_prefetch((const char*)(base +  128), _MM_HINT_T0); \ | ||||
| //       _mm_prefetch((const char*)(base +  192), _MM_HINT_T0); \ | ||||
| //       _mm_prefetch((const char*)(base +  256), _MM_HINT_T0); \ | ||||
| //       _mm_prefetch((const char*)(base +  320), _MM_HINT_T0); \ | ||||
| //     }                                                        \ | ||||
| //                                                              \ | ||||
| //     if ((pf_dist_L2 >= 0) && (sU + pf_dist_L2 < Nsite)) {    \ | ||||
| //       base = (uint64_t)&diag_t()(pf_dist_L2+BASE)(0);        \ | ||||
| //       _mm_prefetch((const char*)(base +    0), _MM_HINT_T1); \ | ||||
| //       _mm_prefetch((const char*)(base +   64), _MM_HINT_T1); \ | ||||
| //       _mm_prefetch((const char*)(base +  128), _MM_HINT_T1); \ | ||||
| //       _mm_prefetch((const char*)(base +  192), _MM_HINT_T1); \ | ||||
| //       _mm_prefetch((const char*)(base +  256), _MM_HINT_T1); \ | ||||
| //       _mm_prefetch((const char*)(base +  320), _MM_HINT_T1); \ | ||||
| //     }                                                        \ | ||||
| //   } | ||||
| #else | ||||
| #define PREFETCH_CLOVER(BASE) | ||||
| #endif | ||||
|  | ||||
|     const uint64_t NN = Nsite * Ls; | ||||
|  | ||||
|     thread_for(ss, NN, { | ||||
|       int sF = ss; | ||||
|       int sU = ss/Ls; | ||||
|       CalcSpinor res; | ||||
|       CalcSpinor in_t = in_v[sF]; | ||||
|       auto diag_t     = diagonal_v[sU]; // "diag" instead of "diagonal" here to make code below easier to read | ||||
|       auto triangle_t = triangle_v[sU]; | ||||
|  | ||||
|       // upper half | ||||
|       PREFETCH_CLOVER(0); | ||||
|  | ||||
|       auto in_cc_0_0 = conjugate(in_t()(0)(0)); // Nils: reduces number | ||||
|       auto in_cc_0_1 = conjugate(in_t()(0)(1)); // of conjugates from | ||||
|       auto in_cc_0_2 = conjugate(in_t()(0)(2)); // 30 to 20 | ||||
|       auto in_cc_1_0 = conjugate(in_t()(1)(0)); | ||||
|       auto in_cc_1_1 = conjugate(in_t()(1)(1)); | ||||
|  | ||||
|       res()(0)(0) =               diag_t()(0)( 0) * in_t()(0)(0) | ||||
|                   +           triangle_t()(0)( 0) * in_t()(0)(1) | ||||
|                   +           triangle_t()(0)( 1) * in_t()(0)(2) | ||||
|                   +           triangle_t()(0)( 2) * in_t()(1)(0) | ||||
|                   +           triangle_t()(0)( 3) * in_t()(1)(1) | ||||
|                   +           triangle_t()(0)( 4) * in_t()(1)(2); | ||||
|  | ||||
|       res()(0)(1) =           triangle_t()(0)( 0) * in_cc_0_0; | ||||
|       res()(0)(1) =               diag_t()(0)( 1) * in_t()(0)(1) | ||||
|                   +           triangle_t()(0)( 5) * in_t()(0)(2) | ||||
|                   +           triangle_t()(0)( 6) * in_t()(1)(0) | ||||
|                   +           triangle_t()(0)( 7) * in_t()(1)(1) | ||||
|                   +           triangle_t()(0)( 8) * in_t()(1)(2) | ||||
|                   + conjugate(       res()(0)( 1)); | ||||
|  | ||||
|       res()(0)(2) =           triangle_t()(0)( 1) * in_cc_0_0 | ||||
|                   +           triangle_t()(0)( 5) * in_cc_0_1; | ||||
|       res()(0)(2) =               diag_t()(0)( 2) * in_t()(0)(2) | ||||
|                   +           triangle_t()(0)( 9) * in_t()(1)(0) | ||||
|                   +           triangle_t()(0)(10) * in_t()(1)(1) | ||||
|                   +           triangle_t()(0)(11) * in_t()(1)(2) | ||||
|                   + conjugate(       res()(0)( 2)); | ||||
|  | ||||
|       res()(1)(0) =           triangle_t()(0)( 2) * in_cc_0_0 | ||||
|                   +           triangle_t()(0)( 6) * in_cc_0_1 | ||||
|                   +           triangle_t()(0)( 9) * in_cc_0_2; | ||||
|       res()(1)(0) =               diag_t()(0)( 3) * in_t()(1)(0) | ||||
|                   +           triangle_t()(0)(12) * in_t()(1)(1) | ||||
|                   +           triangle_t()(0)(13) * in_t()(1)(2) | ||||
|                   + conjugate(       res()(1)( 0)); | ||||
|  | ||||
|       res()(1)(1) =           triangle_t()(0)( 3) * in_cc_0_0 | ||||
|                   +           triangle_t()(0)( 7) * in_cc_0_1 | ||||
|                   +           triangle_t()(0)(10) * in_cc_0_2 | ||||
|                   +           triangle_t()(0)(12) * in_cc_1_0; | ||||
|       res()(1)(1) =               diag_t()(0)( 4) * in_t()(1)(1) | ||||
|                   +           triangle_t()(0)(14) * in_t()(1)(2) | ||||
|                   + conjugate(       res()(1)( 1)); | ||||
|  | ||||
|       res()(1)(2) =           triangle_t()(0)( 4) * in_cc_0_0 | ||||
|                   +           triangle_t()(0)( 8) * in_cc_0_1 | ||||
|                   +           triangle_t()(0)(11) * in_cc_0_2 | ||||
|                   +           triangle_t()(0)(13) * in_cc_1_0 | ||||
|                   +           triangle_t()(0)(14) * in_cc_1_1; | ||||
|       res()(1)(2) =               diag_t()(0)( 5) * in_t()(1)(2) | ||||
|                   + conjugate(       res()(1)( 2)); | ||||
|  | ||||
|       vstream(out_v[sF]()(0)(0), res()(0)(0)); | ||||
|       vstream(out_v[sF]()(0)(1), res()(0)(1)); | ||||
|       vstream(out_v[sF]()(0)(2), res()(0)(2)); | ||||
|       vstream(out_v[sF]()(1)(0), res()(1)(0)); | ||||
|       vstream(out_v[sF]()(1)(1), res()(1)(1)); | ||||
|       vstream(out_v[sF]()(1)(2), res()(1)(2)); | ||||
|  | ||||
|       // lower half | ||||
|       PREFETCH_CLOVER(1); | ||||
|  | ||||
|       auto in_cc_2_0 = conjugate(in_t()(2)(0)); | ||||
|       auto in_cc_2_1 = conjugate(in_t()(2)(1)); | ||||
|       auto in_cc_2_2 = conjugate(in_t()(2)(2)); | ||||
|       auto in_cc_3_0 = conjugate(in_t()(3)(0)); | ||||
|       auto in_cc_3_1 = conjugate(in_t()(3)(1)); | ||||
|  | ||||
|       res()(2)(0) =               diag_t()(1)( 0) * in_t()(2)(0) | ||||
|                   +           triangle_t()(1)( 0) * in_t()(2)(1) | ||||
|                   +           triangle_t()(1)( 1) * in_t()(2)(2) | ||||
|                   +           triangle_t()(1)( 2) * in_t()(3)(0) | ||||
|                   +           triangle_t()(1)( 3) * in_t()(3)(1) | ||||
|                   +           triangle_t()(1)( 4) * in_t()(3)(2); | ||||
|  | ||||
|       res()(2)(1) =           triangle_t()(1)( 0) * in_cc_2_0; | ||||
|       res()(2)(1) =               diag_t()(1)( 1) * in_t()(2)(1) | ||||
|                   +           triangle_t()(1)( 5) * in_t()(2)(2) | ||||
|                   +           triangle_t()(1)( 6) * in_t()(3)(0) | ||||
|                   +           triangle_t()(1)( 7) * in_t()(3)(1) | ||||
|                   +           triangle_t()(1)( 8) * in_t()(3)(2) | ||||
|                   + conjugate(       res()(2)( 1)); | ||||
|  | ||||
|       res()(2)(2) =           triangle_t()(1)( 1) * in_cc_2_0 | ||||
|                   +           triangle_t()(1)( 5) * in_cc_2_1; | ||||
|       res()(2)(2) =               diag_t()(1)( 2) * in_t()(2)(2) | ||||
|                   +           triangle_t()(1)( 9) * in_t()(3)(0) | ||||
|                   +           triangle_t()(1)(10) * in_t()(3)(1) | ||||
|                   +           triangle_t()(1)(11) * in_t()(3)(2) | ||||
|                   + conjugate(       res()(2)( 2)); | ||||
|  | ||||
|       res()(3)(0) =           triangle_t()(1)( 2) * in_cc_2_0 | ||||
|                   +           triangle_t()(1)( 6) * in_cc_2_1 | ||||
|                   +           triangle_t()(1)( 9) * in_cc_2_2; | ||||
|       res()(3)(0) =               diag_t()(1)( 3) * in_t()(3)(0) | ||||
|                   +           triangle_t()(1)(12) * in_t()(3)(1) | ||||
|                   +           triangle_t()(1)(13) * in_t()(3)(2) | ||||
|                   + conjugate(       res()(3)( 0)); | ||||
|  | ||||
|       res()(3)(1) =           triangle_t()(1)( 3) * in_cc_2_0 | ||||
|                   +           triangle_t()(1)( 7) * in_cc_2_1 | ||||
|                   +           triangle_t()(1)(10) * in_cc_2_2 | ||||
|                   +           triangle_t()(1)(12) * in_cc_3_0; | ||||
|       res()(3)(1) =               diag_t()(1)( 4) * in_t()(3)(1) | ||||
|                   +           triangle_t()(1)(14) * in_t()(3)(2) | ||||
|                   + conjugate(       res()(3)( 1)); | ||||
|  | ||||
|       res()(3)(2) =           triangle_t()(1)( 4) * in_cc_2_0 | ||||
|                   +           triangle_t()(1)( 8) * in_cc_2_1 | ||||
|                   +           triangle_t()(1)(11) * in_cc_2_2 | ||||
|                   +           triangle_t()(1)(13) * in_cc_3_0 | ||||
|                   +           triangle_t()(1)(14) * in_cc_3_1; | ||||
|       res()(3)(2) =               diag_t()(1)( 5) * in_t()(3)(2) | ||||
|                   + conjugate(       res()(3)( 2)); | ||||
|  | ||||
|       vstream(out_v[sF]()(2)(0), res()(2)(0)); | ||||
|       vstream(out_v[sF]()(2)(1), res()(2)(1)); | ||||
|       vstream(out_v[sF]()(2)(2), res()(2)(2)); | ||||
|       vstream(out_v[sF]()(3)(0), res()(3)(0)); | ||||
|       vstream(out_v[sF]()(3)(1), res()(3)(1)); | ||||
|       vstream(out_v[sF]()(3)(2), res()(3)(2)); | ||||
|     }); | ||||
|   } | ||||
|  | ||||
|   static void MooeeKernel(int                        Nsite, | ||||
|                           int                        Ls, | ||||
|                           const FermionField&        in, | ||||
|                           FermionField&              out, | ||||
|                           const CloverDiagonalField& diagonal, | ||||
|                           const CloverTriangleField& triangle) { | ||||
| #if defined(GRID_CUDA) || defined(GRID_HIP) | ||||
|     MooeeKernel_gpu(Nsite, Ls, in, out, diagonal, triangle); | ||||
| #else | ||||
|     MooeeKernel_cpu(Nsite, Ls, in, out, diagonal, triangle); | ||||
| #endif | ||||
|   } | ||||
|  | ||||
|   static void Invert(const CloverDiagonalField& diagonal, | ||||
|                      const CloverTriangleField& triangle, | ||||
|                      CloverDiagonalField&       diagonalInv, | ||||
|                      CloverTriangleField&       triangleInv) { | ||||
|     conformable(diagonal, diagonalInv); | ||||
|     conformable(triangle, triangleInv); | ||||
|     conformable(diagonal, triangle); | ||||
|  | ||||
|     diagonalInv.Checkerboard() = diagonal.Checkerboard(); | ||||
|     triangleInv.Checkerboard() = triangle.Checkerboard(); | ||||
|  | ||||
|     GridBase* grid = diagonal.Grid(); | ||||
|  | ||||
|     long lsites = grid->lSites(); | ||||
|  | ||||
|     typedef typename SiteCloverDiagonal::scalar_object scalar_object_diagonal; | ||||
|     typedef typename SiteCloverTriangle::scalar_object scalar_object_triangle; | ||||
|  | ||||
|     autoView(diagonal_v,  diagonal,  CpuRead); | ||||
|     autoView(triangle_v,  triangle,  CpuRead); | ||||
|     autoView(diagonalInv_v, diagonalInv, CpuWrite); | ||||
|     autoView(triangleInv_v, triangleInv, CpuWrite); | ||||
|  | ||||
|     thread_for(site, lsites, { // NOTE: Not on GPU because of Eigen & (peek/poke)LocalSite | ||||
|       Eigen::MatrixXcd clover_inv_eigen = Eigen::MatrixXcd::Zero(Ns*Nc, Ns*Nc); | ||||
|       Eigen::MatrixXcd clover_eigen = Eigen::MatrixXcd::Zero(Ns*Nc, Ns*Nc); | ||||
|  | ||||
|       scalar_object_diagonal diagonal_tmp     = Zero(); | ||||
|       scalar_object_diagonal diagonal_inv_tmp = Zero(); | ||||
|       scalar_object_triangle triangle_tmp     = Zero(); | ||||
|       scalar_object_triangle triangle_inv_tmp = Zero(); | ||||
|  | ||||
|       Coordinate lcoor; | ||||
|       grid->LocalIndexToLocalCoor(site, lcoor); | ||||
|  | ||||
|       peekLocalSite(diagonal_tmp, diagonal_v, lcoor); | ||||
|       peekLocalSite(triangle_tmp, triangle_v, lcoor); | ||||
|  | ||||
|       // TODO: can we save time here by inverting the two 6x6 hermitian matrices separately? | ||||
|       for (long s_row=0;s_row<Ns;s_row++) { | ||||
|         for (long s_col=0;s_col<Ns;s_col++) { | ||||
|           if(abs(s_row - s_col) > 1 || s_row + s_col == 3) continue; | ||||
|           int block       = s_row / Nhs; | ||||
|           int s_row_block = s_row % Nhs; | ||||
|           int s_col_block = s_col % Nhs; | ||||
|           for (long c_row=0;c_row<Nc;c_row++) { | ||||
|             for (long c_col=0;c_col<Nc;c_col++) { | ||||
|               int i = s_row_block * Nc + c_row; | ||||
|               int j = s_col_block * Nc + c_col; | ||||
|               if(i == j) | ||||
|                 clover_eigen(s_row*Nc+c_row, s_col*Nc+c_col) = static_cast<ComplexD>(TensorRemove(diagonal_tmp()(block)(i))); | ||||
|               else | ||||
|                 clover_eigen(s_row*Nc+c_row, s_col*Nc+c_col) = static_cast<ComplexD>(TensorRemove(triangle_elem(triangle_tmp, block, i, j))); | ||||
|             } | ||||
|           } | ||||
|         } | ||||
|       } | ||||
|  | ||||
|       clover_inv_eigen = clover_eigen.inverse(); | ||||
|  | ||||
|       for (long s_row=0;s_row<Ns;s_row++) { | ||||
|         for (long s_col=0;s_col<Ns;s_col++) { | ||||
|           if(abs(s_row - s_col) > 1 || s_row + s_col == 3) continue; | ||||
|           int block       = s_row / Nhs; | ||||
|           int s_row_block = s_row % Nhs; | ||||
|           int s_col_block = s_col % Nhs; | ||||
|           for (long c_row=0;c_row<Nc;c_row++) { | ||||
|             for (long c_col=0;c_col<Nc;c_col++) { | ||||
|               int i = s_row_block * Nc + c_row; | ||||
|               int j = s_col_block * Nc + c_col; | ||||
|               if(i == j) | ||||
|                 diagonal_inv_tmp()(block)(i) = clover_inv_eigen(s_row*Nc+c_row, s_col*Nc+c_col); | ||||
|               else if(i < j) | ||||
|                 triangle_inv_tmp()(block)(triangle_index(i, j)) = clover_inv_eigen(s_row*Nc+c_row, s_col*Nc+c_col); | ||||
|               else | ||||
|                 continue; | ||||
|             } | ||||
|           } | ||||
|         } | ||||
|       } | ||||
|  | ||||
|       pokeLocalSite(diagonal_inv_tmp, diagonalInv_v, lcoor); | ||||
|       pokeLocalSite(triangle_inv_tmp, triangleInv_v, lcoor); | ||||
|     }); | ||||
|   } | ||||
|  | ||||
|   static void ConvertLayout(const CloverField&   full, | ||||
|                             CloverDiagonalField& diagonal, | ||||
|                             CloverTriangleField& triangle) { | ||||
|     conformable(full, diagonal); | ||||
|     conformable(full, triangle); | ||||
|  | ||||
|     diagonal.Checkerboard() = full.Checkerboard(); | ||||
|     triangle.Checkerboard() = full.Checkerboard(); | ||||
|  | ||||
|     autoView(full_v,     full,     AcceleratorRead); | ||||
|     autoView(diagonal_v, diagonal, AcceleratorWrite); | ||||
|     autoView(triangle_v, triangle, AcceleratorWrite); | ||||
|  | ||||
|     // NOTE: this function cannot be 'private' since nvcc forbids this for kernels | ||||
|     accelerator_for(ss, full.Grid()->oSites(), 1, { | ||||
|       for(int s_row = 0; s_row < Ns; s_row++) { | ||||
|         for(int s_col = 0; s_col < Ns; s_col++) { | ||||
|           if(abs(s_row - s_col) > 1 || s_row + s_col == 3) continue; | ||||
|           int block       = s_row / Nhs; | ||||
|           int s_row_block = s_row % Nhs; | ||||
|           int s_col_block = s_col % Nhs; | ||||
|           for(int c_row = 0; c_row < Nc; c_row++) { | ||||
|             for(int c_col = 0; c_col < Nc; c_col++) { | ||||
|               int i = s_row_block * Nc + c_row; | ||||
|               int j = s_col_block * Nc + c_col; | ||||
|               if(i == j) | ||||
|                 diagonal_v[ss]()(block)(i) = full_v[ss]()(s_row, s_col)(c_row, c_col); | ||||
|               else if(i < j) | ||||
|                 triangle_v[ss]()(block)(triangle_index(i, j)) = full_v[ss]()(s_row, s_col)(c_row, c_col); | ||||
|               else | ||||
|                 continue; | ||||
|             } | ||||
|           } | ||||
|         } | ||||
|       } | ||||
|     }); | ||||
|   } | ||||
|  | ||||
|  | ||||
|   static void ConvertLayout(const CloverDiagonalField& diagonal, | ||||
|                             const CloverTriangleField& triangle, | ||||
|                             CloverField&               full) { | ||||
|     conformable(full, diagonal); | ||||
|     conformable(full, triangle); | ||||
|  | ||||
|     full.Checkerboard() = diagonal.Checkerboard(); | ||||
|  | ||||
|     full = Zero(); | ||||
|  | ||||
|     autoView(diagonal_v, diagonal, AcceleratorRead); | ||||
|     autoView(triangle_v, triangle, AcceleratorRead); | ||||
|     autoView(full_v,     full,     AcceleratorWrite); | ||||
|  | ||||
|     // NOTE: this function cannot be 'private' since nvcc forbids this for kernels | ||||
|     accelerator_for(ss, full.Grid()->oSites(), 1, { | ||||
|       for(int s_row = 0; s_row < Ns; s_row++) { | ||||
|         for(int s_col = 0; s_col < Ns; s_col++) { | ||||
|           if(abs(s_row - s_col) > 1 || s_row + s_col == 3) continue; | ||||
|           int block       = s_row / Nhs; | ||||
|           int s_row_block = s_row % Nhs; | ||||
|           int s_col_block = s_col % Nhs; | ||||
|           for(int c_row = 0; c_row < Nc; c_row++) { | ||||
|             for(int c_col = 0; c_col < Nc; c_col++) { | ||||
|               int i = s_row_block * Nc + c_row; | ||||
|               int j = s_col_block * Nc + c_col; | ||||
|               if(i == j) | ||||
|                 full_v[ss]()(s_row, s_col)(c_row, c_col) = diagonal_v[ss]()(block)(i); | ||||
|               else | ||||
|                 full_v[ss]()(s_row, s_col)(c_row, c_col) = triangle_elem(triangle_v[ss], block, i, j); | ||||
|             } | ||||
|           } | ||||
|         } | ||||
|       } | ||||
|     }); | ||||
|   } | ||||
|  | ||||
|   static void ModifyBoundaries(CloverDiagonalField& diagonal, CloverTriangleField& triangle, RealD csw_t, RealD cF, RealD diag_mass) { | ||||
|     // Checks/grid | ||||
|     double t0 = usecond(); | ||||
|     conformable(diagonal, triangle); | ||||
|     GridBase* grid = diagonal.Grid(); | ||||
|  | ||||
|     // Determine the boundary coordinates/sites | ||||
|     double t1 = usecond(); | ||||
|     int t_dir = Nd - 1; | ||||
|     Lattice<iScalar<vInteger>> t_coor(grid); | ||||
|     LatticeCoordinate(t_coor, t_dir); | ||||
|     int T = grid->GlobalDimensions()[t_dir]; | ||||
|  | ||||
|     // Set off-diagonal parts at boundary to zero -- OK | ||||
|     double t2 = usecond(); | ||||
|     CloverTriangleField zeroTriangle(grid); | ||||
|     zeroTriangle.Checkerboard() = triangle.Checkerboard(); | ||||
|     zeroTriangle = Zero(); | ||||
|     triangle = where(t_coor == 0,   zeroTriangle, triangle); | ||||
|     triangle = where(t_coor == T-1, zeroTriangle, triangle); | ||||
|  | ||||
|     // Set diagonal to unity (scaled correctly) -- OK | ||||
|     double t3 = usecond(); | ||||
|     CloverDiagonalField tmp(grid); | ||||
|     tmp.Checkerboard() = diagonal.Checkerboard(); | ||||
|     tmp                = -1.0 * csw_t + diag_mass; | ||||
|     diagonal           = where(t_coor == 0,   tmp, diagonal); | ||||
|     diagonal           = where(t_coor == T-1, tmp, diagonal); | ||||
|  | ||||
|     // Correct values next to boundary | ||||
|     double t4 = usecond(); | ||||
|     if(cF != 1.0) { | ||||
|       tmp = cF - 1.0; | ||||
|       tmp += diagonal; | ||||
|       diagonal = where(t_coor == 1,   tmp, diagonal); | ||||
|       diagonal = where(t_coor == T-2, tmp, diagonal); | ||||
|     } | ||||
|  | ||||
|     // Report timings | ||||
|     double t5 = usecond(); | ||||
| #if 0 | ||||
|     std::cout << GridLogMessage << "CompactWilsonCloverHelpers::ModifyBoundaries timings:" | ||||
|               << " checks = "          << (t1 - t0) / 1e6 | ||||
|               << ", coordinate = "     << (t2 - t1) / 1e6 | ||||
|               << ", off-diag zero = "  << (t3 - t2) / 1e6 | ||||
|               << ", diagonal unity = " << (t4 - t3) / 1e6 | ||||
|               << ", near-boundary = "  << (t5 - t4) / 1e6 | ||||
|               << ", total = "          << (t5 - t0) / 1e6 | ||||
|               << std::endl; | ||||
| #endif | ||||
|   } | ||||
|  | ||||
|   template<class Field, class Mask> | ||||
|   static strong_inline void ApplyBoundaryMask(Field& f, const Mask& m) { | ||||
|     conformable(f, m); | ||||
|     auto grid  = f.Grid(); | ||||
|     const uint32_t Nsite = grid->oSites(); | ||||
|     const uint32_t Nsimd = grid->Nsimd(); | ||||
|     autoView(f_v, f, AcceleratorWrite); | ||||
|     autoView(m_v, m, AcceleratorRead); | ||||
|     // NOTE: this function cannot be 'private' since nvcc forbids this for kernels | ||||
|     accelerator_for(ss, Nsite, Nsimd, { | ||||
|       coalescedWrite(f_v[ss], m_v(ss) * f_v(ss)); | ||||
|     }); | ||||
|   } | ||||
|  | ||||
|   template<class MaskField> | ||||
|   static void SetupMasks(MaskField& full, MaskField& even, MaskField& odd) { | ||||
|     assert(even.Grid()->_isCheckerBoarded && even.Checkerboard() == Even); | ||||
|     assert(odd.Grid()->_isCheckerBoarded  && odd.Checkerboard()  == Odd); | ||||
|     assert(!full.Grid()->_isCheckerBoarded); | ||||
|  | ||||
|     GridBase* grid = full.Grid(); | ||||
|     int t_dir = Nd-1; | ||||
|     Lattice<iScalar<vInteger>> t_coor(grid); | ||||
|     LatticeCoordinate(t_coor, t_dir); | ||||
|     int T = grid->GlobalDimensions()[t_dir]; | ||||
|  | ||||
|     MaskField zeroMask(grid); zeroMask = Zero(); | ||||
|     full = 1.0; | ||||
|     full = where(t_coor == 0,   zeroMask, full); | ||||
|     full = where(t_coor == T-1, zeroMask, full); | ||||
|  | ||||
|     pickCheckerboard(Even, even, full); | ||||
|     pickCheckerboard(Odd,  odd,  full); | ||||
|   } | ||||
| }; | ||||
|  | ||||
| NAMESPACE_END(Grid); | ||||
| @@ -1,92 +0,0 @@ | ||||
| /************************************************************************************* | ||||
|  | ||||
|     Grid physics library, www.github.com/paboyle/Grid | ||||
|  | ||||
|     Source file: ./lib/qcd/action/fermion/WilsonCloverTypes.h | ||||
|  | ||||
|     Copyright (C) 2021 - 2022 | ||||
|  | ||||
|     Author: Daniel Richtmann <daniel.richtmann@gmail.com> | ||||
|  | ||||
|     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 */ | ||||
|  | ||||
| #pragma once | ||||
|  | ||||
| NAMESPACE_BEGIN(Grid); | ||||
|  | ||||
| template<class Impl> | ||||
| class WilsonCloverTypes { | ||||
| public: | ||||
|   INHERIT_IMPL_TYPES(Impl); | ||||
|  | ||||
|   template <typename vtype> using iImplClover = iScalar<iMatrix<iMatrix<vtype, Impl::Dimension>, Ns>>; | ||||
|  | ||||
|   typedef iImplClover<Simd> SiteClover; | ||||
|  | ||||
|   typedef Lattice<SiteClover> CloverField; | ||||
| }; | ||||
|  | ||||
| template<class Impl> | ||||
| class CompactWilsonCloverTypes { | ||||
| public: | ||||
|   INHERIT_IMPL_TYPES(Impl); | ||||
|  | ||||
|   static_assert(Nd == 4 && Nc == 3 && Ns == 4 && Impl::Dimension == 3, "Wrong dimensions"); | ||||
|  | ||||
|   static constexpr int Nred      = Nc * Nhs;        // 6 | ||||
|   static constexpr int Nblock    = Nhs;             // 2 | ||||
|   static constexpr int Ndiagonal = Nred;            // 6 | ||||
|   static constexpr int Ntriangle = (Nred - 1) * Nc; // 15 | ||||
|  | ||||
|   template<typename vtype> using iImplCloverDiagonal = iScalar<iVector<iVector<vtype, Ndiagonal>, Nblock>>; | ||||
|   template<typename vtype> using iImplCloverTriangle = iScalar<iVector<iVector<vtype, Ntriangle>, Nblock>>; | ||||
|  | ||||
|   typedef iImplCloverDiagonal<Simd> SiteCloverDiagonal; | ||||
|   typedef iImplCloverTriangle<Simd> SiteCloverTriangle; | ||||
|   typedef iSinglet<Simd>            SiteMask; | ||||
|  | ||||
|   typedef Lattice<SiteCloverDiagonal> CloverDiagonalField; | ||||
|   typedef Lattice<SiteCloverTriangle> CloverTriangleField; | ||||
|   typedef Lattice<SiteMask>           MaskField; | ||||
| }; | ||||
|  | ||||
| #define INHERIT_CLOVER_TYPES(Impl)                                 \ | ||||
|   typedef typename WilsonCloverTypes<Impl>::SiteClover SiteClover; \ | ||||
|   typedef typename WilsonCloverTypes<Impl>::CloverField CloverField; | ||||
|  | ||||
| #define INHERIT_COMPACT_CLOVER_TYPES(Impl) \ | ||||
|   typedef typename CompactWilsonCloverTypes<Impl>::SiteCloverDiagonal  SiteCloverDiagonal; \ | ||||
|   typedef typename CompactWilsonCloverTypes<Impl>::SiteCloverTriangle  SiteCloverTriangle; \ | ||||
|   typedef typename CompactWilsonCloverTypes<Impl>::SiteMask            SiteMask; \ | ||||
|   typedef typename CompactWilsonCloverTypes<Impl>::CloverDiagonalField CloverDiagonalField; \ | ||||
|   typedef typename CompactWilsonCloverTypes<Impl>::CloverTriangleField CloverTriangleField; \ | ||||
|   typedef typename CompactWilsonCloverTypes<Impl>::MaskField           MaskField; \ | ||||
|   /* ugly duplication but needed inside functionality classes */ \ | ||||
|   template<typename vtype> using iImplCloverDiagonal = \ | ||||
|     iScalar<iVector<iVector<vtype, CompactWilsonCloverTypes<Impl>::Ndiagonal>, CompactWilsonCloverTypes<Impl>::Nblock>>; \ | ||||
|   template<typename vtype> using iImplCloverTriangle = \ | ||||
|     iScalar<iVector<iVector<vtype, CompactWilsonCloverTypes<Impl>::Ntriangle>, CompactWilsonCloverTypes<Impl>::Nblock>>; | ||||
|  | ||||
| #define INHERIT_COMPACT_CLOVER_SIZES(Impl)                                    \ | ||||
|   static constexpr int Nred      = CompactWilsonCloverTypes<Impl>::Nred;      \ | ||||
|   static constexpr int Nblock    = CompactWilsonCloverTypes<Impl>::Nblock;    \ | ||||
|   static constexpr int Ndiagonal = CompactWilsonCloverTypes<Impl>::Ndiagonal; \ | ||||
|   static constexpr int Ntriangle = CompactWilsonCloverTypes<Impl>::Ntriangle; | ||||
|  | ||||
| NAMESPACE_END(Grid); | ||||
| @@ -68,12 +68,11 @@ public: | ||||
|   /*****************************************************/ | ||||
|   /* Compress includes precision change if mpi data is not same */ | ||||
|   /*****************************************************/ | ||||
|   accelerator_inline void Compress(SiteHalfSpinor &buf,const SiteSpinor &in) const { | ||||
|     typedef decltype(coalescedRead(buf)) sobj; | ||||
|     sobj sp; | ||||
|     auto sin = coalescedRead(in); | ||||
|     projector::Proj(sp,sin,mu,dag); | ||||
|     coalescedWrite(buf,sp); | ||||
|   template<class _SiteHalfSpinor, class _SiteSpinor> | ||||
|   accelerator_inline void Compress(_SiteHalfSpinor *buf,Integer o,const _SiteSpinor &in) const { | ||||
|     _SiteHalfSpinor tmp; | ||||
|     projector::Proj(tmp,in,mu,dag); | ||||
|     vstream(buf[o],tmp); | ||||
|   } | ||||
|  | ||||
|   /*****************************************************/ | ||||
| @@ -83,18 +82,13 @@ public: | ||||
| 				   const SiteHalfSpinor * __restrict__ vp0, | ||||
| 				   const SiteHalfSpinor * __restrict__ vp1, | ||||
| 				   Integer type,Integer o) const { | ||||
| #ifdef GRID_SIMT | ||||
|     exchangeSIMT(mp[2*o],mp[2*o+1],vp0[o],vp1[o],type); | ||||
| #else | ||||
|     SiteHalfSpinor tmp1; | ||||
|     SiteHalfSpinor tmp2; | ||||
|     exchange(tmp1,tmp2,vp0[o],vp1[o],type); | ||||
|     vstream(mp[2*o  ],tmp1); | ||||
|     vstream(mp[2*o+1],tmp2); | ||||
| #endif | ||||
|   } | ||||
|  | ||||
|  | ||||
|   /*****************************************************/ | ||||
|   /* Have a decompression step if mpi data is not same */ | ||||
|   /*****************************************************/ | ||||
| @@ -111,28 +105,6 @@ public: | ||||
| 					   const SiteSpinor * __restrict__ in, | ||||
| 					   Integer j,Integer k, Integer m,Integer type) const | ||||
|   { | ||||
| #ifdef GRID_SIMT | ||||
|     typedef SiteSpinor vobj; | ||||
|     typedef SiteHalfSpinor hvobj; | ||||
|     typedef decltype(coalescedRead(*in))    sobj; | ||||
|     typedef decltype(coalescedRead(*out0)) hsobj; | ||||
|  | ||||
|     unsigned int Nsimd = vobj::Nsimd(); | ||||
|     unsigned int mask = Nsimd >> (type + 1); | ||||
|     int lane = acceleratorSIMTlane(Nsimd); | ||||
|     int j0 = lane &(~mask); // inner coor zero | ||||
|     int j1 = lane |(mask) ; // inner coor one | ||||
|     const vobj *vp0 = &in[k]; | ||||
|     const vobj *vp1 = &in[m]; | ||||
|     const vobj *vp = (lane&mask) ? vp1:vp0; | ||||
|     auto sa = coalescedRead(*vp,j0); | ||||
|     auto sb = coalescedRead(*vp,j1); | ||||
|     hsobj psa, psb; | ||||
|     projector::Proj(psa,sa,mu,dag); | ||||
|     projector::Proj(psb,sb,mu,dag); | ||||
|     coalescedWrite(out0[j],psa); | ||||
|     coalescedWrite(out1[j],psb); | ||||
| #else | ||||
|     SiteHalfSpinor temp1, temp2; | ||||
|     SiteHalfSpinor temp3, temp4; | ||||
|     projector::Proj(temp1,in[k],mu,dag); | ||||
| @@ -140,7 +112,6 @@ public: | ||||
|     exchange(temp3,temp4,temp1,temp2,type); | ||||
|     vstream(out0[j],temp3); | ||||
|     vstream(out1[j],temp4); | ||||
| #endif | ||||
|   } | ||||
|  | ||||
|   /*****************************************************/ | ||||
| @@ -150,7 +121,6 @@ public: | ||||
|  | ||||
| }; | ||||
|  | ||||
| #if 0 | ||||
| template<class _HCspinor,class _Hspinor,class _Spinor, class projector> | ||||
| class WilsonCompressorTemplate< _HCspinor, _Hspinor, _Spinor, projector, | ||||
| 				typename std::enable_if<!std::is_same<_HCspinor,_Hspinor>::value>::type > | ||||
| @@ -179,23 +149,13 @@ public: | ||||
|   /*****************************************************/ | ||||
|   /* Compress includes precision change if mpi data is not same */ | ||||
|   /*****************************************************/ | ||||
|   accelerator_inline void Compress(SiteHalfSpinor &buf,const SiteSpinor &in) const { | ||||
|     SiteHalfSpinor hsp; | ||||
|   template<class _SiteHalfSpinor, class _SiteSpinor> | ||||
|   accelerator_inline void Compress(_SiteHalfSpinor *buf,Integer o,const _SiteSpinor &in) const { | ||||
|     _SiteHalfSpinor hsp; | ||||
|     SiteHalfCommSpinor *hbuf = (SiteHalfCommSpinor *)buf; | ||||
|     projector::Proj(hsp,in,mu,dag); | ||||
|     precisionChange((vComplexLow *)&hbuf[o],(vComplexHigh *)&hsp,Nw); | ||||
|   } | ||||
|   accelerator_inline void Compress(SiteHalfSpinor &buf,const SiteSpinor &in) const { | ||||
| #ifdef GRID_SIMT | ||||
|     typedef decltype(coalescedRead(buf)) sobj; | ||||
|     sobj sp; | ||||
|     auto sin = coalescedRead(in); | ||||
|     projector::Proj(sp,sin,mu,dag); | ||||
|     coalescedWrite(buf,sp); | ||||
| #else | ||||
|     projector::Proj(buf,in,mu,dag); | ||||
| #endif | ||||
|   } | ||||
|  | ||||
|   /*****************************************************/ | ||||
|   /* Exchange includes precision change if mpi data is not same */ | ||||
| @@ -243,7 +203,6 @@ public: | ||||
|   accelerator_inline bool DecompressionStep(void) const { return true; } | ||||
|  | ||||
| }; | ||||
| #endif | ||||
|  | ||||
| #define DECLARE_PROJ(Projector,Compressor,spProj)			\ | ||||
|   class Projector {							\ | ||||
| @@ -294,8 +253,33 @@ public: | ||||
|   typedef typename Base::View_type View_type; | ||||
|   typedef typename Base::StencilVector StencilVector; | ||||
|  | ||||
|   void ZeroCountersi(void)  {  } | ||||
|   void Reporti(int calls)  {  } | ||||
|   double timer0; | ||||
|   double timer1; | ||||
|   double timer2; | ||||
|   double timer3; | ||||
|   double timer4; | ||||
|   double timer5; | ||||
|   double timer6; | ||||
|   uint64_t callsi; | ||||
|   void ZeroCountersi(void) | ||||
|   { | ||||
|     timer0=0; | ||||
|     timer1=0; | ||||
|     timer2=0; | ||||
|     timer3=0; | ||||
|     timer4=0; | ||||
|     timer5=0; | ||||
|     timer6=0; | ||||
|     callsi=0; | ||||
|   } | ||||
|   void Reporti(int calls) | ||||
|   { | ||||
|     if ( timer0 ) std::cout << GridLogMessage << " timer0 (HaloGatherOpt) " <<timer0/calls <<std::endl; | ||||
|     if ( timer1 ) std::cout << GridLogMessage << " timer1 (Communicate)   " <<timer1/calls <<std::endl; | ||||
|     if ( timer2 ) std::cout << GridLogMessage << " timer2 (CommsMerge )   " <<timer2/calls <<std::endl; | ||||
|     if ( timer3 ) std::cout << GridLogMessage << " timer3 (commsMergeShm) " <<timer3/calls <<std::endl; | ||||
|     if ( timer4 ) std::cout << GridLogMessage << " timer4 " <<timer4 <<std::endl; | ||||
|   } | ||||
|  | ||||
|   std::vector<int> surface_list; | ||||
|  | ||||
| @@ -337,18 +321,26 @@ public: | ||||
|   { | ||||
|     std::vector<std::vector<CommsRequest_t> > reqs; | ||||
|     this->HaloExchangeOptGather(source,compress); | ||||
|     double t1=usecond(); | ||||
|     // Asynchronous MPI calls multidirectional, Isend etc... | ||||
|     // Non-overlapped directions within a thread. Asynchronous calls except MPI3, threaded up to comm threads ways. | ||||
|     this->Communicate(); | ||||
|     double t2=usecond(); timer1 += t2-t1; | ||||
|     this->CommsMerge(compress); | ||||
|     double t3=usecond(); timer2 += t3-t2; | ||||
|     this->CommsMergeSHM(compress); | ||||
|     double t4=usecond(); timer3 += t4-t3; | ||||
|   } | ||||
|    | ||||
|   template <class compressor> | ||||
|   void HaloExchangeOptGather(const Lattice<vobj> &source,compressor &compress)  | ||||
|   { | ||||
|     this->Prepare(); | ||||
|     double t0=usecond(); | ||||
|     this->HaloGatherOpt(source,compress); | ||||
|     double t1=usecond(); | ||||
|     timer0 += t1-t0; | ||||
|     callsi++; | ||||
|   } | ||||
|  | ||||
|   template <class compressor> | ||||
| @@ -360,9 +352,12 @@ public: | ||||
|     typedef typename compressor::SiteHalfSpinor     SiteHalfSpinor; | ||||
|     typedef typename compressor::SiteHalfCommSpinor SiteHalfCommSpinor; | ||||
|  | ||||
|     this->mpi3synctime_g-=usecond(); | ||||
|     this->_grid->StencilBarrier(); | ||||
|     this->mpi3synctime_g+=usecond(); | ||||
|  | ||||
|     assert(source.Grid()==this->_grid); | ||||
|     this->halogtime-=usecond(); | ||||
|      | ||||
|     this->u_comm_offset=0; | ||||
|        | ||||
| @@ -398,6 +393,7 @@ public: | ||||
|     } | ||||
|     this->face_table_computed=1; | ||||
|     assert(this->u_comm_offset==this->_unified_buffer_size); | ||||
|     this->halogtime+=usecond(); | ||||
|     accelerator_barrier(); | ||||
|   } | ||||
|  | ||||
|   | ||||
| @@ -243,17 +243,17 @@ typedef WilsonImpl<vComplex,  FundamentalRepresentation, CoeffReal > WilsonImplR | ||||
| typedef WilsonImpl<vComplexF, FundamentalRepresentation, CoeffReal > WilsonImplF;  // Float | ||||
| typedef WilsonImpl<vComplexD, FundamentalRepresentation, CoeffReal > WilsonImplD;  // Double | ||||
|  | ||||
| //typedef WilsonImpl<vComplex,  FundamentalRepresentation, CoeffRealHalfComms > WilsonImplRL;  // Real.. whichever prec | ||||
| //typedef WilsonImpl<vComplexF, FundamentalRepresentation, CoeffRealHalfComms > WilsonImplFH;  // Float | ||||
| //typedef WilsonImpl<vComplexD, FundamentalRepresentation, CoeffRealHalfComms > WilsonImplDF;  // Double | ||||
| typedef WilsonImpl<vComplex,  FundamentalRepresentation, CoeffRealHalfComms > WilsonImplRL;  // Real.. whichever prec | ||||
| typedef WilsonImpl<vComplexF, FundamentalRepresentation, CoeffRealHalfComms > WilsonImplFH;  // Float | ||||
| typedef WilsonImpl<vComplexD, FundamentalRepresentation, CoeffRealHalfComms > WilsonImplDF;  // Double | ||||
|  | ||||
| typedef WilsonImpl<vComplex,  FundamentalRepresentation, CoeffComplex > ZWilsonImplR; // Real.. whichever prec | ||||
| typedef WilsonImpl<vComplexF, FundamentalRepresentation, CoeffComplex > ZWilsonImplF; // Float | ||||
| typedef WilsonImpl<vComplexD, FundamentalRepresentation, CoeffComplex > ZWilsonImplD; // Double | ||||
|  | ||||
| //typedef WilsonImpl<vComplex,  FundamentalRepresentation, CoeffComplexHalfComms > ZWilsonImplRL; // Real.. whichever prec | ||||
| //typedef WilsonImpl<vComplexF, FundamentalRepresentation, CoeffComplexHalfComms > ZWilsonImplFH; // Float | ||||
| //typedef WilsonImpl<vComplexD, FundamentalRepresentation, CoeffComplexHalfComms > ZWilsonImplDF; // Double | ||||
| typedef WilsonImpl<vComplex,  FundamentalRepresentation, CoeffComplexHalfComms > ZWilsonImplRL; // Real.. whichever prec | ||||
| typedef WilsonImpl<vComplexF, FundamentalRepresentation, CoeffComplexHalfComms > ZWilsonImplFH; // Float | ||||
| typedef WilsonImpl<vComplexD, FundamentalRepresentation, CoeffComplexHalfComms > ZWilsonImplDF; // Double | ||||
|   | ||||
| typedef WilsonImpl<vComplex,  AdjointRepresentation, CoeffReal > WilsonAdjImplR;   // Real.. whichever prec | ||||
| typedef WilsonImpl<vComplexF, AdjointRepresentation, CoeffReal > WilsonAdjImplF;  // Float | ||||
|   | ||||
| @@ -828,7 +828,6 @@ void CayleyFermion5D<Impl>::SeqConservedCurrent(PropagatorField &q_in, | ||||
|  | ||||
| #if (!defined(GRID_HIP)) | ||||
|   int tshift = (mu == Nd-1) ? 1 : 0; | ||||
|   unsigned int LLt    = GridDefaultLatt()[Tp]; | ||||
|   //////////////////////////////////////////////// | ||||
|   // GENERAL CAYLEY CASE | ||||
|   //////////////////////////////////////////////// | ||||
| @@ -881,7 +880,7 @@ void CayleyFermion5D<Impl>::SeqConservedCurrent(PropagatorField &q_in, | ||||
|   } | ||||
|  | ||||
|   std::vector<RealD> G_s(Ls,1.0); | ||||
|   RealD sign = 1.0; // sign flip for vector/tadpole | ||||
|   Integer sign = 1; // sign flip for vector/tadpole | ||||
|   if ( curr_type == Current::Axial ) { | ||||
|     for(int s=0;s<Ls/2;s++){ | ||||
|       G_s[s] = -1.0; | ||||
| @@ -891,7 +890,7 @@ void CayleyFermion5D<Impl>::SeqConservedCurrent(PropagatorField &q_in, | ||||
|     auto b=this->_b; | ||||
|     auto c=this->_c; | ||||
|     if ( b == 1 && c == 0 ) { | ||||
|       sign = -1.0;     | ||||
|       sign = -1;     | ||||
|     } | ||||
|     else { | ||||
|       std::cerr << "Error: Tadpole implementation currently unavailable for non-Shamir actions." << std::endl; | ||||
| @@ -902,8 +901,8 @@ void CayleyFermion5D<Impl>::SeqConservedCurrent(PropagatorField &q_in, | ||||
|   for(int s=0;s<Ls;s++){ | ||||
|  | ||||
|     int sp = (s+1)%Ls; | ||||
|     //    int sr = Ls-1-s; | ||||
|     //    int srp= (sr+1)%Ls; | ||||
|     int sr = Ls-1-s; | ||||
|     int srp= (sr+1)%Ls; | ||||
|  | ||||
|     // Mobius parameters | ||||
|     auto b=this->bs[s]; | ||||
| @@ -935,13 +934,7 @@ void CayleyFermion5D<Impl>::SeqConservedCurrent(PropagatorField &q_in, | ||||
|     tmp    = Cshift(tmp,mu,-1); | ||||
|     Impl::multLinkField(Utmp,this->Umu,tmp,mu+Nd); // Adjoint link | ||||
|     tmp = -G_s[s]*( Utmp + gmu*Utmp ); | ||||
|     // Mask the time | ||||
|     if (tmax == LLt - 1 && tshift == 1){ // quick fix to include timeslice 0 if tmax + tshift is over the last timeslice | ||||
|       unsigned int t0 = 0; | ||||
|       tmp    = where(((lcoor==t0) || (lcoor>=tmin+tshift)),tmp,zz); | ||||
|     } else { | ||||
|       tmp    = where((lcoor>=tmin+tshift),tmp,zz); | ||||
|     } | ||||
|     tmp    = where((lcoor>=tmin+tshift),tmp,zz); // Mask the time  | ||||
|     L_Q   += where((lcoor<=tmax+tshift),tmp,zz); // Position of current complicated | ||||
|  | ||||
|     InsertSlice(L_Q, q_out, s , 0); | ||||
|   | ||||
| @@ -1,363 +0,0 @@ | ||||
| /************************************************************************************* | ||||
|  | ||||
|     Grid physics library, www.github.com/paboyle/Grid | ||||
|  | ||||
|     Source file: ./lib/qcd/action/fermion/CompactWilsonCloverFermionImplementation.h | ||||
|  | ||||
|     Copyright (C) 2017 - 2022 | ||||
|  | ||||
|     Author: paboyle <paboyle@ph.ed.ac.uk> | ||||
|     Author: Guido Cossu <guido.cossu@ed.ac.uk> | ||||
|     Author: Daniel Richtmann <daniel.richtmann@gmail.com> | ||||
|  | ||||
|     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/spin/Dirac.h> | ||||
| #include <Grid/qcd/action/fermion/CompactWilsonCloverFermion.h> | ||||
|  | ||||
| NAMESPACE_BEGIN(Grid); | ||||
| template<class Impl> | ||||
| CompactWilsonCloverFermion<Impl>::CompactWilsonCloverFermion(GaugeField& _Umu, | ||||
|                                                              GridCartesian& Fgrid, | ||||
|                                                              GridRedBlackCartesian& Hgrid, | ||||
|                                                              const RealD _mass, | ||||
|                                                              const RealD _csw_r, | ||||
|                                                              const RealD _csw_t, | ||||
|                                                              const RealD _cF, | ||||
|                                                              const WilsonAnisotropyCoefficients& clover_anisotropy, | ||||
|                                                              const ImplParams& impl_p) | ||||
|   : WilsonBase(_Umu, Fgrid, Hgrid, _mass, impl_p, clover_anisotropy) | ||||
|   , csw_r(_csw_r) | ||||
|   , csw_t(_csw_t) | ||||
|   , cF(_cF) | ||||
|   , open_boundaries(impl_p.boundary_phases[Nd-1] == 0.0) | ||||
|   , Diagonal(&Fgrid),        Triangle(&Fgrid) | ||||
|   , DiagonalEven(&Hgrid),    TriangleEven(&Hgrid) | ||||
|   , DiagonalOdd(&Hgrid),     TriangleOdd(&Hgrid) | ||||
|   , DiagonalInv(&Fgrid),     TriangleInv(&Fgrid) | ||||
|   , DiagonalInvEven(&Hgrid), TriangleInvEven(&Hgrid) | ||||
|   , DiagonalInvOdd(&Hgrid),  TriangleInvOdd(&Hgrid) | ||||
|   , Tmp(&Fgrid) | ||||
|   , BoundaryMask(&Fgrid) | ||||
|   , BoundaryMaskEven(&Hgrid), BoundaryMaskOdd(&Hgrid) | ||||
| { | ||||
|   csw_r *= 0.5; | ||||
|   csw_t *= 0.5; | ||||
|   if (clover_anisotropy.isAnisotropic) | ||||
|     csw_r /= clover_anisotropy.xi_0; | ||||
|  | ||||
|   ImportGauge(_Umu); | ||||
|   if (open_boundaries) | ||||
|     CompactHelpers::SetupMasks(this->BoundaryMask, this->BoundaryMaskEven, this->BoundaryMaskOdd); | ||||
| } | ||||
|  | ||||
| template<class Impl> | ||||
| void CompactWilsonCloverFermion<Impl>::Dhop(const FermionField& in, FermionField& out, int dag) { | ||||
|   WilsonBase::Dhop(in, out, dag); | ||||
|   if(open_boundaries) ApplyBoundaryMask(out); | ||||
| } | ||||
|  | ||||
| template<class Impl> | ||||
| void CompactWilsonCloverFermion<Impl>::DhopOE(const FermionField& in, FermionField& out, int dag) { | ||||
|   WilsonBase::DhopOE(in, out, dag); | ||||
|   if(open_boundaries) ApplyBoundaryMask(out); | ||||
| } | ||||
|  | ||||
| template<class Impl> | ||||
| void CompactWilsonCloverFermion<Impl>::DhopEO(const FermionField& in, FermionField& out, int dag) { | ||||
|   WilsonBase::DhopEO(in, out, dag); | ||||
|   if(open_boundaries) ApplyBoundaryMask(out); | ||||
| } | ||||
|  | ||||
| template<class Impl> | ||||
| void CompactWilsonCloverFermion<Impl>::DhopDir(const FermionField& in, FermionField& out, int dir, int disp) { | ||||
|   WilsonBase::DhopDir(in, out, dir, disp); | ||||
|   if(this->open_boundaries) ApplyBoundaryMask(out); | ||||
| } | ||||
|  | ||||
| template<class Impl> | ||||
| void CompactWilsonCloverFermion<Impl>::DhopDirAll(const FermionField& in, std::vector<FermionField>& out) { | ||||
|   WilsonBase::DhopDirAll(in, out); | ||||
|   if(this->open_boundaries) { | ||||
|     for(auto& o : out) ApplyBoundaryMask(o); | ||||
|   } | ||||
| } | ||||
|  | ||||
| template<class Impl> | ||||
| void CompactWilsonCloverFermion<Impl>::M(const FermionField& in, FermionField& out) { | ||||
|   out.Checkerboard() = in.Checkerboard(); | ||||
|   WilsonBase::Dhop(in, out, DaggerNo); // call base to save applying bc | ||||
|   Mooee(in, Tmp); | ||||
|   axpy(out, 1.0, out, Tmp); | ||||
|   if(open_boundaries) ApplyBoundaryMask(out); | ||||
| } | ||||
|  | ||||
| template<class Impl> | ||||
| void CompactWilsonCloverFermion<Impl>::Mdag(const FermionField& in, FermionField& out) { | ||||
|   out.Checkerboard() = in.Checkerboard(); | ||||
|   WilsonBase::Dhop(in, out, DaggerYes);  // call base to save applying bc | ||||
|   MooeeDag(in, Tmp); | ||||
|   axpy(out, 1.0, out, Tmp); | ||||
|   if(open_boundaries) ApplyBoundaryMask(out); | ||||
| } | ||||
|  | ||||
| template<class Impl> | ||||
| void CompactWilsonCloverFermion<Impl>::Meooe(const FermionField& in, FermionField& out) { | ||||
|   WilsonBase::Meooe(in, out); | ||||
|   if(open_boundaries) ApplyBoundaryMask(out); | ||||
| } | ||||
|  | ||||
| template<class Impl> | ||||
| void CompactWilsonCloverFermion<Impl>::MeooeDag(const FermionField& in, FermionField& out) { | ||||
|   WilsonBase::MeooeDag(in, out); | ||||
|   if(open_boundaries) ApplyBoundaryMask(out); | ||||
| } | ||||
|  | ||||
| template<class Impl> | ||||
| void CompactWilsonCloverFermion<Impl>::Mooee(const FermionField& in, FermionField& out) { | ||||
|   if(in.Grid()->_isCheckerBoarded) { | ||||
|     if(in.Checkerboard() == Odd) { | ||||
|       MooeeInternal(in, out, DiagonalOdd, TriangleOdd); | ||||
|     } else { | ||||
|       MooeeInternal(in, out, DiagonalEven, TriangleEven); | ||||
|     } | ||||
|   } else { | ||||
|     MooeeInternal(in, out, Diagonal, Triangle); | ||||
|   } | ||||
|   if(open_boundaries) ApplyBoundaryMask(out); | ||||
| } | ||||
|  | ||||
| template<class Impl> | ||||
| void CompactWilsonCloverFermion<Impl>::MooeeDag(const FermionField& in, FermionField& out) { | ||||
|   Mooee(in, out); // blocks are hermitian | ||||
| } | ||||
|  | ||||
| template<class Impl> | ||||
| void CompactWilsonCloverFermion<Impl>::MooeeInv(const FermionField& in, FermionField& out) { | ||||
|   if(in.Grid()->_isCheckerBoarded) { | ||||
|     if(in.Checkerboard() == Odd) { | ||||
|       MooeeInternal(in, out, DiagonalInvOdd, TriangleInvOdd); | ||||
|     } else { | ||||
|       MooeeInternal(in, out, DiagonalInvEven, TriangleInvEven); | ||||
|     } | ||||
|   } else { | ||||
|     MooeeInternal(in, out, DiagonalInv, TriangleInv); | ||||
|   } | ||||
|   if(open_boundaries) ApplyBoundaryMask(out); | ||||
| } | ||||
|  | ||||
| template<class Impl> | ||||
| void CompactWilsonCloverFermion<Impl>::MooeeInvDag(const FermionField& in, FermionField& out) { | ||||
|   MooeeInv(in, out); // blocks are hermitian | ||||
| } | ||||
|  | ||||
| template<class Impl> | ||||
| void CompactWilsonCloverFermion<Impl>::Mdir(const FermionField& in, FermionField& out, int dir, int disp) { | ||||
|   DhopDir(in, out, dir, disp); | ||||
| } | ||||
|  | ||||
| template<class Impl> | ||||
| void CompactWilsonCloverFermion<Impl>::MdirAll(const FermionField& in, std::vector<FermionField>& out) { | ||||
|   DhopDirAll(in, out); | ||||
| } | ||||
|  | ||||
| template<class Impl> | ||||
| void CompactWilsonCloverFermion<Impl>::MDeriv(GaugeField& force, const FermionField& X, const FermionField& Y, int dag) { | ||||
|   assert(!open_boundaries); // TODO check for changes required for open bc | ||||
|  | ||||
|   // NOTE: code copied from original clover term | ||||
|   conformable(X.Grid(), Y.Grid()); | ||||
|   conformable(X.Grid(), force.Grid()); | ||||
|   GaugeLinkField force_mu(force.Grid()), lambda(force.Grid()); | ||||
|   GaugeField clover_force(force.Grid()); | ||||
|   PropagatorField Lambda(force.Grid()); | ||||
|  | ||||
|   // Guido: Here we are hitting some performance issues: | ||||
|   // need to extract the components of the DoubledGaugeField | ||||
|   // for each call | ||||
|   // Possible solution | ||||
|   // Create a vector object to store them? (cons: wasting space) | ||||
|   std::vector<GaugeLinkField> U(Nd, this->Umu.Grid()); | ||||
|  | ||||
|   Impl::extractLinkField(U, this->Umu); | ||||
|  | ||||
|   force = Zero(); | ||||
|   // Derivative of the Wilson hopping term | ||||
|   this->DhopDeriv(force, X, Y, dag); | ||||
|  | ||||
|   /////////////////////////////////////////////////////////// | ||||
|   // Clover term derivative | ||||
|   /////////////////////////////////////////////////////////// | ||||
|   Impl::outerProductImpl(Lambda, X, Y); | ||||
|   //std::cout << "Lambda:" << Lambda << std::endl; | ||||
|  | ||||
|   Gamma::Algebra sigma[] = { | ||||
|       Gamma::Algebra::SigmaXY, | ||||
|       Gamma::Algebra::SigmaXZ, | ||||
|       Gamma::Algebra::SigmaXT, | ||||
|       Gamma::Algebra::MinusSigmaXY, | ||||
|       Gamma::Algebra::SigmaYZ, | ||||
|       Gamma::Algebra::SigmaYT, | ||||
|       Gamma::Algebra::MinusSigmaXZ, | ||||
|       Gamma::Algebra::MinusSigmaYZ, | ||||
|       Gamma::Algebra::SigmaZT, | ||||
|       Gamma::Algebra::MinusSigmaXT, | ||||
|       Gamma::Algebra::MinusSigmaYT, | ||||
|       Gamma::Algebra::MinusSigmaZT}; | ||||
|  | ||||
|   /* | ||||
|     sigma_{\mu \nu}= | ||||
|     | 0         sigma[0]  sigma[1]  sigma[2] | | ||||
|     | sigma[3]    0       sigma[4]  sigma[5] | | ||||
|     | sigma[6]  sigma[7]     0      sigma[8] | | ||||
|     | sigma[9]  sigma[10] sigma[11]   0      | | ||||
|   */ | ||||
|  | ||||
|   int count = 0; | ||||
|   clover_force = Zero(); | ||||
|   for (int mu = 0; mu < 4; mu++) | ||||
|   { | ||||
|     force_mu = Zero(); | ||||
|     for (int nu = 0; nu < 4; nu++) | ||||
|     { | ||||
|       if (mu == nu) | ||||
|         continue; | ||||
|  | ||||
|       RealD factor; | ||||
|       if (nu == 4 || mu == 4) | ||||
|       { | ||||
|         factor = 2.0 * csw_t; | ||||
|       } | ||||
|       else | ||||
|       { | ||||
|         factor = 2.0 * csw_r; | ||||
|       } | ||||
|       PropagatorField Slambda = Gamma(sigma[count]) * Lambda; // sigma checked | ||||
|       Impl::TraceSpinImpl(lambda, Slambda);                   // traceSpin ok | ||||
|       force_mu -= factor*Helpers::Cmunu(U, lambda, mu, nu);   // checked | ||||
|       count++; | ||||
|     } | ||||
|  | ||||
|     pokeLorentz(clover_force, U[mu] * force_mu, mu); | ||||
|   } | ||||
|   //clover_force *= csw; | ||||
|   force += clover_force; | ||||
| } | ||||
|  | ||||
| template<class Impl> | ||||
| void CompactWilsonCloverFermion<Impl>::MooDeriv(GaugeField& mat, const FermionField& U, const FermionField& V, int dag) { | ||||
|   assert(0); | ||||
| } | ||||
|  | ||||
| template<class Impl> | ||||
| void CompactWilsonCloverFermion<Impl>::MeeDeriv(GaugeField& mat, const FermionField& U, const FermionField& V, int dag) { | ||||
|   assert(0); | ||||
| } | ||||
|  | ||||
| template<class Impl> | ||||
| void CompactWilsonCloverFermion<Impl>::MooeeInternal(const FermionField&        in, | ||||
|                     FermionField&              out, | ||||
|                     const CloverDiagonalField& diagonal, | ||||
|                     const CloverTriangleField& triangle) { | ||||
|   assert(in.Checkerboard() == Odd || in.Checkerboard() == Even); | ||||
|   out.Checkerboard() = in.Checkerboard(); | ||||
|   conformable(in, out); | ||||
|   conformable(in, diagonal); | ||||
|   conformable(in, triangle); | ||||
|  | ||||
|   CompactHelpers::MooeeKernel(diagonal.oSites(), 1, in, out, diagonal, triangle); | ||||
| } | ||||
|  | ||||
| template<class Impl> | ||||
| void CompactWilsonCloverFermion<Impl>::ImportGauge(const GaugeField& _Umu) { | ||||
|   // NOTE: parts copied from original implementation | ||||
|  | ||||
|   // Import gauge into base class | ||||
|   double t0 = usecond(); | ||||
|   WilsonBase::ImportGauge(_Umu); // NOTE: called here and in wilson constructor -> performed twice, but can't avoid that | ||||
|  | ||||
|   // Initialize temporary variables | ||||
|   double t1 = usecond(); | ||||
|   conformable(_Umu.Grid(), this->GaugeGrid()); | ||||
|   GridBase* grid = _Umu.Grid(); | ||||
|   typename Impl::GaugeLinkField Bx(grid), By(grid), Bz(grid), Ex(grid), Ey(grid), Ez(grid); | ||||
|   CloverField TmpOriginal(grid); | ||||
|  | ||||
|   // Compute the field strength terms mu>nu | ||||
|   double t2 = usecond(); | ||||
|   WilsonLoops<Impl>::FieldStrength(Bx, _Umu, Zdir, Ydir); | ||||
|   WilsonLoops<Impl>::FieldStrength(By, _Umu, Zdir, Xdir); | ||||
|   WilsonLoops<Impl>::FieldStrength(Bz, _Umu, Ydir, Xdir); | ||||
|   WilsonLoops<Impl>::FieldStrength(Ex, _Umu, Tdir, Xdir); | ||||
|   WilsonLoops<Impl>::FieldStrength(Ey, _Umu, Tdir, Ydir); | ||||
|   WilsonLoops<Impl>::FieldStrength(Ez, _Umu, Tdir, Zdir); | ||||
|  | ||||
|   // Compute the Clover Operator acting on Colour and Spin | ||||
|   // multiply here by the clover coefficients for the anisotropy | ||||
|   double t3 = usecond(); | ||||
|   TmpOriginal  = Helpers::fillCloverYZ(Bx) * csw_r; | ||||
|   TmpOriginal += Helpers::fillCloverXZ(By) * csw_r; | ||||
|   TmpOriginal += Helpers::fillCloverXY(Bz) * csw_r; | ||||
|   TmpOriginal += Helpers::fillCloverXT(Ex) * csw_t; | ||||
|   TmpOriginal += Helpers::fillCloverYT(Ey) * csw_t; | ||||
|   TmpOriginal += Helpers::fillCloverZT(Ez) * csw_t; | ||||
|   TmpOriginal += this->diag_mass; | ||||
|  | ||||
|   // Convert the data layout of the clover term | ||||
|   double t4 = usecond(); | ||||
|   CompactHelpers::ConvertLayout(TmpOriginal, Diagonal, Triangle); | ||||
|  | ||||
|   // Possible modify the boundary values | ||||
|   double t5 = usecond(); | ||||
|   if(open_boundaries) CompactHelpers::ModifyBoundaries(Diagonal, Triangle, csw_t, cF, this->diag_mass); | ||||
|  | ||||
|   // Invert the clover term in the improved layout | ||||
|   double t6 = usecond(); | ||||
|   CompactHelpers::Invert(Diagonal, Triangle, DiagonalInv, TriangleInv); | ||||
|  | ||||
|   // Fill the remaining clover fields | ||||
|   double t7 = usecond(); | ||||
|   pickCheckerboard(Even, DiagonalEven,    Diagonal); | ||||
|   pickCheckerboard(Even, TriangleEven,    Triangle); | ||||
|   pickCheckerboard(Odd,  DiagonalOdd,     Diagonal); | ||||
|   pickCheckerboard(Odd,  TriangleOdd,     Triangle); | ||||
|   pickCheckerboard(Even, DiagonalInvEven, DiagonalInv); | ||||
|   pickCheckerboard(Even, TriangleInvEven, TriangleInv); | ||||
|   pickCheckerboard(Odd,  DiagonalInvOdd,  DiagonalInv); | ||||
|   pickCheckerboard(Odd,  TriangleInvOdd,  TriangleInv); | ||||
|  | ||||
|   // Report timings | ||||
|   double t8 = usecond(); | ||||
| #if 0 | ||||
|   std::cout << GridLogMessage << "CompactWilsonCloverFermion::ImportGauge timings:" | ||||
|             << " WilsonFermion::Importgauge = " << (t1 - t0) / 1e6 | ||||
|             << ", allocations = "               << (t2 - t1) / 1e6 | ||||
|             << ", field strength = "            << (t3 - t2) / 1e6 | ||||
|             << ", fill clover = "               << (t4 - t3) / 1e6 | ||||
|             << ", convert = "                   << (t5 - t4) / 1e6 | ||||
|             << ", boundaries = "                << (t6 - t5) / 1e6 | ||||
|             << ", inversions = "                << (t7 - t6) / 1e6 | ||||
|             << ", pick cbs = "                  << (t8 - t7) / 1e6 | ||||
|             << ", total = "                     << (t8 - t0) / 1e6 | ||||
|             << std::endl; | ||||
| #endif | ||||
| } | ||||
|  | ||||
| NAMESPACE_END(Grid); | ||||
| @@ -2,13 +2,12 @@ | ||||
|  | ||||
|     Grid physics library, www.github.com/paboyle/Grid | ||||
|  | ||||
|     Source file: ./lib/qcd/action/fermion/WilsonCloverFermionImplementation.h | ||||
|     Source file: ./lib/qcd/action/fermion/WilsonCloverFermion.cc | ||||
|  | ||||
|     Copyright (C) 2017 - 2022 | ||||
|     Copyright (C) 2017 | ||||
|  | ||||
|     Author: paboyle <paboyle@ph.ed.ac.uk> | ||||
|     Author: Guido Cossu <guido.cossu@ed.ac.uk> | ||||
|     Author: Daniel Richtmann <daniel.richtmann@gmail.com> | ||||
|  | ||||
|     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 | ||||
| @@ -34,45 +33,6 @@ | ||||
|  | ||||
| NAMESPACE_BEGIN(Grid); | ||||
|  | ||||
| template<class Impl> | ||||
| WilsonCloverFermion<Impl>::WilsonCloverFermion(GaugeField&                         _Umu, | ||||
|                                                GridCartesian&                      Fgrid, | ||||
|                                                GridRedBlackCartesian&              Hgrid, | ||||
|                                                const RealD                         _mass, | ||||
|                                                const RealD                         _csw_r, | ||||
|                                                const RealD                         _csw_t, | ||||
|                                                const WilsonAnisotropyCoefficients& clover_anisotropy, | ||||
|                                                const ImplParams&                   impl_p) | ||||
|   : WilsonFermion<Impl>(_Umu, Fgrid, Hgrid, _mass, impl_p, clover_anisotropy) | ||||
|   , CloverTerm(&Fgrid) | ||||
|   , CloverTermInv(&Fgrid) | ||||
|   , CloverTermEven(&Hgrid) | ||||
|   , CloverTermOdd(&Hgrid) | ||||
|   , CloverTermInvEven(&Hgrid) | ||||
|   , CloverTermInvOdd(&Hgrid) | ||||
|   , CloverTermDagEven(&Hgrid) | ||||
|   , CloverTermDagOdd(&Hgrid) | ||||
|   , CloverTermInvDagEven(&Hgrid) | ||||
|   , CloverTermInvDagOdd(&Hgrid) { | ||||
|   assert(Nd == 4); // require 4 dimensions | ||||
|  | ||||
|   if(clover_anisotropy.isAnisotropic) { | ||||
|     csw_r     = _csw_r * 0.5 / clover_anisotropy.xi_0; | ||||
|     diag_mass = _mass + 1.0 + (Nd - 1) * (clover_anisotropy.nu / clover_anisotropy.xi_0); | ||||
|   } else { | ||||
|     csw_r     = _csw_r * 0.5; | ||||
|     diag_mass = 4.0 + _mass; | ||||
|   } | ||||
|   csw_t = _csw_t * 0.5; | ||||
|  | ||||
|   if(csw_r == 0) | ||||
|     std::cout << GridLogWarning << "Initializing WilsonCloverFermion with csw_r = 0" << std::endl; | ||||
|   if(csw_t == 0) | ||||
|     std::cout << GridLogWarning << "Initializing WilsonCloverFermion with csw_t = 0" << std::endl; | ||||
|  | ||||
|   ImportGauge(_Umu); | ||||
| } | ||||
|  | ||||
| // *NOT* EO | ||||
| template <class Impl> | ||||
| void WilsonCloverFermion<Impl>::M(const FermionField &in, FermionField &out) | ||||
| @@ -107,13 +67,10 @@ void WilsonCloverFermion<Impl>::Mdag(const FermionField &in, FermionField &out) | ||||
| template <class Impl> | ||||
| void WilsonCloverFermion<Impl>::ImportGauge(const GaugeField &_Umu) | ||||
| { | ||||
|   double t0 = usecond(); | ||||
|   WilsonFermion<Impl>::ImportGauge(_Umu); | ||||
|   double t1 = usecond(); | ||||
|   GridBase *grid = _Umu.Grid(); | ||||
|   typename Impl::GaugeLinkField Bx(grid), By(grid), Bz(grid), Ex(grid), Ey(grid), Ez(grid); | ||||
|  | ||||
|   double t2 = usecond(); | ||||
|   // Compute the field strength terms mu>nu | ||||
|   WilsonLoops<Impl>::FieldStrength(Bx, _Umu, Zdir, Ydir); | ||||
|   WilsonLoops<Impl>::FieldStrength(By, _Umu, Zdir, Xdir); | ||||
| @@ -122,22 +79,19 @@ void WilsonCloverFermion<Impl>::ImportGauge(const GaugeField &_Umu) | ||||
|   WilsonLoops<Impl>::FieldStrength(Ey, _Umu, Tdir, Ydir); | ||||
|   WilsonLoops<Impl>::FieldStrength(Ez, _Umu, Tdir, Zdir); | ||||
|  | ||||
|   double t3 = usecond(); | ||||
|   // Compute the Clover Operator acting on Colour and Spin | ||||
|   // multiply here by the clover coefficients for the anisotropy | ||||
|   CloverTerm  = Helpers::fillCloverYZ(Bx) * csw_r; | ||||
|   CloverTerm += Helpers::fillCloverXZ(By) * csw_r; | ||||
|   CloverTerm += Helpers::fillCloverXY(Bz) * csw_r; | ||||
|   CloverTerm += Helpers::fillCloverXT(Ex) * csw_t; | ||||
|   CloverTerm += Helpers::fillCloverYT(Ey) * csw_t; | ||||
|   CloverTerm += Helpers::fillCloverZT(Ez) * csw_t; | ||||
|   CloverTerm  = fillCloverYZ(Bx) * csw_r; | ||||
|   CloverTerm += fillCloverXZ(By) * csw_r; | ||||
|   CloverTerm += fillCloverXY(Bz) * csw_r; | ||||
|   CloverTerm += fillCloverXT(Ex) * csw_t; | ||||
|   CloverTerm += fillCloverYT(Ey) * csw_t; | ||||
|   CloverTerm += fillCloverZT(Ez) * csw_t; | ||||
|   CloverTerm += diag_mass; | ||||
|  | ||||
|   double t4 = usecond(); | ||||
|   int lvol = _Umu.Grid()->lSites(); | ||||
|   int DimRep = Impl::Dimension; | ||||
|  | ||||
|   double t5 = usecond(); | ||||
|   { | ||||
|     autoView(CTv,CloverTerm,CpuRead); | ||||
|     autoView(CTIv,CloverTermInv,CpuWrite); | ||||
| @@ -146,7 +100,7 @@ void WilsonCloverFermion<Impl>::ImportGauge(const GaugeField &_Umu) | ||||
|       grid->LocalIndexToLocalCoor(site, lcoor); | ||||
|       Eigen::MatrixXcd EigenCloverOp = Eigen::MatrixXcd::Zero(Ns * DimRep, Ns * DimRep); | ||||
|       Eigen::MatrixXcd EigenInvCloverOp = Eigen::MatrixXcd::Zero(Ns * DimRep, Ns * DimRep); | ||||
|       typename SiteClover::scalar_object Qx = Zero(), Qxinv = Zero(); | ||||
|       typename SiteCloverType::scalar_object Qx = Zero(), Qxinv = Zero(); | ||||
|       peekLocalSite(Qx, CTv, lcoor); | ||||
|       //if (csw!=0){ | ||||
|       for (int j = 0; j < Ns; j++) | ||||
| @@ -171,7 +125,6 @@ void WilsonCloverFermion<Impl>::ImportGauge(const GaugeField &_Umu) | ||||
|     }); | ||||
|   } | ||||
|  | ||||
|   double t6 = usecond(); | ||||
|   // Separate the even and odd parts | ||||
|   pickCheckerboard(Even, CloverTermEven, CloverTerm); | ||||
|   pickCheckerboard(Odd, CloverTermOdd, CloverTerm); | ||||
| @@ -184,20 +137,6 @@ void WilsonCloverFermion<Impl>::ImportGauge(const GaugeField &_Umu) | ||||
|  | ||||
|   pickCheckerboard(Even, CloverTermInvDagEven, adj(CloverTermInv)); | ||||
|   pickCheckerboard(Odd, CloverTermInvDagOdd, adj(CloverTermInv)); | ||||
|   double t7 = usecond(); | ||||
|  | ||||
| #if 0 | ||||
|   std::cout << GridLogMessage << "WilsonCloverFermion::ImportGauge timings:" | ||||
|             << " WilsonFermion::Importgauge = " << (t1 - t0) / 1e6 | ||||
|             << ", allocations = "               << (t2 - t1) / 1e6 | ||||
|             << ", field strength = "            << (t3 - t2) / 1e6 | ||||
|             << ", fill clover = "               << (t4 - t3) / 1e6 | ||||
|             << ", misc = "                      << (t5 - t4) / 1e6 | ||||
|             << ", inversions = "                << (t6 - t5) / 1e6 | ||||
|             << ", pick cbs = "                  << (t7 - t6) / 1e6 | ||||
|             << ", total = "                     << (t7 - t0) / 1e6 | ||||
|             << std::endl; | ||||
| #endif | ||||
| } | ||||
|  | ||||
| template <class Impl> | ||||
| @@ -228,7 +167,7 @@ template <class Impl> | ||||
| void WilsonCloverFermion<Impl>::MooeeInternal(const FermionField &in, FermionField &out, int dag, int inv) | ||||
| { | ||||
|   out.Checkerboard() = in.Checkerboard(); | ||||
|   CloverField *Clover; | ||||
|   CloverFieldType *Clover; | ||||
|   assert(in.Checkerboard() == Odd || in.Checkerboard() == Even); | ||||
|  | ||||
|   if (dag) | ||||
| @@ -243,12 +182,12 @@ void WilsonCloverFermion<Impl>::MooeeInternal(const FermionField &in, FermionFie | ||||
|       { | ||||
|         Clover = (inv) ? &CloverTermInvDagEven : &CloverTermDagEven; | ||||
|       } | ||||
|       Helpers::multCloverField(out, *Clover, in); | ||||
|       out = *Clover * in; | ||||
|     } | ||||
|     else | ||||
|     { | ||||
|       Clover = (inv) ? &CloverTermInv : &CloverTerm; | ||||
|       Helpers::multCloverField(out, *Clover, in); // don't bother with adj, hermitian anyway | ||||
|       out = adj(*Clover) * in; | ||||
|     } | ||||
|   } | ||||
|   else | ||||
| @@ -266,98 +205,18 @@ void WilsonCloverFermion<Impl>::MooeeInternal(const FermionField &in, FermionFie | ||||
|         //  std::cout << "Calling clover term Even" << std::endl; | ||||
|         Clover = (inv) ? &CloverTermInvEven : &CloverTermEven; | ||||
|       } | ||||
|       Helpers::multCloverField(out, *Clover, in); | ||||
|       out = *Clover * in; | ||||
|       //  std::cout << GridLogMessage << "*Clover.Checkerboard() "  << (*Clover).Checkerboard() << std::endl; | ||||
|     } | ||||
|     else | ||||
|     { | ||||
|       Clover = (inv) ? &CloverTermInv : &CloverTerm; | ||||
|       Helpers::multCloverField(out, *Clover, in); | ||||
|       out = *Clover * in; | ||||
|     } | ||||
|   } | ||||
|  | ||||
| } // MooeeInternal | ||||
|  | ||||
| // Derivative parts unpreconditioned pseudofermions | ||||
| template <class Impl> | ||||
| void WilsonCloverFermion<Impl>::MDeriv(GaugeField &force, const FermionField &X, const FermionField &Y, int dag) | ||||
| { | ||||
|   conformable(X.Grid(), Y.Grid()); | ||||
|   conformable(X.Grid(), force.Grid()); | ||||
|   GaugeLinkField force_mu(force.Grid()), lambda(force.Grid()); | ||||
|   GaugeField clover_force(force.Grid()); | ||||
|   PropagatorField Lambda(force.Grid()); | ||||
|  | ||||
|   // Guido: Here we are hitting some performance issues: | ||||
|   // need to extract the components of the DoubledGaugeField | ||||
|   // for each call | ||||
|   // Possible solution | ||||
|   // Create a vector object to store them? (cons: wasting space) | ||||
|   std::vector<GaugeLinkField> U(Nd, this->Umu.Grid()); | ||||
|  | ||||
|   Impl::extractLinkField(U, this->Umu); | ||||
|  | ||||
|   force = Zero(); | ||||
|   // Derivative of the Wilson hopping term | ||||
|   this->DhopDeriv(force, X, Y, dag); | ||||
|  | ||||
|   /////////////////////////////////////////////////////////// | ||||
|   // Clover term derivative | ||||
|   /////////////////////////////////////////////////////////// | ||||
|   Impl::outerProductImpl(Lambda, X, Y); | ||||
|   //std::cout << "Lambda:" << Lambda << std::endl; | ||||
|  | ||||
|   Gamma::Algebra sigma[] = { | ||||
|       Gamma::Algebra::SigmaXY, | ||||
|       Gamma::Algebra::SigmaXZ, | ||||
|       Gamma::Algebra::SigmaXT, | ||||
|       Gamma::Algebra::MinusSigmaXY, | ||||
|       Gamma::Algebra::SigmaYZ, | ||||
|       Gamma::Algebra::SigmaYT, | ||||
|       Gamma::Algebra::MinusSigmaXZ, | ||||
|       Gamma::Algebra::MinusSigmaYZ, | ||||
|       Gamma::Algebra::SigmaZT, | ||||
|       Gamma::Algebra::MinusSigmaXT, | ||||
|       Gamma::Algebra::MinusSigmaYT, | ||||
|       Gamma::Algebra::MinusSigmaZT}; | ||||
|  | ||||
|   /* | ||||
|     sigma_{\mu \nu}= | ||||
|     | 0         sigma[0]  sigma[1]  sigma[2] | | ||||
|     | sigma[3]    0       sigma[4]  sigma[5] | | ||||
|     | sigma[6]  sigma[7]     0      sigma[8] | | ||||
|     | sigma[9]  sigma[10] sigma[11]   0      | | ||||
|   */ | ||||
|  | ||||
|   int count = 0; | ||||
|   clover_force = Zero(); | ||||
|   for (int mu = 0; mu < 4; mu++) | ||||
|   { | ||||
|     force_mu = Zero(); | ||||
|     for (int nu = 0; nu < 4; nu++) | ||||
|     { | ||||
|       if (mu == nu) | ||||
|       continue; | ||||
|  | ||||
|       RealD factor; | ||||
|       if (nu == 4 || mu == 4) | ||||
|       { | ||||
|         factor = 2.0 * csw_t; | ||||
|       } | ||||
|       else | ||||
|       { | ||||
|         factor = 2.0 * csw_r; | ||||
|       } | ||||
|       PropagatorField Slambda = Gamma(sigma[count]) * Lambda; // sigma checked | ||||
|       Impl::TraceSpinImpl(lambda, Slambda);                   // traceSpin ok | ||||
|       force_mu -= factor*Helpers::Cmunu(U, lambda, mu, nu);                   // checked | ||||
|       count++; | ||||
|     } | ||||
|  | ||||
|     pokeLorentz(clover_force, U[mu] * force_mu, mu); | ||||
|   } | ||||
|   //clover_force *= csw; | ||||
|   force += clover_force; | ||||
| } | ||||
|  | ||||
| // Derivative parts | ||||
| template <class Impl> | ||||
|   | ||||
| @@ -73,17 +73,17 @@ WilsonKernels<ZWilsonImplF>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
| //#pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| //template<> void | ||||
| //WilsonKernels<WilsonImplFH>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| template<> void | ||||
| WilsonKernels<WilsonImplFH>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
| //#pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| //template<> void | ||||
| //WilsonKernels<ZWilsonImplFH>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| template<> void | ||||
| WilsonKernels<ZWilsonImplFH>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
|  | ||||
| #undef INTERIOR_AND_EXTERIOR | ||||
| @@ -102,17 +102,17 @@ WilsonKernels<ZWilsonImplF>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldVi | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
| //#pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| //template<> void | ||||
| //WilsonKernels<WilsonImplFH>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| template<> void | ||||
| WilsonKernels<WilsonImplFH>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
| //#pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| //template<> void | ||||
| //WilsonKernels<ZWilsonImplFH>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| template<> void | ||||
| WilsonKernels<ZWilsonImplFH>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
|  | ||||
| #undef INTERIOR_AND_EXTERIOR | ||||
| @@ -131,17 +131,17 @@ WilsonKernels<ZWilsonImplF>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldVi | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
| //#pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| //template<> void | ||||
| //WilsonKernels<WilsonImplFH>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| template<> void | ||||
| WilsonKernels<WilsonImplFH>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
| //#pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| //template<> void | ||||
| //WilsonKernels<ZWilsonImplFH>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| template<> void | ||||
| WilsonKernels<ZWilsonImplFH>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
|  | ||||
|  | ||||
| @@ -165,17 +165,17 @@ WilsonKernels<ZWilsonImplF>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldVi | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
| //#pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| //template<> void | ||||
| //WilsonKernels<WilsonImplFH>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| template<> void | ||||
| WilsonKernels<WilsonImplFH>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
| //#pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| //template<> void | ||||
| //WilsonKernels<ZWilsonImplFH>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| template<> void | ||||
| WilsonKernels<ZWilsonImplFH>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
|  | ||||
| #undef INTERIOR_AND_EXTERIOR | ||||
| @@ -194,17 +194,17 @@ WilsonKernels<ZWilsonImplF>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFiel | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
| //#pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| //template<> void | ||||
| //WilsonKernels<WilsonImplFH>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| template<> void | ||||
| WilsonKernels<WilsonImplFH>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
| //#pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| //template<> void | ||||
| //WilsonKernels<ZWilsonImplFH>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| template<> void | ||||
| WilsonKernels<ZWilsonImplFH>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
|  | ||||
| #undef INTERIOR_AND_EXTERIOR | ||||
| @@ -223,17 +223,17 @@ WilsonKernels<ZWilsonImplF>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFiel | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
| //#pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| //template<> void | ||||
| //WilsonKernels<WilsonImplFH>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| template<> void | ||||
| WilsonKernels<WilsonImplFH>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
| //#pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| //template<> void | ||||
| //WilsonKernels<ZWilsonImplFH>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| template<> void | ||||
| WilsonKernels<ZWilsonImplFH>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
|  | ||||
|  | ||||
| @@ -280,17 +280,17 @@ WilsonKernels<ZWilsonImplD>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
| // #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| // template<> void | ||||
| // WilsonKernels<WilsonImplDF>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| // 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| // #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| template<> void | ||||
| WilsonKernels<WilsonImplDF>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
| // #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| // template<> void | ||||
| // WilsonKernels<ZWilsonImplDF>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| // 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| // #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| template<> void | ||||
| WilsonKernels<ZWilsonImplDF>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
|  | ||||
| #undef INTERIOR_AND_EXTERIOR | ||||
| @@ -309,17 +309,17 @@ WilsonKernels<ZWilsonImplD>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldVi | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
| // #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| // template<> void | ||||
| // WilsonKernels<WilsonImplDF>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| // 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| // #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| template<> void | ||||
| WilsonKernels<WilsonImplDF>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
| // #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| // template<> void | ||||
| // WilsonKernels<ZWilsonImplDF>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| // 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| // #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| template<> void | ||||
| WilsonKernels<ZWilsonImplDF>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
|  | ||||
| #undef INTERIOR_AND_EXTERIOR | ||||
| @@ -338,17 +338,17 @@ WilsonKernels<ZWilsonImplD>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldVi | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
| // #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| // template<> void | ||||
| // WilsonKernels<WilsonImplDF>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| // 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| // #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| template<> void | ||||
| WilsonKernels<WilsonImplDF>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
| // #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| // template<> void | ||||
| // WilsonKernels<ZWilsonImplDF>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| // 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| // #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| template<> void | ||||
| WilsonKernels<ZWilsonImplDF>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
|  | ||||
| ///////////////////////////////////////////////////////////////// | ||||
| @@ -371,17 +371,17 @@ WilsonKernels<ZWilsonImplD>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldVi | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
| // #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| // template<> void | ||||
| // WilsonKernels<WilsonImplDF>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| // 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| // #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| template<> void | ||||
| WilsonKernels<WilsonImplDF>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
| // #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| // template<> void | ||||
| // WilsonKernels<ZWilsonImplDF>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| // 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| // #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| template<> void | ||||
| WilsonKernels<ZWilsonImplDF>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
|  | ||||
| #undef INTERIOR_AND_EXTERIOR | ||||
| @@ -400,17 +400,17 @@ WilsonKernels<ZWilsonImplD>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFiel | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
| // #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| // template<> void | ||||
| // WilsonKernels<WilsonImplDF>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| // 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| // #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| template<> void | ||||
| WilsonKernels<WilsonImplDF>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
| // #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| // template<> void | ||||
| // WilsonKernels<ZWilsonImplDF>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| // 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| // #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| template<> void | ||||
| WilsonKernels<ZWilsonImplDF>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
|  | ||||
| #undef INTERIOR_AND_EXTERIOR | ||||
| @@ -429,17 +429,17 @@ WilsonKernels<ZWilsonImplD>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFiel | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
| // #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| // template<> void | ||||
| // WilsonKernels<WilsonImplDF>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| // 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| // #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| template<> void | ||||
| WilsonKernels<WilsonImplDF>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
| // #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| // template<> void | ||||
| // WilsonKernels<ZWilsonImplDF>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| // 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| // #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | ||||
| template<> void | ||||
| WilsonKernels<ZWilsonImplDF>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||
|  | ||||
|  | ||||
|  | ||||
|   | ||||
| @@ -74,15 +74,15 @@ WilsonKernels<ZWilsonImplF>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| //template<> void | ||||
| //WilsonKernels<WilsonImplFH>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| // | ||||
| //template<> void | ||||
| //WilsonKernels<ZWilsonImplFH>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<WilsonImplFH>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| template<> void  | ||||
| WilsonKernels<ZWilsonImplFH>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| #undef INTERIOR_AND_EXTERIOR | ||||
| #define INTERIOR | ||||
| @@ -97,15 +97,15 @@ WilsonKernels<ZWilsonImplF>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldVi | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| //template<> void | ||||
| //WilsonKernels<WilsonImplFH>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| // | ||||
| //template<> void | ||||
| //WilsonKernels<ZWilsonImplFH>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<WilsonImplFH>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| template<> void  | ||||
| WilsonKernels<ZWilsonImplFH>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
|  | ||||
| #undef INTERIOR_AND_EXTERIOR | ||||
| @@ -121,15 +121,15 @@ WilsonKernels<ZWilsonImplF>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldVi | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| //template<> void | ||||
| //WilsonKernels<WilsonImplFH>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| // | ||||
| //template<> void | ||||
| //WilsonKernels<ZWilsonImplFH>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<WilsonImplFH>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| template<> void  | ||||
| WilsonKernels<ZWilsonImplFH>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|        | ||||
| ///////////////////////////////////////////////////////////////// | ||||
| // XYZT vectorised, dag Kernel, single | ||||
| @@ -148,15 +148,15 @@ WilsonKernels<ZWilsonImplF>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldVi | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| //template<> void | ||||
| //WilsonKernels<WilsonImplFH>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| // | ||||
| //template<> void | ||||
| //WilsonKernels<ZWilsonImplFH>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<WilsonImplFH>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| template<> void  | ||||
| WilsonKernels<ZWilsonImplFH>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| #undef INTERIOR_AND_EXTERIOR | ||||
| #define INTERIOR | ||||
| @@ -171,15 +171,15 @@ WilsonKernels<ZWilsonImplF>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFiel | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| //template<> void | ||||
| //WilsonKernels<WilsonImplFH>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| // | ||||
| //template<> void | ||||
| //WilsonKernels<ZWilsonImplFH>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<WilsonImplFH>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| template<> void  | ||||
| WilsonKernels<ZWilsonImplFH>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| #undef INTERIOR_AND_EXTERIOR | ||||
| #undef INTERIOR | ||||
| @@ -194,15 +194,15 @@ WilsonKernels<ZWilsonImplF>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFiel | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| 				     | ||||
| //template<> void | ||||
| //WilsonKernels<WilsonImplFH>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| // | ||||
| //template<> void | ||||
| //WilsonKernels<ZWilsonImplFH>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<WilsonImplFH>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| 				     | ||||
| template<> void  | ||||
| WilsonKernels<ZWilsonImplFH>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| 				     | ||||
| #undef MAYBEPERM | ||||
| #undef MULT_2SPIN | ||||
| @@ -228,14 +228,14 @@ WilsonKernels<ZDomainWallVec5dImplF>::AsmDhopSite(StencilView &st, DoubledGaugeF | ||||
| 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| //template<> void | ||||
| //WilsonKernels<DomainWallVec5dImplFH>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| //template<> void | ||||
| //WilsonKernels<ZDomainWallVec5dImplFH>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<DomainWallVec5dImplFH>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<ZDomainWallVec5dImplFH>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| #undef INTERIOR_AND_EXTERIOR | ||||
| #define INTERIOR | ||||
| @@ -249,14 +249,14 @@ WilsonKernels<ZDomainWallVec5dImplF>::AsmDhopSiteInt(StencilView &st, DoubledGau | ||||
| 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| //template<> void | ||||
| //WilsonKernels<DomainWallVec5dImplFH>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| //template<> void | ||||
| //WilsonKernels<ZDomainWallVec5dImplFH>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<DomainWallVec5dImplFH>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<ZDomainWallVec5dImplFH>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| #undef INTERIOR_AND_EXTERIOR | ||||
| #undef INTERIOR | ||||
| @@ -273,15 +273,15 @@ WilsonKernels<ZDomainWallVec5dImplF>::AsmDhopSiteExt(StencilView &st, DoubledGau | ||||
| 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| 				     | ||||
| //template<> void | ||||
| //WilsonKernels<DomainWallVec5dImplFH>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| // | ||||
| //template<> void | ||||
| //WilsonKernels<ZDomainWallVec5dImplFH>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<DomainWallVec5dImplFH>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| 				     | ||||
| template<> void  | ||||
| WilsonKernels<ZDomainWallVec5dImplFH>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| 				     | ||||
| ///////////////////////////////////////////////////////////////// | ||||
| // Ls vectorised, dag Kernel, single | ||||
| @@ -299,14 +299,14 @@ WilsonKernels<ZDomainWallVec5dImplF>::AsmDhopSiteDag(StencilView &st, DoubledGau | ||||
| 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| //template<> void | ||||
| //WilsonKernels<DomainWallVec5dImplFH>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||
| //							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| //template<> void | ||||
| //WilsonKernels<ZDomainWallVec5dImplFH>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||
| //							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<DomainWallVec5dImplFH>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||
| 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<ZDomainWallVec5dImplFH>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||
| 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| #undef INTERIOR_AND_EXTERIOR | ||||
| #define INTERIOR | ||||
| @@ -320,14 +320,14 @@ WilsonKernels<ZDomainWallVec5dImplF>::AsmDhopSiteDagInt(StencilView &st, Doubled | ||||
| 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| //template<> void | ||||
| //WilsonKernels<DomainWallVec5dImplFH>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||
| //							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| //template<> void | ||||
| //WilsonKernels<ZDomainWallVec5dImplFH>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||
| //							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<DomainWallVec5dImplFH>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||
| 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<ZDomainWallVec5dImplFH>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||
| 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| #undef INTERIOR_AND_EXTERIOR | ||||
| #undef INTERIOR | ||||
| @@ -341,14 +341,14 @@ WilsonKernels<ZDomainWallVec5dImplF>::AsmDhopSiteDagExt(StencilView &st, Doubled | ||||
| 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| //template<> void | ||||
| //WilsonKernels<DomainWallVec5dImplFH>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||
| //							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| //template<> void | ||||
| //WilsonKernels<ZDomainWallVec5dImplFH>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||
| //							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<DomainWallVec5dImplFH>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||
| 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<ZDomainWallVec5dImplFH>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||
| 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| #endif  // VEC 5D | ||||
|  | ||||
| @@ -392,14 +392,14 @@ WilsonKernels<ZWilsonImplD>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| //template<> void | ||||
| //WilsonKernels<WilsonImplDF>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| //template<> void | ||||
| //WilsonKernels<ZWilsonImplDF>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<WilsonImplDF>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<ZWilsonImplDF>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| #undef INTERIOR_AND_EXTERIOR | ||||
| #define INTERIOR | ||||
| @@ -413,14 +413,14 @@ WilsonKernels<ZWilsonImplD>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldVi | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| //template<> void | ||||
| //WilsonKernels<WilsonImplDF>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| //template<> void | ||||
| //WilsonKernels<ZWilsonImplDF>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<WilsonImplDF>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<ZWilsonImplDF>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| #undef INTERIOR_AND_EXTERIOR | ||||
| #undef INTERIOR | ||||
| @@ -434,14 +434,14 @@ WilsonKernels<ZWilsonImplD>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldVi | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|        | ||||
| //template<> void | ||||
| //WilsonKernels<WilsonImplDF>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| //template<> void | ||||
| //WilsonKernels<ZWilsonImplDF>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<WilsonImplDF>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<ZWilsonImplDF>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|        | ||||
| ///////////////////////////////////////////////////////////////// | ||||
| // XYZT vectorised, dag Kernel, single | ||||
| @@ -459,14 +459,14 @@ WilsonKernels<ZWilsonImplD>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldVi | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| //template<> void | ||||
| //WilsonKernels<WilsonImplDF>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| //template<> void | ||||
| //WilsonKernels<ZWilsonImplDF>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<WilsonImplDF>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<ZWilsonImplDF>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| #undef INTERIOR_AND_EXTERIOR | ||||
| #define INTERIOR | ||||
| @@ -480,14 +480,14 @@ WilsonKernels<ZWilsonImplD>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFiel | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| //template<> void | ||||
| //WilsonKernels<WilsonImplDF>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| //template<> void | ||||
| //WilsonKernels<ZWilsonImplDF>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<WilsonImplDF>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<ZWilsonImplDF>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| #undef INTERIOR_AND_EXTERIOR | ||||
| #undef INTERIOR | ||||
| @@ -501,14 +501,14 @@ WilsonKernels<ZWilsonImplD>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFiel | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| 				     | ||||
| //template<> void | ||||
| //WilsonKernels<WilsonImplDF>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| //template<> void | ||||
| //WilsonKernels<ZWilsonImplDF>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<WilsonImplDF>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<ZWilsonImplDF>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| 				     | ||||
| #undef MAYBEPERM | ||||
| #undef MULT_2SPIN | ||||
| @@ -533,14 +533,14 @@ WilsonKernels<ZDomainWallVec5dImplD>::AsmDhopSite(StencilView &st, DoubledGaugeF | ||||
| 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| //template<> void | ||||
| //WilsonKernels<DomainWallVec5dImplDF>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| //template<> void | ||||
| //WilsonKernels<ZDomainWallVec5dImplDF>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<DomainWallVec5dImplDF>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<ZDomainWallVec5dImplDF>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| #undef INTERIOR_AND_EXTERIOR | ||||
| #define INTERIOR | ||||
| @@ -554,14 +554,14 @@ WilsonKernels<ZDomainWallVec5dImplD>::AsmDhopSiteInt(StencilView &st, DoubledGau | ||||
| 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| //template<> void | ||||
| //WilsonKernels<DomainWallVec5dImplDF>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| //template<> void | ||||
| //WilsonKernels<ZDomainWallVec5dImplDF>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<DomainWallVec5dImplDF>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<ZDomainWallVec5dImplDF>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| #undef INTERIOR_AND_EXTERIOR | ||||
| #undef INTERIOR | ||||
| @@ -577,14 +577,14 @@ WilsonKernels<ZDomainWallVec5dImplD>::AsmDhopSiteExt(StencilView &st, DoubledGau | ||||
| 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| 				     | ||||
| //template<> void | ||||
| //WilsonKernels<DomainWallVec5dImplDF>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| //template<> void | ||||
| //WilsonKernels<ZDomainWallVec5dImplDF>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| //							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<DomainWallVec5dImplDF>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<ZDomainWallVec5dImplDF>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||
| 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| 				     | ||||
| ///////////////////////////////////////////////////////////////// | ||||
| // Ls vectorised, dag Kernel, single | ||||
| @@ -602,14 +602,14 @@ WilsonKernels<ZDomainWallVec5dImplD>::AsmDhopSiteDag(StencilView &st, DoubledGau | ||||
| 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| //template<> void | ||||
| //WilsonKernels<DomainWallVec5dImplDF>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||
| //							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| //template<> void | ||||
| //WilsonKernels<ZDomainWallVec5dImplDF>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||
| //							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<DomainWallVec5dImplDF>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||
| 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<ZDomainWallVec5dImplDF>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||
| 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| #undef INTERIOR_AND_EXTERIOR | ||||
| #define INTERIOR | ||||
| @@ -623,14 +623,14 @@ WilsonKernels<ZDomainWallVec5dImplD>::AsmDhopSiteDagInt(StencilView &st, Doubled | ||||
| 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| //template<> void | ||||
| //WilsonKernels<DomainWallVec5dImplDF>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||
| //							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| //template<> void | ||||
| //WilsonKernels<ZDomainWallVec5dImplDF>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||
| //							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<DomainWallVec5dImplDF>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||
| 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<ZDomainWallVec5dImplDF>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||
| 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| #undef INTERIOR_AND_EXTERIOR | ||||
| #undef INTERIOR | ||||
| @@ -645,14 +645,14 @@ WilsonKernels<ZDomainWallVec5dImplD>::AsmDhopSiteDagExt(StencilView &st, Doubled | ||||
| 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| //template<> void | ||||
| //WilsonKernels<DomainWallVec5dImplDF>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||
| //							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| //template<> void | ||||
| //WilsonKernels<ZDomainWallVec5dImplDF>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||
| //							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<DomainWallVec5dImplDF>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||
| 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
| template<> void  | ||||
| WilsonKernels<ZDomainWallVec5dImplDF>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||
| 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||
|  | ||||
| #endif  // VEC 5D | ||||
|  | ||||
|   | ||||
| @@ -77,23 +77,23 @@ Author: paboyle <paboyle@ph.ed.ac.uk> | ||||
| #define REGISTER | ||||
|  | ||||
| #ifdef GRID_SIMT | ||||
| #define LOAD_CHIMU(Ptype)		\ | ||||
| #define LOAD_CHIMU(ptype)		\ | ||||
|   {const SiteSpinor & ref (in[offset]);	\ | ||||
|     Chimu_00=coalescedReadPermute<Ptype>(ref()(0)(0),perm,lane);	\ | ||||
|     Chimu_01=coalescedReadPermute<Ptype>(ref()(0)(1),perm,lane);		\ | ||||
|     Chimu_02=coalescedReadPermute<Ptype>(ref()(0)(2),perm,lane);		\ | ||||
|     Chimu_10=coalescedReadPermute<Ptype>(ref()(1)(0),perm,lane);		\ | ||||
|     Chimu_11=coalescedReadPermute<Ptype>(ref()(1)(1),perm,lane);		\ | ||||
|     Chimu_12=coalescedReadPermute<Ptype>(ref()(1)(2),perm,lane);		\ | ||||
|     Chimu_20=coalescedReadPermute<Ptype>(ref()(2)(0),perm,lane);		\ | ||||
|     Chimu_21=coalescedReadPermute<Ptype>(ref()(2)(1),perm,lane);		\ | ||||
|     Chimu_22=coalescedReadPermute<Ptype>(ref()(2)(2),perm,lane);		\ | ||||
|     Chimu_30=coalescedReadPermute<Ptype>(ref()(3)(0),perm,lane);		\ | ||||
|     Chimu_31=coalescedReadPermute<Ptype>(ref()(3)(1),perm,lane);		\ | ||||
|     Chimu_32=coalescedReadPermute<Ptype>(ref()(3)(2),perm,lane);	} | ||||
|     Chimu_00=coalescedReadPermute<ptype>(ref()(0)(0),perm,lane);	\ | ||||
|     Chimu_01=coalescedReadPermute<ptype>(ref()(0)(1),perm,lane);		\ | ||||
|     Chimu_02=coalescedReadPermute<ptype>(ref()(0)(2),perm,lane);		\ | ||||
|     Chimu_10=coalescedReadPermute<ptype>(ref()(1)(0),perm,lane);		\ | ||||
|     Chimu_11=coalescedReadPermute<ptype>(ref()(1)(1),perm,lane);		\ | ||||
|     Chimu_12=coalescedReadPermute<ptype>(ref()(1)(2),perm,lane);		\ | ||||
|     Chimu_20=coalescedReadPermute<ptype>(ref()(2)(0),perm,lane);		\ | ||||
|     Chimu_21=coalescedReadPermute<ptype>(ref()(2)(1),perm,lane);		\ | ||||
|     Chimu_22=coalescedReadPermute<ptype>(ref()(2)(2),perm,lane);		\ | ||||
|     Chimu_30=coalescedReadPermute<ptype>(ref()(3)(0),perm,lane);		\ | ||||
|     Chimu_31=coalescedReadPermute<ptype>(ref()(3)(1),perm,lane);		\ | ||||
|     Chimu_32=coalescedReadPermute<ptype>(ref()(3)(2),perm,lane);	} | ||||
| #define PERMUTE_DIR(dir) ; | ||||
| #else | ||||
| #define LOAD_CHIMU(Ptype)		\ | ||||
| #define LOAD_CHIMU(ptype)		\ | ||||
|   {const SiteSpinor & ref (in[offset]);	\ | ||||
|     Chimu_00=ref()(0)(0);\ | ||||
|     Chimu_01=ref()(0)(1);\ | ||||
| @@ -109,12 +109,12 @@ Author: paboyle <paboyle@ph.ed.ac.uk> | ||||
|     Chimu_32=ref()(3)(2);} | ||||
|  | ||||
| #define PERMUTE_DIR(dir)			\ | ||||
|   permute##dir(Chi_00,Chi_00);			\ | ||||
|   permute##dir(Chi_01,Chi_01);			\ | ||||
|   permute##dir(Chi_02,Chi_02);			\ | ||||
|   permute##dir(Chi_10,Chi_10);			\ | ||||
|   permute##dir(Chi_11,Chi_11);			\ | ||||
|   permute##dir(Chi_12,Chi_12); | ||||
|   permute##dir(Chi_00,Chi_00);	\ | ||||
|       permute##dir(Chi_01,Chi_01);\ | ||||
|       permute##dir(Chi_02,Chi_02);\ | ||||
|       permute##dir(Chi_10,Chi_10);	\ | ||||
|       permute##dir(Chi_11,Chi_11);\ | ||||
|       permute##dir(Chi_12,Chi_12); | ||||
|  | ||||
| #endif | ||||
|  | ||||
| @@ -371,91 +371,88 @@ Author: paboyle <paboyle@ph.ed.ac.uk> | ||||
|   result_32-= UChi_12; | ||||
|  | ||||
| #define HAND_STENCIL_LEGB(PROJ,PERM,DIR,RECON)	\ | ||||
|   {int ptype;					\ | ||||
|    SE=st.GetEntry(ptype,DIR,ss);		\ | ||||
|    auto offset = SE->_offset;			\ | ||||
|    auto local  = SE->_is_local;			\ | ||||
|    auto perm   = SE->_permute;			\ | ||||
|    if ( local ) {				\ | ||||
|      LOAD_CHIMU(PERM);				\ | ||||
|      PROJ;					\ | ||||
|      if ( perm) {				\ | ||||
|        PERMUTE_DIR(PERM);			\ | ||||
|      }						\ | ||||
|    } else {					\ | ||||
|      LOAD_CHI;					\ | ||||
|    }						\ | ||||
|    acceleratorSynchronise();			\ | ||||
|    MULT_2SPIN(DIR);				\ | ||||
|    RECON;					} | ||||
|   SE=st.GetEntry(ptype,DIR,ss);			\ | ||||
|   offset = SE->_offset;				\ | ||||
|   local  = SE->_is_local;			\ | ||||
|   perm   = SE->_permute;			\ | ||||
|   if ( local ) {				\ | ||||
|     LOAD_CHIMU(PERM);				\ | ||||
|     PROJ;					\ | ||||
|     if ( perm) {				\ | ||||
|       PERMUTE_DIR(PERM);			\ | ||||
|     }						\ | ||||
|   } else {					\ | ||||
|     LOAD_CHI;					\ | ||||
|   }						\ | ||||
|   acceleratorSynchronise();			\ | ||||
|   MULT_2SPIN(DIR);				\ | ||||
|   RECON;					 | ||||
|  | ||||
| #define HAND_STENCIL_LEG(PROJ,PERM,DIR,RECON)		\ | ||||
|   { SE=&st_p[DIR+8*ss];						\ | ||||
|   auto ptype=st_perm[DIR];					\ | ||||
|   auto offset = SE->_offset;					\ | ||||
|   auto local  = SE->_is_local;					\ | ||||
|   auto perm   = SE->_permute;					\ | ||||
|   if ( local ) {						\ | ||||
|     LOAD_CHIMU(PERM);						\ | ||||
|     PROJ;							\ | ||||
|     if ( perm) {						\ | ||||
|       PERMUTE_DIR(PERM);					\ | ||||
|     }								\ | ||||
|   } else {							\ | ||||
|     LOAD_CHI;							\ | ||||
|   }								\ | ||||
|   acceleratorSynchronise();					\ | ||||
|   MULT_2SPIN(DIR);						\ | ||||
|   RECON;					} | ||||
| #define HAND_STENCIL_LEG(PROJ,PERM,DIR,RECON)	\ | ||||
|   SE=&st_p[DIR+8*ss];				\ | ||||
|   ptype=st_perm[DIR];				\ | ||||
|   offset = SE->_offset;				\ | ||||
|   local  = SE->_is_local;			\ | ||||
|   perm   = SE->_permute;			\ | ||||
|   if ( local ) {				\ | ||||
|     LOAD_CHIMU(PERM);				\ | ||||
|     PROJ;					\ | ||||
|     if ( perm) {				\ | ||||
|       PERMUTE_DIR(PERM);			\ | ||||
|     }						\ | ||||
|   } else {					\ | ||||
|     LOAD_CHI;					\ | ||||
|   }						\ | ||||
|   acceleratorSynchronise();			\ | ||||
|   MULT_2SPIN(DIR);				\ | ||||
|   RECON;					 | ||||
|  | ||||
| #define HAND_STENCIL_LEGA(PROJ,PERM,DIR,RECON)				\ | ||||
|   { SE=&st_p[DIR+8*ss];							\ | ||||
|     auto ptype=st_perm[DIR];						\ | ||||
|     /*SE=st.GetEntry(ptype,DIR,ss);*/					\ | ||||
|     auto offset = SE->_offset;						\ | ||||
|     auto perm   = SE->_permute;						\ | ||||
|     LOAD_CHIMU(PERM);							\ | ||||
|     PROJ;								\ | ||||
|     MULT_2SPIN(DIR);							\ | ||||
|     RECON;					} | ||||
|   SE=&st_p[DIR+8*ss];							\ | ||||
|   ptype=st_perm[DIR];							\ | ||||
|  /*SE=st.GetEntry(ptype,DIR,ss);*/					\ | ||||
|   offset = SE->_offset;				\ | ||||
|   perm   = SE->_permute;			\ | ||||
|   LOAD_CHIMU(PERM);				\ | ||||
|   PROJ;						\ | ||||
|   MULT_2SPIN(DIR);				\ | ||||
|   RECON;					 | ||||
|  | ||||
| #define HAND_STENCIL_LEG_INT(PROJ,PERM,DIR,RECON)	\ | ||||
|   { int ptype;						\ | ||||
|   SE=st.GetEntry(ptype,DIR,ss);				\ | ||||
|   auto offset = SE->_offset;					\ | ||||
|   auto local  = SE->_is_local;					\ | ||||
|   auto perm   = SE->_permute;					\ | ||||
|   if ( local ) {						\ | ||||
|     LOAD_CHIMU(PERM);						\ | ||||
|     PROJ;							\ | ||||
|     if ( perm) {						\ | ||||
|       PERMUTE_DIR(PERM);					\ | ||||
|     }								\ | ||||
|   } else if ( st.same_node[DIR] ) {				\ | ||||
|     LOAD_CHI;							\ | ||||
|   }								\ | ||||
|   acceleratorSynchronise();					\ | ||||
|   if (local || st.same_node[DIR] ) {				\ | ||||
|     MULT_2SPIN(DIR);						\ | ||||
|     RECON;							\ | ||||
|   }								\ | ||||
|   acceleratorSynchronise();			} | ||||
|   SE=st.GetEntry(ptype,DIR,ss);			\ | ||||
|   offset = SE->_offset;				\ | ||||
|   local  = SE->_is_local;			\ | ||||
|   perm   = SE->_permute;			\ | ||||
|   if ( local ) {				\ | ||||
|     LOAD_CHIMU(PERM);				\ | ||||
|     PROJ;					\ | ||||
|     if ( perm) {				\ | ||||
|       PERMUTE_DIR(PERM);			\ | ||||
|     }						\ | ||||
|   } else if ( st.same_node[DIR] ) {		\ | ||||
|     LOAD_CHI;					\ | ||||
|   }						\ | ||||
|   acceleratorSynchronise();			\ | ||||
|   if (local || st.same_node[DIR] ) {		\ | ||||
|     MULT_2SPIN(DIR);				\ | ||||
|     RECON;					\ | ||||
|   }						\ | ||||
|   acceleratorSynchronise();			 | ||||
|  | ||||
| #define HAND_STENCIL_LEG_EXT(PROJ,PERM,DIR,RECON)	\ | ||||
|   { int ptype;						\ | ||||
|   SE=st.GetEntry(ptype,DIR,ss);				\ | ||||
|   auto offset = SE->_offset;				\ | ||||
|   if((!SE->_is_local)&&(!st.same_node[DIR]) ) {		\ | ||||
|     LOAD_CHI;						\ | ||||
|     MULT_2SPIN(DIR);					\ | ||||
|     RECON;						\ | ||||
|     nmu++;						\ | ||||
|   }							\ | ||||
|   acceleratorSynchronise();			} | ||||
|   SE=st.GetEntry(ptype,DIR,ss);			\ | ||||
|   offset = SE->_offset;				\ | ||||
|   if((!SE->_is_local)&&(!st.same_node[DIR]) ) {	\ | ||||
|     LOAD_CHI;					\ | ||||
|     MULT_2SPIN(DIR);				\ | ||||
|     RECON;					\ | ||||
|     nmu++;					\ | ||||
|   }						\ | ||||
|   acceleratorSynchronise();			 | ||||
|  | ||||
| #define HAND_RESULT(ss)					\ | ||||
|   {							\ | ||||
|     SiteSpinor & ref (out[ss]);				\ | ||||
| #define HAND_RESULT(ss)				\ | ||||
|   {						\ | ||||
|     SiteSpinor & ref (out[ss]);			\ | ||||
|     coalescedWrite(ref()(0)(0),result_00,lane);		\ | ||||
|     coalescedWrite(ref()(0)(1),result_01,lane);		\ | ||||
|     coalescedWrite(ref()(0)(2),result_02,lane);		\ | ||||
| @@ -566,6 +563,7 @@ WilsonKernels<Impl>::HandDhopSiteSycl(StencilVector st_perm,StencilEntry *st_p, | ||||
|  | ||||
|   HAND_DECLARATIONS(Simt); | ||||
|  | ||||
|   int offset,local,perm, ptype; | ||||
|   StencilEntry *SE; | ||||
|   HAND_STENCIL_LEG(XM_PROJ,3,Xp,XM_RECON); | ||||
|   HAND_STENCIL_LEG(YM_PROJ,2,Yp,YM_RECON_ACCUM); | ||||
| @@ -595,7 +593,9 @@ WilsonKernels<Impl>::HandDhopSite(StencilView &st, DoubledGaugeFieldView &U,Site | ||||
|  | ||||
|   HAND_DECLARATIONS(Simt); | ||||
|  | ||||
|   int offset,local,perm, ptype; | ||||
|   StencilEntry *SE; | ||||
|  | ||||
|   HAND_STENCIL_LEG(XM_PROJ,3,Xp,XM_RECON); | ||||
|   HAND_STENCIL_LEG(YM_PROJ,2,Yp,YM_RECON_ACCUM); | ||||
|   HAND_STENCIL_LEG(ZM_PROJ,1,Zp,ZM_RECON_ACCUM); | ||||
| @@ -623,6 +623,8 @@ void WilsonKernels<Impl>::HandDhopSiteDag(StencilView &st,DoubledGaugeFieldView | ||||
|   HAND_DECLARATIONS(Simt); | ||||
|  | ||||
|   StencilEntry *SE; | ||||
|   int offset,local,perm, ptype; | ||||
|    | ||||
|   HAND_STENCIL_LEG(XP_PROJ,3,Xp,XP_RECON); | ||||
|   HAND_STENCIL_LEG(YP_PROJ,2,Yp,YP_RECON_ACCUM); | ||||
|   HAND_STENCIL_LEG(ZP_PROJ,1,Zp,ZP_RECON_ACCUM); | ||||
| @@ -638,8 +640,8 @@ template<class Impl>  accelerator_inline void | ||||
| WilsonKernels<Impl>::HandDhopSiteInt(StencilView &st,DoubledGaugeFieldView &U,SiteHalfSpinor  *buf, | ||||
| 					  int ss,int sU,const FermionFieldView &in, FermionFieldView &out) | ||||
| { | ||||
|   //  auto st_p = st._entries_p;						 | ||||
|   //  auto st_perm = st._permute_type;					 | ||||
|   auto st_p = st._entries_p;						 | ||||
|   auto st_perm = st._permute_type;					 | ||||
| // T==0, Z==1, Y==2, Z==3 expect 1,2,2,2 simd layout etc... | ||||
|   typedef typename Simd::scalar_type S; | ||||
|   typedef typename Simd::vector_type V; | ||||
| @@ -650,6 +652,7 @@ WilsonKernels<Impl>::HandDhopSiteInt(StencilView &st,DoubledGaugeFieldView &U,Si | ||||
|  | ||||
|   HAND_DECLARATIONS(Simt); | ||||
|  | ||||
|   int offset,local,perm, ptype; | ||||
|   StencilEntry *SE; | ||||
|   ZERO_RESULT; | ||||
|   HAND_STENCIL_LEG_INT(XM_PROJ,3,Xp,XM_RECON_ACCUM); | ||||
| @@ -667,8 +670,8 @@ template<class Impl> accelerator_inline | ||||
| void WilsonKernels<Impl>::HandDhopSiteDagInt(StencilView &st,DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||
| 						  int ss,int sU,const FermionFieldView &in, FermionFieldView &out) | ||||
| { | ||||
|   //  auto st_p = st._entries_p;						 | ||||
|   //  auto st_perm = st._permute_type;					 | ||||
|   auto st_p = st._entries_p;						 | ||||
|   auto st_perm = st._permute_type;					 | ||||
|   typedef typename Simd::scalar_type S; | ||||
|   typedef typename Simd::vector_type V; | ||||
|   typedef decltype( coalescedRead( in[0]()(0)(0) )) Simt; | ||||
| @@ -679,6 +682,7 @@ void WilsonKernels<Impl>::HandDhopSiteDagInt(StencilView &st,DoubledGaugeFieldVi | ||||
|   HAND_DECLARATIONS(Simt); | ||||
|  | ||||
|   StencilEntry *SE; | ||||
|   int offset,local,perm, ptype; | ||||
|   ZERO_RESULT; | ||||
|   HAND_STENCIL_LEG_INT(XP_PROJ,3,Xp,XP_RECON_ACCUM); | ||||
|   HAND_STENCIL_LEG_INT(YP_PROJ,2,Yp,YP_RECON_ACCUM); | ||||
| @@ -695,8 +699,8 @@ template<class Impl>  accelerator_inline void | ||||
| WilsonKernels<Impl>::HandDhopSiteExt(StencilView &st,DoubledGaugeFieldView &U,SiteHalfSpinor  *buf, | ||||
| 					  int ss,int sU,const FermionFieldView &in, FermionFieldView &out) | ||||
| { | ||||
|   //  auto st_p = st._entries_p;						 | ||||
|   //  auto st_perm = st._permute_type;					 | ||||
|   auto st_p = st._entries_p;						 | ||||
|   auto st_perm = st._permute_type;					 | ||||
| // T==0, Z==1, Y==2, Z==3 expect 1,2,2,2 simd layout etc... | ||||
|   typedef typename Simd::scalar_type S; | ||||
|   typedef typename Simd::vector_type V; | ||||
| @@ -707,7 +711,7 @@ WilsonKernels<Impl>::HandDhopSiteExt(StencilView &st,DoubledGaugeFieldView &U,Si | ||||
|  | ||||
|   HAND_DECLARATIONS(Simt); | ||||
|  | ||||
|   //  int offset, ptype; | ||||
|   int offset, ptype; | ||||
|   StencilEntry *SE; | ||||
|   int nmu=0; | ||||
|   ZERO_RESULT; | ||||
| @@ -726,8 +730,8 @@ template<class Impl>  accelerator_inline | ||||
| void WilsonKernels<Impl>::HandDhopSiteDagExt(StencilView &st,DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||
| 						  int ss,int sU,const FermionFieldView &in, FermionFieldView &out) | ||||
| { | ||||
|   //  auto st_p = st._entries_p;						 | ||||
|   //  auto st_perm = st._permute_type;					 | ||||
|   auto st_p = st._entries_p;						 | ||||
|   auto st_perm = st._permute_type;					 | ||||
|   typedef typename Simd::scalar_type S; | ||||
|   typedef typename Simd::vector_type V; | ||||
|   typedef decltype( coalescedRead( in[0]()(0)(0) )) Simt; | ||||
| @@ -738,7 +742,7 @@ void WilsonKernels<Impl>::HandDhopSiteDagExt(StencilView &st,DoubledGaugeFieldVi | ||||
|   HAND_DECLARATIONS(Simt); | ||||
|  | ||||
|   StencilEntry *SE; | ||||
|   //  int offset, ptype; | ||||
|   int offset, ptype; | ||||
|   int nmu=0; | ||||
|   ZERO_RESULT; | ||||
|   HAND_STENCIL_LEG_EXT(XP_PROJ,3,Xp,XP_RECON_ACCUM); | ||||
|   | ||||
| @@ -1,41 +0,0 @@ | ||||
| /************************************************************************************* | ||||
|  | ||||
|     Grid physics library, www.github.com/paboyle/Grid | ||||
|  | ||||
|     Source file: ./lib/ qcd/action/fermion/instantiation/CompactWilsonCloverFermionInstantiation.cc.master | ||||
|  | ||||
|     Copyright (C) 2017 - 2022 | ||||
|  | ||||
|     Author: paboyle <paboyle@ph.ed.ac.uk> | ||||
|     Author: Guido Cossu <guido.cossu@ed.ac.uk> | ||||
|     Author: Daniel Richtmann <daniel.richtmann@gmail.com> | ||||
|  | ||||
|     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/spin/Dirac.h> | ||||
| #include <Grid/qcd/action/fermion/CompactWilsonCloverFermion.h> | ||||
| #include <Grid/qcd/action/fermion/implementation/CompactWilsonCloverFermionImplementation.h> | ||||
|  | ||||
| NAMESPACE_BEGIN(Grid); | ||||
|  | ||||
| #include "impl.h" | ||||
| template class CompactWilsonCloverFermion<IMPLEMENTATION>;  | ||||
|  | ||||
| NAMESPACE_END(Grid); | ||||
| @@ -0,0 +1 @@ | ||||
| ../CayleyFermion5DInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../ContinuedFractionFermion5DInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../DomainWallEOFAFermionInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../MobiusEOFAFermionInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../PartialFractionFermion5DInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../WilsonCloverFermionInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../WilsonFermion5DInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../WilsonFermionInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../WilsonKernelsInstantiationGparity.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../WilsonTMFermionInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| #define IMPLEMENTATION GparityWilsonImplDF | ||||
| @@ -0,0 +1 @@ | ||||
| ../CayleyFermion5DInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../ContinuedFractionFermion5DInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../DomainWallEOFAFermionInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../MobiusEOFAFermionInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../PartialFractionFermion5DInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../WilsonCloverFermionInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../WilsonFermion5DInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../WilsonFermionInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../WilsonKernelsInstantiationGparity.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../WilsonTMFermionInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| #define IMPLEMENTATION GparityWilsonImplFH | ||||
| @@ -1 +0,0 @@ | ||||
| ../CompactWilsonCloverFermionInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../CayleyFermion5DInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../ContinuedFractionFermion5DInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../DomainWallEOFAFermionInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../MobiusEOFAFermionInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../PartialFractionFermion5DInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../WilsonCloverFermionInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../WilsonFermion5DInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../WilsonFermionInstantiation.cc.master | ||||
| @@ -2,11 +2,14 @@ | ||||
| 
 | ||||
| Grid physics library, www.github.com/paboyle/Grid | ||||
| 
 | ||||
| Source file: ./lib/serialisation/BaseIO.h | ||||
| Source file: ./lib/qcd/action/fermion/WilsonKernels.cc | ||||
| 
 | ||||
| Copyright (C) 2015 | ||||
| Copyright (C) 2015, 2020 | ||||
| 
 | ||||
| Author: Michael Marshall <michael.marshall@ed.ac.uk> | ||||
| Author: Peter Boyle <paboyle@ph.ed.ac.uk> | ||||
| Author: Peter Boyle <peterboyle@Peters-MacBook-Pro-2.local> | ||||
| Author: paboyle <paboyle@ph.ed.ac.uk> | ||||
| Author: Nils Meyer <nils.meyer@ur.de> Regensburg University | ||||
| 
 | ||||
| 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 | ||||
| @@ -22,14 +25,27 @@ 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 | ||||
| See the full license in the file "LICENSE" in the top level distribution | ||||
| directory | ||||
| *************************************************************************************/ | ||||
| /*  END LEGAL */ | ||||
| #include <Grid/qcd/action/fermion/FermionCore.h> | ||||
| #include <Grid/qcd/action/fermion/implementation/WilsonKernelsImplementation.h> | ||||
| #include <Grid/qcd/action/fermion/implementation/WilsonKernelsHandImplementation.h> | ||||
| 
 | ||||
| #include <Grid/GridCore.h> | ||||
| #ifndef AVX512 | ||||
| #ifndef QPX | ||||
| #ifndef A64FX | ||||
| #ifndef A64FXFIXEDSIZE | ||||
| #include <Grid/qcd/action/fermion/implementation/WilsonKernelsAsmImplementation.h> | ||||
| #endif | ||||
| #endif | ||||
| #endif | ||||
| #endif | ||||
| 
 | ||||
| NAMESPACE_BEGIN(Grid) | ||||
| NAMESPACE_BEGIN(Grid); | ||||
| 
 | ||||
| std::uint64_t EigenIO::EigenResizeCounter(0); | ||||
| #include "impl.h" | ||||
| template class WilsonKernels<IMPLEMENTATION>; | ||||
| 
 | ||||
| NAMESPACE_END(Grid) | ||||
| NAMESPACE_END(Grid); | ||||
| @@ -0,0 +1 @@ | ||||
| ../WilsonTMFermionInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| #define IMPLEMENTATION WilsonImplDF | ||||
| @@ -1 +0,0 @@ | ||||
| ../CompactWilsonCloverFermionInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../CayleyFermion5DInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../ContinuedFractionFermion5DInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../DomainWallEOFAFermionInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../MobiusEOFAFermionInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../PartialFractionFermion5DInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../WilsonCloverFermionInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../WilsonFermion5DInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../WilsonFermionInstantiation.cc.master | ||||
| @@ -0,0 +1,51 @@ | ||||
| /************************************************************************************* | ||||
|  | ||||
| Grid physics library, www.github.com/paboyle/Grid | ||||
|  | ||||
| Source file: ./lib/qcd/action/fermion/WilsonKernels.cc | ||||
|  | ||||
| Copyright (C) 2015, 2020 | ||||
|  | ||||
| Author: Peter Boyle <paboyle@ph.ed.ac.uk> | ||||
| Author: Peter Boyle <peterboyle@Peters-MacBook-Pro-2.local> | ||||
| Author: paboyle <paboyle@ph.ed.ac.uk> | ||||
| Author: Nils Meyer <nils.meyer@ur.de> Regensburg University | ||||
|  | ||||
| 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/qcd/action/fermion/FermionCore.h> | ||||
| #include <Grid/qcd/action/fermion/implementation/WilsonKernelsImplementation.h> | ||||
| #include <Grid/qcd/action/fermion/implementation/WilsonKernelsHandImplementation.h> | ||||
|  | ||||
| #ifndef AVX512 | ||||
| #ifndef QPX | ||||
| #ifndef A64FX | ||||
| #ifndef A64FXFIXEDSIZE | ||||
| #include <Grid/qcd/action/fermion/implementation/WilsonKernelsAsmImplementation.h> | ||||
| #endif | ||||
| #endif | ||||
| #endif | ||||
| #endif | ||||
|  | ||||
| NAMESPACE_BEGIN(Grid); | ||||
|  | ||||
| #include "impl.h" | ||||
| template class WilsonKernels<IMPLEMENTATION>; | ||||
|  | ||||
| NAMESPACE_END(Grid); | ||||
| @@ -0,0 +1 @@ | ||||
| ../WilsonTMFermionInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| #define IMPLEMENTATION WilsonImplFH | ||||
| @@ -0,0 +1 @@ | ||||
| ../CayleyFermion5DInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../ContinuedFractionFermion5DInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../DomainWallEOFAFermionInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../MobiusEOFAFermionInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../PartialFractionFermion5DInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../WilsonFermion5DInstantiation.cc.master | ||||
| @@ -0,0 +1,51 @@ | ||||
| /************************************************************************************* | ||||
|  | ||||
| Grid physics library, www.github.com/paboyle/Grid | ||||
|  | ||||
| Source file: ./lib/qcd/action/fermion/WilsonKernels.cc | ||||
|  | ||||
| Copyright (C) 2015, 2020 | ||||
|  | ||||
| Author: Peter Boyle <paboyle@ph.ed.ac.uk> | ||||
| Author: Peter Boyle <peterboyle@Peters-MacBook-Pro-2.local> | ||||
| Author: paboyle <paboyle@ph.ed.ac.uk> | ||||
| Author: Nils Meyer <nils.meyer@ur.de> Regensburg University | ||||
|  | ||||
| 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/qcd/action/fermion/FermionCore.h> | ||||
| #include <Grid/qcd/action/fermion/implementation/WilsonKernelsImplementation.h> | ||||
| #include <Grid/qcd/action/fermion/implementation/WilsonKernelsHandImplementation.h> | ||||
|  | ||||
| #ifndef AVX512 | ||||
| #ifndef QPX | ||||
| #ifndef A64FX | ||||
| #ifndef A64FXFIXEDSIZE | ||||
| #include <Grid/qcd/action/fermion/implementation/WilsonKernelsAsmImplementation.h> | ||||
| #endif | ||||
| #endif | ||||
| #endif | ||||
| #endif | ||||
|  | ||||
| NAMESPACE_BEGIN(Grid); | ||||
|  | ||||
| #include "impl.h" | ||||
| template class WilsonKernels<IMPLEMENTATION>; | ||||
|  | ||||
| NAMESPACE_END(Grid); | ||||
| @@ -0,0 +1 @@ | ||||
| #define IMPLEMENTATION ZWilsonImplDF | ||||
| @@ -0,0 +1 @@ | ||||
| ../CayleyFermion5DInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../ContinuedFractionFermion5DInstantiation.cc.master | ||||
| @@ -0,0 +1 @@ | ||||
| ../DomainWallEOFAFermionInstantiation.cc.master | ||||
Some files were not shown because too many files have changed in this diff Show More
		Reference in New Issue
	
	Block a user