mirror of
				https://github.com/paboyle/Grid.git
				synced 2025-10-31 12:04:33 +00:00 
			
		
		
		
	Compare commits
	
		
			5 Commits
		
	
	
		
			gauge-grou
			...
			sycl-linki
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
|  | 6d25badce5 | ||
|  | adcc97cbbb | ||
|  | bc8c5fb16c | ||
|  | 845d757bb0 | ||
|  | 0d5470c363 | 
							
								
								
									
										1
									
								
								.gitignore
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										1
									
								
								.gitignore
									
									
									
									
										vendored
									
									
								
							| @@ -88,7 +88,6 @@ Thumbs.db | |||||||
| # build directory # | # build directory # | ||||||
| ################### | ################### | ||||||
| build*/* | build*/* | ||||||
| Documentation/_build |  | ||||||
|  |  | ||||||
| # IDE related files # | # IDE related files # | ||||||
| ##################### | ##################### | ||||||
|   | |||||||
							
								
								
									
										56
									
								
								.travis.yml
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										56
									
								
								.travis.yml
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,56 @@ | |||||||
|  | language: cpp | ||||||
|  |  | ||||||
|  | cache: | ||||||
|  |   directories: | ||||||
|  |     - clang | ||||||
|  |  | ||||||
|  | matrix: | ||||||
|  |   include: | ||||||
|  |     - os:        osx | ||||||
|  |       osx_image: xcode8.3 | ||||||
|  |       compiler: clang | ||||||
|  |        | ||||||
|  | before_install: | ||||||
|  |     - export GRIDDIR=`pwd` | ||||||
|  |     - if [[ "$TRAVIS_OS_NAME" == "linux" ]] && [[ "$CC" == "clang" ]] && [ ! -e clang/bin ]; then wget $CLANG_LINK; tar -xf `basename $CLANG_LINK`; mkdir clang; mv clang+*/* clang/; fi | ||||||
|  |     - if [[ "$TRAVIS_OS_NAME" == "linux" ]] && [[ "$CC" == "clang" ]]; then export PATH="${GRIDDIR}/clang/bin:${PATH}"; fi | ||||||
|  |     - if [[ "$TRAVIS_OS_NAME" == "linux" ]] && [[ "$CC" == "clang" ]]; then export LD_LIBRARY_PATH="${GRIDDIR}/clang/lib:${LD_LIBRARY_PATH}"; fi | ||||||
|  |     - if [[ "$TRAVIS_OS_NAME" == "osx" ]]; then brew update; fi | ||||||
|  |     - if [[ "$TRAVIS_OS_NAME" == "osx" ]]; then brew install libmpc openssl; fi | ||||||
|  |      | ||||||
|  | install: | ||||||
|  |     - export CWD=`pwd` | ||||||
|  |     - echo $CWD | ||||||
|  |     - export CC=$CC$VERSION | ||||||
|  |     - export CXX=$CXX$VERSION | ||||||
|  |     - echo $PATH | ||||||
|  |     - which autoconf | ||||||
|  |     - autoconf  --version | ||||||
|  |     - which automake | ||||||
|  |     - automake  --version | ||||||
|  |     - which $CC | ||||||
|  |     - $CC  --version | ||||||
|  |     - which $CXX | ||||||
|  |     - $CXX --version | ||||||
|  |     - if [[ "$TRAVIS_OS_NAME" == "osx" ]]; then export LDFLAGS='-L/usr/local/lib'; fi | ||||||
|  |     - if [[ "$TRAVIS_OS_NAME" == "osx" ]]; then export EXTRACONF='--with-openssl=/usr/local/opt/openssl'; fi | ||||||
|  |      | ||||||
|  | script: | ||||||
|  |     - ./bootstrap.sh | ||||||
|  |     - mkdir build | ||||||
|  |     - cd build | ||||||
|  |     - mkdir lime | ||||||
|  |     - cd lime | ||||||
|  |     - mkdir build | ||||||
|  |     - cd build | ||||||
|  |     - wget http://usqcd-software.github.io/downloads/c-lime/lime-1.3.2.tar.gz | ||||||
|  |     - tar xf lime-1.3.2.tar.gz | ||||||
|  |     - cd lime-1.3.2 | ||||||
|  |     - ./configure --prefix=$CWD/build/lime/install | ||||||
|  |     - make -j4 | ||||||
|  |     - make install | ||||||
|  |     - cd $CWD/build | ||||||
|  |     - ../configure --enable-simd=SSE4 --enable-comms=none --with-lime=$CWD/build/lime/install ${EXTRACONF} | ||||||
|  |     - make -j4  | ||||||
|  |     - ./benchmarks/Benchmark_dwf --threads 1 --debug-signals | ||||||
|  |     - make check | ||||||
| @@ -37,9 +37,7 @@ directory | |||||||
| #endif | #endif | ||||||
|  |  | ||||||
|  //disables and intel compiler specific warning (in json.hpp) |  //disables and intel compiler specific warning (in json.hpp) | ||||||
| #ifdef __ICC |  | ||||||
| #pragma warning disable 488   | #pragma warning disable 488   | ||||||
| #endif |  | ||||||
|  |  | ||||||
| #ifdef __NVCC__ | #ifdef __NVCC__ | ||||||
|  //disables nvcc specific warning in json.hpp |  //disables nvcc specific warning in json.hpp | ||||||
|   | |||||||
| @@ -54,11 +54,9 @@ Version.h: version-cache | |||||||
| include Make.inc | include Make.inc | ||||||
| include Eigen.inc | include Eigen.inc | ||||||
|  |  | ||||||
|  | extra_sources+=$(ZWILS_FERMION_FILES) | ||||||
| extra_sources+=$(WILS_FERMION_FILES) | extra_sources+=$(WILS_FERMION_FILES) | ||||||
| extra_sources+=$(STAG_FERMION_FILES) | extra_sources+=$(STAG_FERMION_FILES) | ||||||
| if BUILD_ZMOBIUS |  | ||||||
|   extra_sources+=$(ZWILS_FERMION_FILES) |  | ||||||
| endif |  | ||||||
| if BUILD_GPARITY | if BUILD_GPARITY | ||||||
|   extra_sources+=$(GP_FERMION_FILES) |   extra_sources+=$(GP_FERMION_FILES) | ||||||
| endif | endif | ||||||
|   | |||||||
| @@ -31,7 +31,6 @@ Author: paboyle <paboyle@ph.ed.ac.uk> | |||||||
| #ifndef  GRID_ALGORITHM_COARSENED_MATRIX_H | #ifndef  GRID_ALGORITHM_COARSENED_MATRIX_H | ||||||
| #define  GRID_ALGORITHM_COARSENED_MATRIX_H | #define  GRID_ALGORITHM_COARSENED_MATRIX_H | ||||||
|  |  | ||||||
| #include <Grid/qcd/QCD.h> // needed for Dagger(Yes|No), Inverse(Yes|No) |  | ||||||
|  |  | ||||||
| NAMESPACE_BEGIN(Grid); | NAMESPACE_BEGIN(Grid); | ||||||
|  |  | ||||||
| @@ -60,14 +59,12 @@ inline void blockMaskedInnerProduct(Lattice<CComplex> &CoarseInner, | |||||||
| class Geometry { | class Geometry { | ||||||
| public: | public: | ||||||
|   int npoint; |   int npoint; | ||||||
|   int base; |  | ||||||
|   std::vector<int> directions   ; |   std::vector<int> directions   ; | ||||||
|   std::vector<int> displacements; |   std::vector<int> displacements; | ||||||
|   std::vector<int> points_dagger; |  | ||||||
|  |  | ||||||
|   Geometry(int _d)  { |   Geometry(int _d)  { | ||||||
|      |      | ||||||
|     base = (_d==5) ? 1:0; |     int base = (_d==5) ? 1:0; | ||||||
|  |  | ||||||
|     // make coarse grid stencil for 4d , not 5d |     // make coarse grid stencil for 4d , not 5d | ||||||
|     if ( _d==5 ) _d=4; |     if ( _d==5 ) _d=4; | ||||||
| @@ -75,51 +72,16 @@ public: | |||||||
|     npoint = 2*_d+1; |     npoint = 2*_d+1; | ||||||
|     directions.resize(npoint); |     directions.resize(npoint); | ||||||
|     displacements.resize(npoint); |     displacements.resize(npoint); | ||||||
|     points_dagger.resize(npoint); |  | ||||||
|     for(int d=0;d<_d;d++){ |     for(int d=0;d<_d;d++){ | ||||||
|       directions[d   ] = d+base; |       directions[d   ] = d+base; | ||||||
|       directions[d+_d] = d+base; |       directions[d+_d] = d+base; | ||||||
|       displacements[d  ] = +1; |       displacements[d  ] = +1; | ||||||
|       displacements[d+_d]= -1; |       displacements[d+_d]= -1; | ||||||
|       points_dagger[d   ] = d+_d; |  | ||||||
|       points_dagger[d+_d] = d; |  | ||||||
|     } |     } | ||||||
|     directions   [2*_d]=0; |     directions   [2*_d]=0; | ||||||
|     displacements[2*_d]=0; |     displacements[2*_d]=0; | ||||||
|     points_dagger[2*_d]=2*_d; |  | ||||||
|   } |   } | ||||||
|  |  | ||||||
|   int point(int dir, int disp) { |  | ||||||
|     assert(disp == -1 || disp == 0 || disp == 1); |  | ||||||
|     assert(base+0 <= dir && dir < base+4); |  | ||||||
|  |  | ||||||
|     // directions faster index = new indexing |  | ||||||
|     // 4d (base = 0): |  | ||||||
|     // point 0  1  2  3  4  5  6  7  8 |  | ||||||
|     // dir   0  1  2  3  0  1  2  3  0 |  | ||||||
|     // disp +1 +1 +1 +1 -1 -1 -1 -1  0 |  | ||||||
|     // 5d (base = 1): |  | ||||||
|     // point 0  1  2  3  4  5  6  7  8 |  | ||||||
|     // dir   1  2  3  4  1  2  3  4  0 |  | ||||||
|     // disp +1 +1 +1 +1 -1 -1 -1 -1  0 |  | ||||||
|  |  | ||||||
|     // displacements faster index = old indexing |  | ||||||
|     // 4d (base = 0): |  | ||||||
|     // point 0  1  2  3  4  5  6  7  8 |  | ||||||
|     // dir   0  0  1  1  2  2  3  3  0 |  | ||||||
|     // disp +1 -1 +1 -1 +1 -1 +1 -1  0 |  | ||||||
|     // 5d (base = 1): |  | ||||||
|     // point 0  1  2  3  4  5  6  7  8 |  | ||||||
|     // dir   1  1  2  2  3  3  4  4  0 |  | ||||||
|     // disp +1 -1 +1 -1 +1 -1 +1 -1  0 |  | ||||||
|  |  | ||||||
|     if(dir == 0 and disp == 0) |  | ||||||
|       return 8; |  | ||||||
|     else // New indexing |  | ||||||
|       return (1 - disp) / 2 * 4 + dir - base; |  | ||||||
|     // else // Old indexing |  | ||||||
|     //   return (4 * (dir - base) + 1 - disp) / 2; |  | ||||||
|   } |  | ||||||
| }; | }; | ||||||
|    |    | ||||||
| template<class Fobj,class CComplex,int nbasis> | template<class Fobj,class CComplex,int nbasis> | ||||||
| @@ -296,7 +258,7 @@ public: | |||||||
| // Fine Object == (per site) type of fine field | // Fine Object == (per site) type of fine field | ||||||
| // nbasis      == number of deflation vectors | // nbasis      == number of deflation vectors | ||||||
| template<class Fobj,class CComplex,int nbasis> | template<class Fobj,class CComplex,int nbasis> | ||||||
| class CoarsenedMatrix : public CheckerBoardedSparseMatrixBase<Lattice<iVector<CComplex,nbasis > > >  { | class CoarsenedMatrix : public SparseMatrixBase<Lattice<iVector<CComplex,nbasis > > >  { | ||||||
| public: | public: | ||||||
|      |      | ||||||
|   typedef iVector<CComplex,nbasis >           siteVector; |   typedef iVector<CComplex,nbasis >           siteVector; | ||||||
| @@ -306,59 +268,33 @@ public: | |||||||
|   typedef iMatrix<CComplex,nbasis >  Cobj; |   typedef iMatrix<CComplex,nbasis >  Cobj; | ||||||
|   typedef Lattice< CComplex >   CoarseScalar; // used for inner products on fine field |   typedef Lattice< CComplex >   CoarseScalar; // used for inner products on fine field | ||||||
|   typedef Lattice<Fobj >        FineField; |   typedef Lattice<Fobj >        FineField; | ||||||
|   typedef CoarseVector FermionField; |  | ||||||
|  |  | ||||||
|   // enrich interface, use default implementation as in FermionOperator /////// |  | ||||||
|   void Dminus(CoarseVector const& in, CoarseVector& out) { out = in; } |  | ||||||
|   void DminusDag(CoarseVector const& in, CoarseVector& out) { out = in; } |  | ||||||
|   void ImportPhysicalFermionSource(CoarseVector const& input, CoarseVector& imported) { imported = input; } |  | ||||||
|   void ImportUnphysicalFermion(CoarseVector const& input, CoarseVector& imported) { imported = input; } |  | ||||||
|   void ExportPhysicalFermionSolution(CoarseVector const& solution, CoarseVector& exported) { exported = solution; }; |  | ||||||
|   void ExportPhysicalFermionSource(CoarseVector const& solution, CoarseVector& exported) { exported = solution; }; |  | ||||||
|  |  | ||||||
|   //////////////////// |   //////////////////// | ||||||
|   // Data members |   // Data members | ||||||
|   //////////////////// |   //////////////////// | ||||||
|   Geometry         geom; |   Geometry         geom; | ||||||
|   GridBase *       _grid;  |   GridBase *       _grid;  | ||||||
|   GridBase*        _cbgrid; |  | ||||||
|   int hermitian; |   int hermitian; | ||||||
|  |  | ||||||
|   CartesianStencil<siteVector,siteVector,int> Stencil;  |   CartesianStencil<siteVector,siteVector,int> Stencil;  | ||||||
|   CartesianStencil<siteVector,siteVector,int> StencilEven; |  | ||||||
|   CartesianStencil<siteVector,siteVector,int> StencilOdd; |  | ||||||
|  |  | ||||||
|   std::vector<CoarseMatrix> A; |   std::vector<CoarseMatrix> A; | ||||||
|   std::vector<CoarseMatrix> Aeven; |  | ||||||
|   std::vector<CoarseMatrix> Aodd; |  | ||||||
|  |  | ||||||
|   CoarseMatrix AselfInv; |  | ||||||
|   CoarseMatrix AselfInvEven; |  | ||||||
|   CoarseMatrix AselfInvOdd; |  | ||||||
|  |  | ||||||
|   Vector<RealD> dag_factor; |  | ||||||
|      |      | ||||||
|   /////////////////////// |   /////////////////////// | ||||||
|   // Interface |   // Interface | ||||||
|   /////////////////////// |   /////////////////////// | ||||||
|   GridBase * Grid(void)         { return _grid; };   // this is all the linalg routines need to know |   GridBase * Grid(void)         { return _grid; };   // this is all the linalg routines need to know | ||||||
|   GridBase * RedBlackGrid()     { return _cbgrid; }; |  | ||||||
|  |  | ||||||
|   int ConstEE() { return 0; } |  | ||||||
|  |  | ||||||
|   void M (const CoarseVector &in, CoarseVector &out) |   void M (const CoarseVector &in, CoarseVector &out) | ||||||
|   { |   { | ||||||
|     conformable(_grid,in.Grid()); |     conformable(_grid,in.Grid()); | ||||||
|     conformable(in.Grid(),out.Grid()); |     conformable(in.Grid(),out.Grid()); | ||||||
|     out.Checkerboard() = in.Checkerboard(); |  | ||||||
|  |  | ||||||
|     SimpleCompressor<siteVector> compressor; |     SimpleCompressor<siteVector> compressor; | ||||||
|  |  | ||||||
|     Stencil.HaloExchange(in,compressor); |     Stencil.HaloExchange(in,compressor); | ||||||
|     autoView( in_v , in, AcceleratorRead); |     autoView( in_v , in, AcceleratorRead); | ||||||
|     autoView( out_v , out, AcceleratorWrite); |     autoView( out_v , out, AcceleratorWrite); | ||||||
|     autoView( Stencil_v  , Stencil, AcceleratorRead); |  | ||||||
|     auto& geom_v = geom; |  | ||||||
|     typedef LatticeView<Cobj> Aview; |     typedef LatticeView<Cobj> Aview; | ||||||
|        |        | ||||||
|     Vector<Aview> AcceleratorViewContainer; |     Vector<Aview> AcceleratorViewContainer; | ||||||
| @@ -380,14 +316,14 @@ public: | |||||||
|       int ptype; |       int ptype; | ||||||
|       StencilEntry *SE; |       StencilEntry *SE; | ||||||
|  |  | ||||||
|       for(int point=0;point<geom_v.npoint;point++){ |       for(int point=0;point<geom.npoint;point++){ | ||||||
|  |  | ||||||
| 	SE=Stencil_v.GetEntry(ptype,point,ss); | 	SE=Stencil.GetEntry(ptype,point,ss); | ||||||
| 	   | 	   | ||||||
| 	if(SE->_is_local) {  | 	if(SE->_is_local) {  | ||||||
| 	  nbr = coalescedReadPermute(in_v[SE->_offset],ptype,SE->_permute); | 	  nbr = coalescedReadPermute(in_v[SE->_offset],ptype,SE->_permute); | ||||||
| 	} else { | 	} else { | ||||||
| 	  nbr = coalescedRead(Stencil_v.CommBuf()[SE->_offset]); | 	  nbr = coalescedRead(Stencil.CommBuf()[SE->_offset]); | ||||||
| 	} | 	} | ||||||
| 	acceleratorSynchronise(); | 	acceleratorSynchronise(); | ||||||
|  |  | ||||||
| @@ -408,74 +344,12 @@ public: | |||||||
|       return M(in,out); |       return M(in,out); | ||||||
|     } else { |     } else { | ||||||
|       // corresponds to Galerkin coarsening |       // corresponds to Galerkin coarsening | ||||||
|       return MdagNonHermitian(in, out); |       CoarseVector tmp(Grid()); | ||||||
|  |       G5C(tmp, in);  | ||||||
|  |       M(tmp, out); | ||||||
|  |       G5C(out, out); | ||||||
|     } |     } | ||||||
|   }; |   }; | ||||||
|  |  | ||||||
|   void MdagNonHermitian(const CoarseVector &in, CoarseVector &out) |  | ||||||
|   { |  | ||||||
|     conformable(_grid,in.Grid()); |  | ||||||
|     conformable(in.Grid(),out.Grid()); |  | ||||||
|     out.Checkerboard() = in.Checkerboard(); |  | ||||||
|  |  | ||||||
|     SimpleCompressor<siteVector> compressor; |  | ||||||
|  |  | ||||||
|     Stencil.HaloExchange(in,compressor); |  | ||||||
|     autoView( in_v , in, AcceleratorRead); |  | ||||||
|     autoView( out_v , out, AcceleratorWrite); |  | ||||||
|     autoView( Stencil_v  , Stencil, AcceleratorRead); |  | ||||||
|     auto& geom_v = geom; |  | ||||||
|     typedef LatticeView<Cobj> Aview; |  | ||||||
|  |  | ||||||
|     Vector<Aview> AcceleratorViewContainer; |  | ||||||
|  |  | ||||||
|     for(int p=0;p<geom.npoint;p++) AcceleratorViewContainer.push_back(A[p].View(AcceleratorRead)); |  | ||||||
|     Aview *Aview_p = & AcceleratorViewContainer[0]; |  | ||||||
|  |  | ||||||
|     const int Nsimd = CComplex::Nsimd(); |  | ||||||
|     typedef decltype(coalescedRead(in_v[0])) calcVector; |  | ||||||
|     typedef decltype(coalescedRead(in_v[0](0))) calcComplex; |  | ||||||
|  |  | ||||||
|     int osites=Grid()->oSites(); |  | ||||||
|  |  | ||||||
|     Vector<int> points(geom.npoint, 0); |  | ||||||
|     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, { |  | ||||||
|       int ss = sss/nbasis; |  | ||||||
|       int b  = sss%nbasis; |  | ||||||
|       calcComplex res = Zero(); |  | ||||||
|       calcVector nbr; |  | ||||||
|       int ptype; |  | ||||||
|       StencilEntry *SE; |  | ||||||
|  |  | ||||||
|       for(int p=0;p<geom_v.npoint;p++){ |  | ||||||
|         int point = points_p[p]; |  | ||||||
|  |  | ||||||
| 	SE=Stencil_v.GetEntry(ptype,point,ss); |  | ||||||
|  |  | ||||||
| 	if(SE->_is_local) { |  | ||||||
| 	  nbr = coalescedReadPermute(in_v[SE->_offset],ptype,SE->_permute); |  | ||||||
| 	} else { |  | ||||||
| 	  nbr = coalescedRead(Stencil_v.CommBuf()[SE->_offset]); |  | ||||||
| 	} |  | ||||||
| 	acceleratorSynchronise(); |  | ||||||
|  |  | ||||||
| 	for(int bb=0;bb<nbasis;bb++) { |  | ||||||
| 	  res = res + dag_factor_p[b*nbasis+bb]*coalescedRead(Aview_p[point][ss](b,bb))*nbr(bb); |  | ||||||
| 	} |  | ||||||
|       } |  | ||||||
|       coalescedWrite(out_v[ss](b),res); |  | ||||||
|       }); |  | ||||||
|  |  | ||||||
|     for(int p=0;p<geom.npoint;p++) AcceleratorViewContainer[p].ViewClose(); |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   void MdirComms(const CoarseVector &in) |   void MdirComms(const CoarseVector &in) | ||||||
|   { |   { | ||||||
|     SimpleCompressor<siteVector> compressor; |     SimpleCompressor<siteVector> compressor; | ||||||
| @@ -485,7 +359,6 @@ public: | |||||||
|   { |   { | ||||||
|     conformable(_grid,in.Grid()); |     conformable(_grid,in.Grid()); | ||||||
|     conformable(_grid,out.Grid()); |     conformable(_grid,out.Grid()); | ||||||
|     out.Checkerboard() = in.Checkerboard(); |  | ||||||
|  |  | ||||||
|     typedef LatticeView<Cobj> Aview; |     typedef LatticeView<Cobj> Aview; | ||||||
|     Vector<Aview> AcceleratorViewContainer; |     Vector<Aview> AcceleratorViewContainer; | ||||||
| @@ -494,7 +367,6 @@ public: | |||||||
|  |  | ||||||
|     autoView( out_v , out, AcceleratorWrite); |     autoView( out_v , out, AcceleratorWrite); | ||||||
|     autoView( in_v  , in, AcceleratorRead); |     autoView( in_v  , in, AcceleratorRead); | ||||||
|     autoView( Stencil_v  , Stencil, AcceleratorRead); |  | ||||||
|  |  | ||||||
|     const int Nsimd = CComplex::Nsimd(); |     const int Nsimd = CComplex::Nsimd(); | ||||||
|     typedef decltype(coalescedRead(in_v[0])) calcVector; |     typedef decltype(coalescedRead(in_v[0])) calcVector; | ||||||
| @@ -508,12 +380,12 @@ public: | |||||||
|       int ptype; |       int ptype; | ||||||
|       StencilEntry *SE; |       StencilEntry *SE; | ||||||
|  |  | ||||||
|       SE=Stencil_v.GetEntry(ptype,point,ss); |       SE=Stencil.GetEntry(ptype,point,ss); | ||||||
| 	   | 	   | ||||||
|       if(SE->_is_local) {  |       if(SE->_is_local) {  | ||||||
| 	nbr = coalescedReadPermute(in_v[SE->_offset],ptype,SE->_permute); | 	nbr = coalescedReadPermute(in_v[SE->_offset],ptype,SE->_permute); | ||||||
|       } else { |       } else { | ||||||
| 	nbr = coalescedRead(Stencil_v.CommBuf()[SE->_offset]); | 	nbr = coalescedRead(Stencil.CommBuf()[SE->_offset]); | ||||||
|       } |       } | ||||||
|       acceleratorSynchronise(); |       acceleratorSynchronise(); | ||||||
|  |  | ||||||
| @@ -541,7 +413,34 @@ public: | |||||||
|  |  | ||||||
|     this->MdirComms(in); |     this->MdirComms(in); | ||||||
|  |  | ||||||
|     MdirCalc(in,out,geom.point(dir,disp)); |     int ndim = in.Grid()->Nd(); | ||||||
|  |  | ||||||
|  |     ////////////// | ||||||
|  |     // 4D action like wilson | ||||||
|  |     // 0+ => 0  | ||||||
|  |     // 0- => 1 | ||||||
|  |     // 1+ => 2  | ||||||
|  |     // 1- => 3 | ||||||
|  |     // etc.. | ||||||
|  |     ////////////// | ||||||
|  |     // 5D action like DWF | ||||||
|  |     // 1+ => 0  | ||||||
|  |     // 1- => 1 | ||||||
|  |     // 2+ => 2  | ||||||
|  |     // 2- => 3 | ||||||
|  |     // etc.. | ||||||
|  |     auto point = [dir, disp, ndim](){ | ||||||
|  |       if(dir == 0 and disp == 0) | ||||||
|  | 	return 8; | ||||||
|  |       else if ( ndim==4 ) {  | ||||||
|  | 	return (4 * dir + 1 - disp) / 2; | ||||||
|  |       } else {  | ||||||
|  | 	return (4 * (dir-1) + 1 - disp) / 2; | ||||||
|  |       } | ||||||
|  |     }(); | ||||||
|  |  | ||||||
|  |     MdirCalc(in,out,point); | ||||||
|  |  | ||||||
|   }; |   }; | ||||||
|  |  | ||||||
|   void Mdiag(const CoarseVector &in, CoarseVector &out) |   void Mdiag(const CoarseVector &in, CoarseVector &out) | ||||||
| @@ -550,298 +449,23 @@ public: | |||||||
|     MdirCalc(in, out, point); // No comms |     MdirCalc(in, out, point); // No comms | ||||||
|   }; |   }; | ||||||
|  |  | ||||||
|   void Mooee(const CoarseVector &in, CoarseVector &out) { |  | ||||||
|     MooeeInternal(in, out, DaggerNo, InverseNo); |  | ||||||
|   } |  | ||||||
|    |    | ||||||
|   void MooeeInv(const CoarseVector &in, CoarseVector &out) { |  CoarsenedMatrix(GridCartesian &CoarseGrid, int hermitian_=0) 	:  | ||||||
|     MooeeInternal(in, out, DaggerNo, InverseYes); |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   void MooeeDag(const CoarseVector &in, CoarseVector &out) { |  | ||||||
|     MooeeInternal(in, out, DaggerYes, InverseNo); |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   void MooeeInvDag(const CoarseVector &in, CoarseVector &out) { |  | ||||||
|     MooeeInternal(in, out, DaggerYes, InverseYes); |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   void Meooe(const CoarseVector &in, CoarseVector &out) { |  | ||||||
|     if(in.Checkerboard() == Odd) { |  | ||||||
|       DhopEO(in, out, DaggerNo); |  | ||||||
|     } else { |  | ||||||
|       DhopOE(in, out, DaggerNo); |  | ||||||
|     } |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   void MeooeDag(const CoarseVector &in, CoarseVector &out) { |  | ||||||
|     if(in.Checkerboard() == Odd) { |  | ||||||
|       DhopEO(in, out, DaggerYes); |  | ||||||
|     } else { |  | ||||||
|       DhopOE(in, out, DaggerYes); |  | ||||||
|     } |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   void Dhop(const CoarseVector &in, CoarseVector &out, int dag) { |  | ||||||
|     conformable(in.Grid(), _grid); // verifies full grid |  | ||||||
|     conformable(in.Grid(), out.Grid()); |  | ||||||
|  |  | ||||||
|     out.Checkerboard() = in.Checkerboard(); |  | ||||||
|  |  | ||||||
|     DhopInternal(Stencil, A, in, out, dag); |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   void DhopOE(const CoarseVector &in, CoarseVector &out, int dag) { |  | ||||||
|     conformable(in.Grid(), _cbgrid);    // verifies half grid |  | ||||||
|     conformable(in.Grid(), out.Grid()); // drops the cb check |  | ||||||
|  |  | ||||||
|     assert(in.Checkerboard() == Even); |  | ||||||
|     out.Checkerboard() = Odd; |  | ||||||
|  |  | ||||||
|     DhopInternal(StencilEven, Aodd, in, out, dag); |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   void DhopEO(const CoarseVector &in, CoarseVector &out, int dag) { |  | ||||||
|     conformable(in.Grid(), _cbgrid);    // verifies half grid |  | ||||||
|     conformable(in.Grid(), out.Grid()); // drops the cb check |  | ||||||
|  |  | ||||||
|     assert(in.Checkerboard() == Odd); |  | ||||||
|     out.Checkerboard() = Even; |  | ||||||
|  |  | ||||||
|     DhopInternal(StencilOdd, Aeven, in, out, dag); |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   void MooeeInternal(const CoarseVector &in, CoarseVector &out, int dag, int inv) { |  | ||||||
|     out.Checkerboard() = in.Checkerboard(); |  | ||||||
|     assert(in.Checkerboard() == Odd || in.Checkerboard() == Even); |  | ||||||
|  |  | ||||||
|     CoarseMatrix *Aself = nullptr; |  | ||||||
|     if(in.Grid()->_isCheckerBoarded) { |  | ||||||
|       if(in.Checkerboard() == Odd) { |  | ||||||
|         Aself = (inv) ? &AselfInvOdd : &Aodd[geom.npoint-1]; |  | ||||||
|         DselfInternal(StencilOdd, *Aself, in, out, dag); |  | ||||||
|       } else { |  | ||||||
|         Aself = (inv) ? &AselfInvEven : &Aeven[geom.npoint-1]; |  | ||||||
|         DselfInternal(StencilEven, *Aself, in, out, dag); |  | ||||||
|       } |  | ||||||
|     } else { |  | ||||||
|       Aself = (inv) ? &AselfInv : &A[geom.npoint-1]; |  | ||||||
|       DselfInternal(Stencil, *Aself, in, out, dag); |  | ||||||
|     } |  | ||||||
|     assert(Aself != nullptr); |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   void DselfInternal(CartesianStencil<siteVector,siteVector,int> &st, CoarseMatrix &a, |  | ||||||
|                        const CoarseVector &in, CoarseVector &out, int dag) { |  | ||||||
|     int point = geom.npoint-1; |  | ||||||
|     autoView( out_v, out, AcceleratorWrite); |  | ||||||
|     autoView( in_v,  in,  AcceleratorRead); |  | ||||||
|     autoView( st_v,  st,  AcceleratorRead); |  | ||||||
|     autoView( a_v,   a,   AcceleratorRead); |  | ||||||
|  |  | ||||||
|     const int Nsimd = CComplex::Nsimd(); |  | ||||||
|     typedef decltype(coalescedRead(in_v[0])) calcVector; |  | ||||||
|     typedef decltype(coalescedRead(in_v[0](0))) calcComplex; |  | ||||||
|  |  | ||||||
|     RealD* dag_factor_p = &dag_factor[0]; |  | ||||||
|  |  | ||||||
|     if(dag) { |  | ||||||
|       accelerator_for(sss, in.Grid()->oSites()*nbasis, Nsimd, { |  | ||||||
|         int ss = sss/nbasis; |  | ||||||
|         int b  = sss%nbasis; |  | ||||||
|         calcComplex res = Zero(); |  | ||||||
|         calcVector nbr; |  | ||||||
|         int ptype; |  | ||||||
|         StencilEntry *SE; |  | ||||||
|  |  | ||||||
|         SE=st_v.GetEntry(ptype,point,ss); |  | ||||||
|  |  | ||||||
|         if(SE->_is_local) { |  | ||||||
|           nbr = coalescedReadPermute(in_v[SE->_offset],ptype,SE->_permute); |  | ||||||
|         } else { |  | ||||||
|           nbr = coalescedRead(st_v.CommBuf()[SE->_offset]); |  | ||||||
|         } |  | ||||||
|         acceleratorSynchronise(); |  | ||||||
|  |  | ||||||
|         for(int bb=0;bb<nbasis;bb++) { |  | ||||||
|           res = res + dag_factor_p[b*nbasis+bb]*coalescedRead(a_v[ss](b,bb))*nbr(bb); |  | ||||||
|         } |  | ||||||
|         coalescedWrite(out_v[ss](b),res); |  | ||||||
|       }); |  | ||||||
|     } else { |  | ||||||
|       accelerator_for(sss, in.Grid()->oSites()*nbasis, Nsimd, { |  | ||||||
|         int ss = sss/nbasis; |  | ||||||
|         int b  = sss%nbasis; |  | ||||||
|         calcComplex res = Zero(); |  | ||||||
|         calcVector nbr; |  | ||||||
|         int ptype; |  | ||||||
|         StencilEntry *SE; |  | ||||||
|  |  | ||||||
|         SE=st_v.GetEntry(ptype,point,ss); |  | ||||||
|  |  | ||||||
|         if(SE->_is_local) { |  | ||||||
|           nbr = coalescedReadPermute(in_v[SE->_offset],ptype,SE->_permute); |  | ||||||
|         } else { |  | ||||||
|           nbr = coalescedRead(st_v.CommBuf()[SE->_offset]); |  | ||||||
|         } |  | ||||||
|         acceleratorSynchronise(); |  | ||||||
|  |  | ||||||
|         for(int bb=0;bb<nbasis;bb++) { |  | ||||||
|           res = res + coalescedRead(a_v[ss](b,bb))*nbr(bb); |  | ||||||
|         } |  | ||||||
|         coalescedWrite(out_v[ss](b),res); |  | ||||||
|       }); |  | ||||||
|     } |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   void DhopInternal(CartesianStencil<siteVector,siteVector,int> &st, std::vector<CoarseMatrix> &a, |  | ||||||
|                     const CoarseVector &in, CoarseVector &out, int dag) { |  | ||||||
|     SimpleCompressor<siteVector> compressor; |  | ||||||
|  |  | ||||||
|     st.HaloExchange(in,compressor); |  | ||||||
|     autoView( in_v,  in,  AcceleratorRead); |  | ||||||
|     autoView( out_v, out, AcceleratorWrite); |  | ||||||
|     autoView( st_v , st,  AcceleratorRead); |  | ||||||
|     typedef LatticeView<Cobj> Aview; |  | ||||||
|  |  | ||||||
|     // determine in what order we need the points |  | ||||||
|     int npoint = geom.npoint-1; |  | ||||||
|     Vector<int> points(npoint, 0); |  | ||||||
|     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]; |  | ||||||
|  |  | ||||||
|     const int Nsimd = CComplex::Nsimd(); |  | ||||||
|     typedef decltype(coalescedRead(in_v[0])) calcVector; |  | ||||||
|     typedef decltype(coalescedRead(in_v[0](0))) calcComplex; |  | ||||||
|  |  | ||||||
|     RealD* dag_factor_p = &dag_factor[0]; |  | ||||||
|  |  | ||||||
|     if(dag) { |  | ||||||
|       accelerator_for(sss, in.Grid()->oSites()*nbasis, Nsimd, { |  | ||||||
|         int ss = sss/nbasis; |  | ||||||
|         int b  = sss%nbasis; |  | ||||||
|         calcComplex res = Zero(); |  | ||||||
|         calcVector nbr; |  | ||||||
|         int ptype; |  | ||||||
|         StencilEntry *SE; |  | ||||||
|  |  | ||||||
|         for(int p=0;p<npoint;p++){ |  | ||||||
|           int point = points_p[p]; |  | ||||||
|           SE=st_v.GetEntry(ptype,point,ss); |  | ||||||
|  |  | ||||||
|           if(SE->_is_local) { |  | ||||||
|             nbr = coalescedReadPermute(in_v[SE->_offset],ptype,SE->_permute); |  | ||||||
|           } else { |  | ||||||
|             nbr = coalescedRead(st_v.CommBuf()[SE->_offset]); |  | ||||||
|           } |  | ||||||
|           acceleratorSynchronise(); |  | ||||||
|  |  | ||||||
|           for(int bb=0;bb<nbasis;bb++) { |  | ||||||
|             res = res + dag_factor_p[b*nbasis+bb]*coalescedRead(Aview_p[point][ss](b,bb))*nbr(bb); |  | ||||||
|           } |  | ||||||
|         } |  | ||||||
|         coalescedWrite(out_v[ss](b),res); |  | ||||||
|       }); |  | ||||||
|     } else { |  | ||||||
|       accelerator_for(sss, in.Grid()->oSites()*nbasis, Nsimd, { |  | ||||||
|         int ss = sss/nbasis; |  | ||||||
|         int b  = sss%nbasis; |  | ||||||
|         calcComplex res = Zero(); |  | ||||||
|         calcVector nbr; |  | ||||||
|         int ptype; |  | ||||||
|         StencilEntry *SE; |  | ||||||
|  |  | ||||||
|         for(int p=0;p<npoint;p++){ |  | ||||||
|           int point = points_p[p]; |  | ||||||
|           SE=st_v.GetEntry(ptype,point,ss); |  | ||||||
|  |  | ||||||
|           if(SE->_is_local) { |  | ||||||
|             nbr = coalescedReadPermute(in_v[SE->_offset],ptype,SE->_permute); |  | ||||||
|           } else { |  | ||||||
|             nbr = coalescedRead(st_v.CommBuf()[SE->_offset]); |  | ||||||
|           } |  | ||||||
|           acceleratorSynchronise(); |  | ||||||
|  |  | ||||||
|           for(int bb=0;bb<nbasis;bb++) { |  | ||||||
|             res = res + coalescedRead(Aview_p[point][ss](b,bb))*nbr(bb); |  | ||||||
|           } |  | ||||||
|         } |  | ||||||
|         coalescedWrite(out_v[ss](b),res); |  | ||||||
|       }); |  | ||||||
|     } |  | ||||||
|  |  | ||||||
|     for(int p=0;p<npoint;p++) AcceleratorViewContainer[p].ViewClose(); |  | ||||||
|   } |  | ||||||
|    |  | ||||||
|   CoarsenedMatrix(GridCartesian &CoarseGrid, int hermitian_=0) 	: |  | ||||||
|     _grid(&CoarseGrid), |     _grid(&CoarseGrid), | ||||||
|     _cbgrid(new GridRedBlackCartesian(&CoarseGrid)), |  | ||||||
|     geom(CoarseGrid._ndimension), |     geom(CoarseGrid._ndimension), | ||||||
|     hermitian(hermitian_), |     hermitian(hermitian_), | ||||||
|     Stencil(&CoarseGrid,geom.npoint,Even,geom.directions,geom.displacements,0), |     Stencil(&CoarseGrid,geom.npoint,Even,geom.directions,geom.displacements,0), | ||||||
|     StencilEven(_cbgrid,geom.npoint,Even,geom.directions,geom.displacements,0), |       A(geom.npoint,&CoarseGrid) | ||||||
|     StencilOdd(_cbgrid,geom.npoint,Odd,geom.directions,geom.displacements,0), |  | ||||||
|     A(geom.npoint,&CoarseGrid), |  | ||||||
|     Aeven(geom.npoint,_cbgrid), |  | ||||||
|     Aodd(geom.npoint,_cbgrid), |  | ||||||
|     AselfInv(&CoarseGrid), |  | ||||||
|     AselfInvEven(_cbgrid), |  | ||||||
|     AselfInvOdd(_cbgrid), |  | ||||||
|     dag_factor(nbasis*nbasis) |  | ||||||
|   { |   { | ||||||
|     fillFactor(); |  | ||||||
|   }; |   }; | ||||||
|  |  | ||||||
|   CoarsenedMatrix(GridCartesian &CoarseGrid, GridRedBlackCartesian &CoarseRBGrid, int hermitian_=0) 	: |  | ||||||
|  |  | ||||||
|     _grid(&CoarseGrid), |  | ||||||
|     _cbgrid(&CoarseRBGrid), |  | ||||||
|     geom(CoarseGrid._ndimension), |  | ||||||
|     hermitian(hermitian_), |  | ||||||
|     Stencil(&CoarseGrid,geom.npoint,Even,geom.directions,geom.displacements,0), |  | ||||||
|     StencilEven(&CoarseRBGrid,geom.npoint,Even,geom.directions,geom.displacements,0), |  | ||||||
|     StencilOdd(&CoarseRBGrid,geom.npoint,Odd,geom.directions,geom.displacements,0), |  | ||||||
|     A(geom.npoint,&CoarseGrid), |  | ||||||
|     Aeven(geom.npoint,&CoarseRBGrid), |  | ||||||
|     Aodd(geom.npoint,&CoarseRBGrid), |  | ||||||
|     AselfInv(&CoarseGrid), |  | ||||||
|     AselfInvEven(&CoarseRBGrid), |  | ||||||
|     AselfInvOdd(&CoarseRBGrid), |  | ||||||
|     dag_factor(nbasis*nbasis) |  | ||||||
|   { |  | ||||||
|     fillFactor(); |  | ||||||
|   }; |  | ||||||
|  |  | ||||||
|   void fillFactor() { |  | ||||||
|     Eigen::MatrixXd dag_factor_eigen = Eigen::MatrixXd::Ones(nbasis, nbasis); |  | ||||||
|     if(!hermitian) { |  | ||||||
|       const int nb = nbasis/2; |  | ||||||
|       dag_factor_eigen.block(0,nb,nb,nb) *= -1.0; |  | ||||||
|       dag_factor_eigen.block(nb,0,nb,nb) *= -1.0; |  | ||||||
|     } |  | ||||||
|  |  | ||||||
|     // GPU readable prefactor |  | ||||||
|     thread_for(i, nbasis*nbasis, { |  | ||||||
|       int j = i/nbasis; |  | ||||||
|       int k = i%nbasis; |  | ||||||
|       dag_factor[i] = dag_factor_eigen(j, k); |  | ||||||
|     }); |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   void CoarsenOperator(GridBase *FineGrid,LinearOperatorBase<Lattice<Fobj> > &linop, |   void CoarsenOperator(GridBase *FineGrid,LinearOperatorBase<Lattice<Fobj> > &linop, | ||||||
| 		       Aggregation<Fobj,CComplex,nbasis> & Subspace) | 		       Aggregation<Fobj,CComplex,nbasis> & Subspace) | ||||||
|   { |   { | ||||||
|     typedef Lattice<typename Fobj::tensor_reduced> FineComplexField; |     typedef Lattice<typename Fobj::tensor_reduced> FineComplexField; | ||||||
|     typedef typename Fobj::scalar_type scalar_type; |     typedef typename Fobj::scalar_type scalar_type; | ||||||
|  |  | ||||||
|     std::cout << GridLogMessage<< "CoarsenMatrix "<< std::endl; |  | ||||||
|  |  | ||||||
|     FineComplexField one(FineGrid); one=scalar_type(1.0,0.0); |     FineComplexField one(FineGrid); one=scalar_type(1.0,0.0); | ||||||
|     FineComplexField zero(FineGrid); zero=scalar_type(0.0,0.0); |     FineComplexField zero(FineGrid); zero=scalar_type(0.0,0.0); | ||||||
|  |  | ||||||
| @@ -872,13 +496,11 @@ public: | |||||||
|  |  | ||||||
|     CoarseScalar InnerProd(Grid());  |     CoarseScalar InnerProd(Grid());  | ||||||
|  |  | ||||||
|     std::cout << GridLogMessage<< "CoarsenMatrix Orthog "<< std::endl; |  | ||||||
|     // Orthogonalise the subblocks over the basis |     // Orthogonalise the subblocks over the basis | ||||||
|     blockOrthogonalise(InnerProd,Subspace.subspace); |     blockOrthogonalise(InnerProd,Subspace.subspace); | ||||||
|  |  | ||||||
|     // Compute the matrix elements of linop between this orthonormal |     // Compute the matrix elements of linop between this orthonormal | ||||||
|     // set of vectors. |     // set of vectors. | ||||||
|     std::cout << GridLogMessage<< "CoarsenMatrix masks "<< std::endl; |  | ||||||
|     int self_stencil=-1; |     int self_stencil=-1; | ||||||
|     for(int p=0;p<geom.npoint;p++) |     for(int p=0;p<geom.npoint;p++) | ||||||
|     {  |     {  | ||||||
| @@ -917,7 +539,7 @@ public: | |||||||
|  |  | ||||||
|       phi=Subspace.subspace[i]; |       phi=Subspace.subspace[i]; | ||||||
|  |  | ||||||
|       std::cout << GridLogMessage<< "CoarsenMatrix vector "<<i << std::endl; |       //      std::cout << GridLogMessage<< "CoarsenMatrix vector "<<i << std::endl; | ||||||
|       linop.OpDirAll(phi,Mphi_p); |       linop.OpDirAll(phi,Mphi_p); | ||||||
|       linop.OpDiag  (phi,Mphi_p[geom.npoint-1]); |       linop.OpDiag  (phi,Mphi_p[geom.npoint-1]); | ||||||
|  |  | ||||||
| @@ -946,18 +568,6 @@ public: | |||||||
| 	    autoView( A_self  , A[self_stencil], AcceleratorWrite); | 	    autoView( A_self  , A[self_stencil], AcceleratorWrite); | ||||||
|  |  | ||||||
| 	    accelerator_for(ss, Grid()->oSites(), Fobj::Nsimd(),{ coalescedWrite(A_p[ss](j,i),oZProj_v(ss)); }); | 	    accelerator_for(ss, Grid()->oSites(), Fobj::Nsimd(),{ coalescedWrite(A_p[ss](j,i),oZProj_v(ss)); }); | ||||||
| 	    if ( hermitian && (disp==-1) ) { |  | ||||||
| 	      for(int pp=0;pp<geom.npoint;pp++){// Find the opposite link and set <j|A|i> = <i|A|j>* |  | ||||||
| 		int dirp   = geom.directions[pp]; |  | ||||||
| 		int dispp  = geom.displacements[pp]; |  | ||||||
| 		if ( (dirp==dir) && (dispp==1) ){ |  | ||||||
| 		  auto sft = conjugate(Cshift(oZProj,dir,1)); |  | ||||||
| 		  autoView( sft_v    ,  sft  , AcceleratorWrite); |  | ||||||
| 		  autoView( A_pp     ,  A[pp], AcceleratorWrite); |  | ||||||
| 		  accelerator_for(ss, Grid()->oSites(), Fobj::Nsimd(),{ coalescedWrite(A_pp[ss](i,j),sft_v(ss)); }); |  | ||||||
| 		} |  | ||||||
| 	      } |  | ||||||
| 	    } |  | ||||||
|  |  | ||||||
| 	  } | 	  } | ||||||
| 	} | 	} | ||||||
| @@ -996,54 +606,28 @@ public: | |||||||
|     } |     } | ||||||
|     if(hermitian) { |     if(hermitian) { | ||||||
|       std::cout << GridLogMessage << " ForceHermitian, new code "<<std::endl; |       std::cout << GridLogMessage << " ForceHermitian, new code "<<std::endl; | ||||||
|  |       ForceHermitian(); | ||||||
|     } |     } | ||||||
|  |  | ||||||
|     InvertSelfStencilLink(); std::cout << GridLogMessage << "Coarse self link inverted" << std::endl; |  | ||||||
|     FillHalfCbs(); std::cout << GridLogMessage << "Coarse half checkerboards filled" << std::endl; |  | ||||||
|   } |   } | ||||||
|  |  | ||||||
|   void InvertSelfStencilLink() { |   void ForceHermitian(void) { | ||||||
|     std::cout << GridLogDebug << "CoarsenedMatrix::InvertSelfStencilLink" << std::endl; |     CoarseMatrix Diff  (Grid()); | ||||||
|     int localVolume = Grid()->lSites(); |     for(int p=0;p<geom.npoint;p++){ | ||||||
|  |       int dir   = geom.directions[p]; | ||||||
|     typedef typename Cobj::scalar_object scalar_object; |       int disp  = geom.displacements[p]; | ||||||
|  |       if(disp==-1) { | ||||||
|     autoView(Aself_v,    A[geom.npoint-1], CpuRead); | 	// Find the opposite link | ||||||
|     autoView(AselfInv_v, AselfInv,         CpuWrite); | 	for(int pp=0;pp<geom.npoint;pp++){ | ||||||
|     thread_for(site, localVolume, { // NOTE: Not able to bring this to GPU because of Eigen + peek/poke | 	  int dirp   = geom.directions[pp]; | ||||||
|       Eigen::MatrixXcd selfLinkEigen    = Eigen::MatrixXcd::Zero(nbasis, nbasis); | 	  int dispp  = geom.displacements[pp]; | ||||||
|       Eigen::MatrixXcd selfLinkInvEigen = Eigen::MatrixXcd::Zero(nbasis, nbasis); | 	  if ( (dirp==dir) && (dispp==1) ){ | ||||||
|  | 	    //	    Diff = adj(Cshift(A[p],dir,1)) - A[pp];  | ||||||
|       scalar_object selfLink    = Zero(); | 	    //	    std::cout << GridLogMessage<<" Replacing stencil leg "<<pp<<" with leg "<<p<< " diff "<<norm2(Diff) <<std::endl; | ||||||
|       scalar_object selfLinkInv = Zero(); | 	    A[pp] = adj(Cshift(A[p],dir,1)); | ||||||
|  | 	  } | ||||||
|       Coordinate lcoor; | 	} | ||||||
|  |       } | ||||||
|       Grid()->LocalIndexToLocalCoor(site, lcoor); |  | ||||||
|       peekLocalSite(selfLink, Aself_v, lcoor); |  | ||||||
|  |  | ||||||
|       for (int i = 0; i < nbasis; ++i) |  | ||||||
|         for (int j = 0; j < nbasis; ++j) |  | ||||||
|           selfLinkEigen(i, j) = static_cast<ComplexD>(TensorRemove(selfLink(i, j))); |  | ||||||
|  |  | ||||||
|       selfLinkInvEigen = selfLinkEigen.inverse(); |  | ||||||
|  |  | ||||||
|       for(int i = 0; i < nbasis; ++i) |  | ||||||
|         for(int j = 0; j < nbasis; ++j) |  | ||||||
|           selfLinkInv(i, j) = selfLinkInvEigen(i, j); |  | ||||||
|  |  | ||||||
|       pokeLocalSite(selfLinkInv, AselfInv_v, lcoor); |  | ||||||
|     }); |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   void FillHalfCbs() { |  | ||||||
|     std::cout << GridLogDebug << "CoarsenedMatrix::FillHalfCbs" << std::endl; |  | ||||||
|     for(int p = 0; p < geom.npoint; ++p) { |  | ||||||
|       pickCheckerboard(Even, Aeven[p], A[p]); |  | ||||||
|       pickCheckerboard(Odd, Aodd[p], A[p]); |  | ||||||
|     } |     } | ||||||
|     pickCheckerboard(Even, AselfInvEven, AselfInv); |  | ||||||
|     pickCheckerboard(Odd, AselfInvOdd, AselfInv); |  | ||||||
|   } |   } | ||||||
| }; | }; | ||||||
|  |  | ||||||
|   | |||||||
| @@ -136,7 +136,7 @@ public: | |||||||
|     flops=0; |     flops=0; | ||||||
|     usec =0; |     usec =0; | ||||||
|     Coordinate layout(Nd,1); |     Coordinate layout(Nd,1); | ||||||
|     sgrid = new GridCartesian(dimensions,layout,processors,*grid); |     sgrid = new GridCartesian(dimensions,layout,processors); | ||||||
|   }; |   }; | ||||||
|      |      | ||||||
|   ~FFT ( void)  { |   ~FFT ( void)  { | ||||||
| @@ -182,7 +182,7 @@ public: | |||||||
|     pencil_gd[dim] = G*processors[dim]; |     pencil_gd[dim] = G*processors[dim]; | ||||||
|        |        | ||||||
|     // Pencil global vol LxLxGxLxL per node |     // Pencil global vol LxLxGxLxL per node | ||||||
|     GridCartesian pencil_g(pencil_gd,layout,processors,*vgrid); |     GridCartesian pencil_g(pencil_gd,layout,processors); | ||||||
|        |        | ||||||
|     // Construct pencils |     // Construct pencils | ||||||
|     typedef typename vobj::scalar_object sobj; |     typedef typename vobj::scalar_object sobj; | ||||||
|   | |||||||
| @@ -530,16 +530,6 @@ public: | |||||||
| template<class Field> class LinearFunction { | template<class Field> class LinearFunction { | ||||||
| public: | public: | ||||||
|   virtual void operator() (const Field &in, Field &out) = 0; |   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> { | template<class Field> class IdentityLinearFunction : public LinearFunction<Field> { | ||||||
|   | |||||||
| @@ -54,23 +54,15 @@ class DeflatedGuesser: public LinearFunction<Field> { | |||||||
| private: | private: | ||||||
|   const std::vector<Field> &evec; |   const std::vector<Field> &evec; | ||||||
|   const std::vector<RealD> &eval; |   const std::vector<RealD> &eval; | ||||||
|   const unsigned int       N; |  | ||||||
|  |  | ||||||
| public: | public: | ||||||
|  |  | ||||||
|   DeflatedGuesser(const std::vector<Field> & _evec,const std::vector<RealD> & _eval) |   DeflatedGuesser(const std::vector<Field> & _evec,const std::vector<RealD> & _eval) : evec(_evec), eval(_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()); |  | ||||||
|   }  |  | ||||||
|  |  | ||||||
|   virtual void operator()(const Field &src,Field &guess) { |   virtual void operator()(const Field &src,Field &guess) { | ||||||
|     guess = Zero(); |     guess = Zero(); | ||||||
|  |     assert(evec.size()==eval.size()); | ||||||
|  |     auto N = evec.size(); | ||||||
|     for (int i=0;i<N;i++) { |     for (int i=0;i<N;i++) { | ||||||
|       const Field& tmp = evec[i]; |       const Field& tmp = evec[i]; | ||||||
|       axpy(guess,TensorRemove(innerProduct(tmp,src)) / eval[i],tmp,guess); |       axpy(guess,TensorRemove(innerProduct(tmp,src)) / eval[i],tmp,guess); | ||||||
|   | |||||||
| @@ -132,31 +132,6 @@ namespace Grid { | |||||||
|       (*this)(_Matrix,in,out,guess); |       (*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> |     template<class Guesser> | ||||||
|     void operator()(Matrix &_Matrix, const std::vector<Field> &in, std::vector<Field> &out,Guesser &guess)  |     void operator()(Matrix &_Matrix, const std::vector<Field> &in, std::vector<Field> &out,Guesser &guess)  | ||||||
|     { |     { | ||||||
| @@ -175,29 +150,24 @@ namespace Grid { | |||||||
|       //////////////////////////////////////////////// |       //////////////////////////////////////////////// | ||||||
|       // Prepare RedBlack source |       // Prepare RedBlack source | ||||||
|       //////////////////////////////////////////////// |       //////////////////////////////////////////////// | ||||||
|       RedBlackSource(_Matrix,in,src_o); |       for(int b=0;b<nblock;b++){ | ||||||
| 	//      for(int b=0;b<nblock;b++){ | 	RedBlackSource(_Matrix,in[b],tmp,src_o[b]); | ||||||
| 	//	RedBlackSource(_Matrix,in[b],tmp,src_o[b]); |       } | ||||||
| 	//      } |  | ||||||
|        |  | ||||||
|       //////////////////////////////////////////////// |       //////////////////////////////////////////////// | ||||||
|       // Make the guesses |       // Make the guesses | ||||||
|       //////////////////////////////////////////////// |       //////////////////////////////////////////////// | ||||||
|       if ( subGuess ) guess_save.resize(nblock,grid); |       if ( subGuess ) guess_save.resize(nblock,grid); | ||||||
|  |  | ||||||
|        |       for(int b=0;b<nblock;b++){ | ||||||
|       if(useSolnAsInitGuess) { |         if(useSolnAsInitGuess) { | ||||||
|         for(int b=0;b<nblock;b++){ |  | ||||||
|           pickCheckerboard(Odd, sol_o[b], out[b]); |           pickCheckerboard(Odd, sol_o[b], out[b]); | ||||||
|  |         } else { | ||||||
|  |           guess(src_o[b],sol_o[b]);  | ||||||
|         } |         } | ||||||
|       } else { |  | ||||||
|         guess(src_o, sol_o);  |  | ||||||
|       } |  | ||||||
|  |  | ||||||
| 	    if ( subGuess ) {  | 	if ( subGuess ) {  | ||||||
|         for(int b=0;b<nblock;b++){ | 	  guess_save[b] = sol_o[b]; | ||||||
|           guess_save[b] = sol_o[b]; | 	} | ||||||
|         } |  | ||||||
|       } |       } | ||||||
|       ////////////////////////////////////////////////////////////// |       ////////////////////////////////////////////////////////////// | ||||||
|       // Call the block solver |       // Call the block solver | ||||||
|   | |||||||
| @@ -173,8 +173,7 @@ template<class T> using cshiftAllocator = devAllocator<T>; | |||||||
| template<class T> using cshiftAllocator = std::allocator<T>; | template<class T> using cshiftAllocator = std::allocator<T>; | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
| template<class T> using Vector        = std::vector<T,uvmAllocator<T> >;            | template<class T> using Vector     = std::vector<T,uvmAllocator<T> >;            | ||||||
| template<class T> using stencilVector = std::vector<T,alignedAllocator<T> >;            |  | ||||||
| template<class T> using commVector = std::vector<T,devAllocator<T> >; | template<class T> using commVector = std::vector<T,devAllocator<T> >; | ||||||
| template<class T> using cshiftVector = std::vector<T,cshiftAllocator<T> >; | template<class T> using cshiftVector = std::vector<T,cshiftAllocator<T> >; | ||||||
|  |  | ||||||
|   | |||||||
| @@ -34,6 +34,8 @@ NAMESPACE_BEGIN(Grid); | |||||||
|  |  | ||||||
| // Move control to configure.ac and Config.h? | // Move control to configure.ac and Config.h? | ||||||
|  |  | ||||||
|  | #define ALLOCATION_CACHE | ||||||
|  | #define GRID_ALLOC_ALIGN (2*1024*1024) | ||||||
| #define GRID_ALLOC_SMALL_LIMIT (4096) | #define GRID_ALLOC_SMALL_LIMIT (4096) | ||||||
|  |  | ||||||
| /*Pinning pages is costly*/ | /*Pinning pages is costly*/ | ||||||
|   | |||||||
| @@ -1,12 +1,11 @@ | |||||||
| #include <Grid/GridCore.h> | #include <Grid/GridCore.h> | ||||||
|  |  | ||||||
| #ifndef GRID_UVM | #ifndef GRID_UVM | ||||||
|  |  | ||||||
| #warning "Using explicit device memory copies" | #warning "Using explicit device memory copies" | ||||||
| NAMESPACE_BEGIN(Grid); | NAMESPACE_BEGIN(Grid); | ||||||
| //define dprintf(...) printf ( __VA_ARGS__ ); fflush(stdout); |  | ||||||
| #define dprintf(...) | #define dprintf(...) | ||||||
|  |  | ||||||
|  |  | ||||||
| //////////////////////////////////////////////////////////// | //////////////////////////////////////////////////////////// | ||||||
| // For caching copies of data on device | // For caching copies of data on device | ||||||
| //////////////////////////////////////////////////////////// | //////////////////////////////////////////////////////////// | ||||||
| @@ -104,7 +103,7 @@ void MemoryManager::AccDiscard(AcceleratorViewEntry &AccCache) | |||||||
|   /////////////////////////////////////////////////////////// |   /////////////////////////////////////////////////////////// | ||||||
|   assert(AccCache.state!=Empty); |   assert(AccCache.state!=Empty); | ||||||
|    |    | ||||||
|    dprintf("MemoryManager: Discard(%llx) %llx\n",(uint64_t)AccCache.CpuPtr,(uint64_t)AccCache.AccPtr);  |   //  dprintf("MemoryManager: Discard(%llx) %llx\n",(uint64_t)AccCache.CpuPtr,(uint64_t)AccCache.AccPtr);  | ||||||
|   assert(AccCache.accLock==0); |   assert(AccCache.accLock==0); | ||||||
|   assert(AccCache.cpuLock==0); |   assert(AccCache.cpuLock==0); | ||||||
|   assert(AccCache.CpuPtr!=(uint64_t)NULL); |   assert(AccCache.CpuPtr!=(uint64_t)NULL); | ||||||
| @@ -112,7 +111,7 @@ void MemoryManager::AccDiscard(AcceleratorViewEntry &AccCache) | |||||||
|     AcceleratorFree((void *)AccCache.AccPtr,AccCache.bytes); |     AcceleratorFree((void *)AccCache.AccPtr,AccCache.bytes); | ||||||
|     DeviceBytes   -=AccCache.bytes; |     DeviceBytes   -=AccCache.bytes; | ||||||
|     LRUremove(AccCache); |     LRUremove(AccCache); | ||||||
|     dprintf("MemoryManager: Free(%llx) LRU %lld Total %lld\n",(uint64_t)AccCache.AccPtr,DeviceLRUBytes,DeviceBytes);   |     //    dprintf("MemoryManager: Free(%llx) LRU %lld Total %lld\n",(uint64_t)AccCache.AccPtr,DeviceLRUBytes,DeviceBytes);   | ||||||
|   } |   } | ||||||
|   uint64_t CpuPtr = AccCache.CpuPtr; |   uint64_t CpuPtr = AccCache.CpuPtr; | ||||||
|   EntryErase(CpuPtr); |   EntryErase(CpuPtr); | ||||||
| @@ -126,7 +125,7 @@ void MemoryManager::Evict(AcceleratorViewEntry &AccCache) | |||||||
|   /////////////////////////////////////////////////////////////////////////// |   /////////////////////////////////////////////////////////////////////////// | ||||||
|   assert(AccCache.state!=Empty); |   assert(AccCache.state!=Empty); | ||||||
|    |    | ||||||
|   dprintf("MemoryManager: Evict(%llx) %llx\n",(uint64_t)AccCache.CpuPtr,(uint64_t)AccCache.AccPtr);  |   //  dprintf("MemoryManager: Evict(%llx) %llx\n",(uint64_t)AccCache.CpuPtr,(uint64_t)AccCache.AccPtr);  | ||||||
|   assert(AccCache.accLock==0); |   assert(AccCache.accLock==0); | ||||||
|   assert(AccCache.cpuLock==0); |   assert(AccCache.cpuLock==0); | ||||||
|   if(AccCache.state==AccDirty) { |   if(AccCache.state==AccDirty) { | ||||||
| @@ -137,7 +136,7 @@ void MemoryManager::Evict(AcceleratorViewEntry &AccCache) | |||||||
|     AcceleratorFree((void *)AccCache.AccPtr,AccCache.bytes); |     AcceleratorFree((void *)AccCache.AccPtr,AccCache.bytes); | ||||||
|     DeviceBytes   -=AccCache.bytes; |     DeviceBytes   -=AccCache.bytes; | ||||||
|     LRUremove(AccCache); |     LRUremove(AccCache); | ||||||
|     dprintf("MemoryManager: Free(%llx) footprint now %lld \n",(uint64_t)AccCache.AccPtr,DeviceBytes);   |     //    dprintf("MemoryManager: Free(%llx) footprint now %lld \n",(uint64_t)AccCache.AccPtr,DeviceBytes);   | ||||||
|   } |   } | ||||||
|   uint64_t CpuPtr = AccCache.CpuPtr; |   uint64_t CpuPtr = AccCache.CpuPtr; | ||||||
|   EntryErase(CpuPtr); |   EntryErase(CpuPtr); | ||||||
| @@ -150,7 +149,7 @@ void MemoryManager::Flush(AcceleratorViewEntry &AccCache) | |||||||
|   assert(AccCache.AccPtr!=(uint64_t)NULL); |   assert(AccCache.AccPtr!=(uint64_t)NULL); | ||||||
|   assert(AccCache.CpuPtr!=(uint64_t)NULL); |   assert(AccCache.CpuPtr!=(uint64_t)NULL); | ||||||
|   acceleratorCopyFromDevice((void *)AccCache.AccPtr,(void *)AccCache.CpuPtr,AccCache.bytes); |   acceleratorCopyFromDevice((void *)AccCache.AccPtr,(void *)AccCache.CpuPtr,AccCache.bytes); | ||||||
|   dprintf("MemoryManager: Flush  %llx -> %llx\n",(uint64_t)AccCache.AccPtr,(uint64_t)AccCache.CpuPtr); fflush(stdout); |   //  dprintf("MemoryManager: Flush  %llx -> %llx\n",(uint64_t)AccCache.AccPtr,(uint64_t)AccCache.CpuPtr); fflush(stdout); | ||||||
|   DeviceToHostBytes+=AccCache.bytes; |   DeviceToHostBytes+=AccCache.bytes; | ||||||
|   DeviceToHostXfer++; |   DeviceToHostXfer++; | ||||||
|   AccCache.state=Consistent; |   AccCache.state=Consistent; | ||||||
| @@ -165,7 +164,7 @@ void MemoryManager::Clone(AcceleratorViewEntry &AccCache) | |||||||
|     AccCache.AccPtr=(uint64_t)AcceleratorAllocate(AccCache.bytes); |     AccCache.AccPtr=(uint64_t)AcceleratorAllocate(AccCache.bytes); | ||||||
|     DeviceBytes+=AccCache.bytes; |     DeviceBytes+=AccCache.bytes; | ||||||
|   } |   } | ||||||
|   dprintf("MemoryManager: Clone %llx <- %llx\n",(uint64_t)AccCache.AccPtr,(uint64_t)AccCache.CpuPtr); fflush(stdout); |   //  dprintf("MemoryManager: Clone %llx <- %llx\n",(uint64_t)AccCache.AccPtr,(uint64_t)AccCache.CpuPtr); fflush(stdout); | ||||||
|   acceleratorCopyToDevice((void *)AccCache.CpuPtr,(void *)AccCache.AccPtr,AccCache.bytes); |   acceleratorCopyToDevice((void *)AccCache.CpuPtr,(void *)AccCache.AccPtr,AccCache.bytes); | ||||||
|   HostToDeviceBytes+=AccCache.bytes; |   HostToDeviceBytes+=AccCache.bytes; | ||||||
|   HostToDeviceXfer++; |   HostToDeviceXfer++; | ||||||
| @@ -228,24 +227,18 @@ uint64_t MemoryManager::AcceleratorViewOpen(uint64_t CpuPtr,size_t bytes,ViewMod | |||||||
|   // Find if present, otherwise get or force an empty |   // Find if present, otherwise get or force an empty | ||||||
|   //////////////////////////////////////////////////////////////////////////// |   //////////////////////////////////////////////////////////////////////////// | ||||||
|   if ( EntryPresent(CpuPtr)==0 ){ |   if ( EntryPresent(CpuPtr)==0 ){ | ||||||
|  |     EvictVictims(bytes); | ||||||
|     EntryCreate(CpuPtr,bytes,mode,hint); |     EntryCreate(CpuPtr,bytes,mode,hint); | ||||||
|   } |   } | ||||||
|  |  | ||||||
|   auto AccCacheIterator = EntryLookup(CpuPtr); |   auto AccCacheIterator = EntryLookup(CpuPtr); | ||||||
|   auto & AccCache = AccCacheIterator->second; |   auto & AccCache = AccCacheIterator->second; | ||||||
|   if (!AccCache.AccPtr) { |    | ||||||
|     EvictVictims(bytes);  |  | ||||||
|   }  |  | ||||||
|   assert((mode==AcceleratorRead)||(mode==AcceleratorWrite)||(mode==AcceleratorWriteDiscard)); |   assert((mode==AcceleratorRead)||(mode==AcceleratorWrite)||(mode==AcceleratorWriteDiscard)); | ||||||
|  |  | ||||||
|   assert(AccCache.cpuLock==0);  // Programming error |   assert(AccCache.cpuLock==0);  // Programming error | ||||||
|  |  | ||||||
|   if(AccCache.state!=Empty) { |   if(AccCache.state!=Empty) { | ||||||
|     dprintf("ViewOpen found entry %llx %llx : %lld %lld\n", |  | ||||||
| 		    (uint64_t)AccCache.CpuPtr, |  | ||||||
| 		    (uint64_t)CpuPtr, |  | ||||||
| 		    (uint64_t)AccCache.bytes, |  | ||||||
| 		    (uint64_t)bytes); |  | ||||||
|     assert(AccCache.CpuPtr == CpuPtr); |     assert(AccCache.CpuPtr == CpuPtr); | ||||||
|     assert(AccCache.bytes  ==bytes); |     assert(AccCache.bytes  ==bytes); | ||||||
|   } |   } | ||||||
| @@ -292,21 +285,21 @@ uint64_t MemoryManager::AcceleratorViewOpen(uint64_t CpuPtr,size_t bytes,ViewMod | |||||||
|       AccCache.state  = Consistent; // CpuDirty + AccRead => Consistent |       AccCache.state  = Consistent; // CpuDirty + AccRead => Consistent | ||||||
|     } |     } | ||||||
|     AccCache.accLock++; |     AccCache.accLock++; | ||||||
|     dprintf("Copied CpuDirty entry into device accLock %d\n",AccCache.accLock); |     //    printf("Copied CpuDirty entry into device accLock %d\n",AccCache.accLock); | ||||||
|   } else if(AccCache.state==Consistent) { |   } else if(AccCache.state==Consistent) { | ||||||
|     if((mode==AcceleratorWrite)||(mode==AcceleratorWriteDiscard)) |     if((mode==AcceleratorWrite)||(mode==AcceleratorWriteDiscard)) | ||||||
|       AccCache.state  = AccDirty;   // Consistent + AcceleratorWrite=> AccDirty |       AccCache.state  = AccDirty;   // Consistent + AcceleratorWrite=> AccDirty | ||||||
|     else |     else | ||||||
|       AccCache.state  = Consistent; // Consistent + AccRead => Consistent |       AccCache.state  = Consistent; // Consistent + AccRead => Consistent | ||||||
|     AccCache.accLock++; |     AccCache.accLock++; | ||||||
|     dprintf("Consistent entry into device accLock %d\n",AccCache.accLock); |     //    printf("Consistent entry into device accLock %d\n",AccCache.accLock); | ||||||
|   } else if(AccCache.state==AccDirty) { |   } else if(AccCache.state==AccDirty) { | ||||||
|     if((mode==AcceleratorWrite)||(mode==AcceleratorWriteDiscard)) |     if((mode==AcceleratorWrite)||(mode==AcceleratorWriteDiscard)) | ||||||
|       AccCache.state  = AccDirty; // AccDirty + AcceleratorWrite=> AccDirty |       AccCache.state  = AccDirty; // AccDirty + AcceleratorWrite=> AccDirty | ||||||
|     else |     else | ||||||
|       AccCache.state  = AccDirty; // AccDirty + AccRead => AccDirty |       AccCache.state  = AccDirty; // AccDirty + AccRead => AccDirty | ||||||
|     AccCache.accLock++; |     AccCache.accLock++; | ||||||
|     dprintf("AccDirty entry into device accLock %d\n",AccCache.accLock); |     //    printf("AccDirty entry into device accLock %d\n",AccCache.accLock); | ||||||
|   } else { |   } else { | ||||||
|     assert(0); |     assert(0); | ||||||
|   } |   } | ||||||
| @@ -368,16 +361,13 @@ uint64_t MemoryManager::CpuViewOpen(uint64_t CpuPtr,size_t bytes,ViewMode mode,V | |||||||
|   // Find if present, otherwise get or force an empty |   // Find if present, otherwise get or force an empty | ||||||
|   //////////////////////////////////////////////////////////////////////////// |   //////////////////////////////////////////////////////////////////////////// | ||||||
|   if ( EntryPresent(CpuPtr)==0 ){ |   if ( EntryPresent(CpuPtr)==0 ){ | ||||||
|  |     EvictVictims(bytes); | ||||||
|     EntryCreate(CpuPtr,bytes,mode,transient); |     EntryCreate(CpuPtr,bytes,mode,transient); | ||||||
|   } |   } | ||||||
|  |  | ||||||
|   auto AccCacheIterator = EntryLookup(CpuPtr); |   auto AccCacheIterator = EntryLookup(CpuPtr); | ||||||
|   auto & AccCache = AccCacheIterator->second; |   auto & AccCache = AccCacheIterator->second; | ||||||
|    |    | ||||||
|   if (!AccCache.AccPtr) { |  | ||||||
|      EvictVictims(bytes); |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   assert((mode==CpuRead)||(mode==CpuWrite)); |   assert((mode==CpuRead)||(mode==CpuWrite)); | ||||||
|   assert(AccCache.accLock==0);  // Programming error |   assert(AccCache.accLock==0);  // Programming error | ||||||
|  |  | ||||||
|   | |||||||
| @@ -1,6 +1,7 @@ | |||||||
| #include <Grid/GridCore.h> | #include <Grid/GridCore.h> | ||||||
| #ifdef GRID_UVM | #ifdef GRID_UVM | ||||||
|  |  | ||||||
|  | #warning "Grid is assuming unified virtual memory address space" | ||||||
| NAMESPACE_BEGIN(Grid); | NAMESPACE_BEGIN(Grid); | ||||||
| ///////////////////////////////////////////////////////////////////////////////// | ///////////////////////////////////////////////////////////////////////////////// | ||||||
| // View management is 1:1 address space mapping | // View management is 1:1 address space mapping | ||||||
|   | |||||||
| @@ -36,7 +36,7 @@ static const int CbBlack=1; | |||||||
| static const int Even   =CbRed; | static const int Even   =CbRed; | ||||||
| static const int Odd    =CbBlack; | static const int Odd    =CbBlack; | ||||||
|  |  | ||||||
| accelerator_inline int RedBlackCheckerBoardFromOindex (int oindex,const Coordinate &rdim,const Coordinate &chk_dim_msk) | accelerator_inline int RedBlackCheckerBoardFromOindex (int oindex, Coordinate &rdim, Coordinate &chk_dim_msk) | ||||||
| { | { | ||||||
|   int nd=rdim.size(); |   int nd=rdim.size(); | ||||||
|   Coordinate coor(nd); |   Coordinate coor(nd); | ||||||
|   | |||||||
| @@ -33,8 +33,6 @@ Author: Peter Boyle <paboyle@ph.ed.ac.uk> | |||||||
|  |  | ||||||
| NAMESPACE_BEGIN(Grid); | NAMESPACE_BEGIN(Grid); | ||||||
|  |  | ||||||
| bool Stencil_force_mpi = true; |  | ||||||
|  |  | ||||||
| /////////////////////////////////////////////////////////////// | /////////////////////////////////////////////////////////////// | ||||||
| // Info that is setup once and indept of cartesian layout | // Info that is setup once and indept of cartesian layout | ||||||
| /////////////////////////////////////////////////////////////// | /////////////////////////////////////////////////////////////// | ||||||
|   | |||||||
| @@ -1,3 +1,4 @@ | |||||||
|  |  | ||||||
| /************************************************************************************* | /************************************************************************************* | ||||||
|  |  | ||||||
|     Grid physics library, www.github.com/paboyle/Grid  |     Grid physics library, www.github.com/paboyle/Grid  | ||||||
| @@ -35,8 +36,6 @@ Author: Peter Boyle <paboyle@ph.ed.ac.uk> | |||||||
|  |  | ||||||
| NAMESPACE_BEGIN(Grid); | NAMESPACE_BEGIN(Grid); | ||||||
|  |  | ||||||
| extern bool Stencil_force_mpi ; |  | ||||||
|  |  | ||||||
| class CartesianCommunicator : public SharedMemory { | class CartesianCommunicator : public SharedMemory { | ||||||
|  |  | ||||||
| public:     | public:     | ||||||
| @@ -109,8 +108,6 @@ public: | |||||||
|   //////////////////////////////////////////////////////////// |   //////////////////////////////////////////////////////////// | ||||||
|   // Reduction |   // Reduction | ||||||
|   //////////////////////////////////////////////////////////// |   //////////////////////////////////////////////////////////// | ||||||
|   void GlobalMax(RealD &); |  | ||||||
|   void GlobalMax(RealF &); |  | ||||||
|   void GlobalSum(RealF &); |   void GlobalSum(RealF &); | ||||||
|   void GlobalSumVector(RealF *,int N); |   void GlobalSumVector(RealF *,int N); | ||||||
|   void GlobalSum(RealD &); |   void GlobalSum(RealD &); | ||||||
|   | |||||||
| @@ -275,16 +275,6 @@ void CartesianCommunicator::GlobalXOR(uint64_t &u){ | |||||||
|   int ierr=MPI_Allreduce(MPI_IN_PLACE,&u,1,MPI_UINT64_T,MPI_BXOR,communicator); |   int ierr=MPI_Allreduce(MPI_IN_PLACE,&u,1,MPI_UINT64_T,MPI_BXOR,communicator); | ||||||
|   assert(ierr==0); |   assert(ierr==0); | ||||||
| } | } | ||||||
| void CartesianCommunicator::GlobalMax(float &f) |  | ||||||
| { |  | ||||||
|   int ierr=MPI_Allreduce(MPI_IN_PLACE,&f,1,MPI_FLOAT,MPI_MAX,communicator); |  | ||||||
|   assert(ierr==0); |  | ||||||
| } |  | ||||||
| void CartesianCommunicator::GlobalMax(double &d) |  | ||||||
| { |  | ||||||
|   int ierr = MPI_Allreduce(MPI_IN_PLACE,&d,1,MPI_DOUBLE,MPI_MAX,communicator); |  | ||||||
|   assert(ierr==0); |  | ||||||
| } |  | ||||||
| void CartesianCommunicator::GlobalSum(float &f){ | void CartesianCommunicator::GlobalSum(float &f){ | ||||||
|   int ierr=MPI_Allreduce(MPI_IN_PLACE,&f,1,MPI_FLOAT,MPI_SUM,communicator); |   int ierr=MPI_Allreduce(MPI_IN_PLACE,&f,1,MPI_FLOAT,MPI_SUM,communicator); | ||||||
|   assert(ierr==0); |   assert(ierr==0); | ||||||
| @@ -370,7 +360,7 @@ double CartesianCommunicator::StencilSendToRecvFromBegin(std::vector<CommsReques | |||||||
|   double off_node_bytes=0.0; |   double off_node_bytes=0.0; | ||||||
|   int tag; |   int tag; | ||||||
|  |  | ||||||
|   if ( (gfrom ==MPI_UNDEFINED) || Stencil_force_mpi ) { |   if ( gfrom ==MPI_UNDEFINED) { | ||||||
|     tag= dir+from*32; |     tag= dir+from*32; | ||||||
|     ierr=MPI_Irecv(recv, bytes, MPI_CHAR,from,tag,communicator_halo[commdir],&rrq); |     ierr=MPI_Irecv(recv, bytes, MPI_CHAR,from,tag,communicator_halo[commdir],&rrq); | ||||||
|     assert(ierr==0); |     assert(ierr==0); | ||||||
| @@ -378,18 +368,12 @@ double CartesianCommunicator::StencilSendToRecvFromBegin(std::vector<CommsReques | |||||||
|     off_node_bytes+=bytes; |     off_node_bytes+=bytes; | ||||||
|   } |   } | ||||||
|  |  | ||||||
|   if ( (gdest == MPI_UNDEFINED) || Stencil_force_mpi ) { |   if ( gdest == MPI_UNDEFINED ) { | ||||||
|     tag= dir+_processor*32; |     tag= dir+_processor*32; | ||||||
|     ierr =MPI_Isend(xmit, bytes, MPI_CHAR,dest,tag,communicator_halo[commdir],&xrq); |     ierr =MPI_Isend(xmit, bytes, MPI_CHAR,dest,tag,communicator_halo[commdir],&xrq); | ||||||
|     assert(ierr==0); |     assert(ierr==0); | ||||||
|     list.push_back(xrq); |     list.push_back(xrq); | ||||||
|     off_node_bytes+=bytes; |     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); |  | ||||||
|     acceleratorCopyDeviceToDeviceAsynch(xmit,shm,bytes); |  | ||||||
|     acceleratorCopySynchronise(); // MPI prob slower |  | ||||||
|   } |   } | ||||||
|  |  | ||||||
|   if ( CommunicatorPolicy == CommunicatorPolicySequential ) { |   if ( CommunicatorPolicy == CommunicatorPolicySequential ) { | ||||||
|   | |||||||
| @@ -67,8 +67,6 @@ CartesianCommunicator::CartesianCommunicator(const Coordinate &processors) | |||||||
|  |  | ||||||
| CartesianCommunicator::~CartesianCommunicator(){} | CartesianCommunicator::~CartesianCommunicator(){} | ||||||
|  |  | ||||||
| void CartesianCommunicator::GlobalMax(float &){} |  | ||||||
| void CartesianCommunicator::GlobalMax(double &){} |  | ||||||
| void CartesianCommunicator::GlobalSum(float &){} | void CartesianCommunicator::GlobalSum(float &){} | ||||||
| void CartesianCommunicator::GlobalSumVector(float *,int N){} | void CartesianCommunicator::GlobalSumVector(float *,int N){} | ||||||
| void CartesianCommunicator::GlobalSum(double &){} | void CartesianCommunicator::GlobalSum(double &){} | ||||||
|   | |||||||
| @@ -102,7 +102,7 @@ public: | |||||||
|   /////////////////////////////////////////////////// |   /////////////////////////////////////////////////// | ||||||
|   static void SharedMemoryAllocate(uint64_t bytes, int flags); |   static void SharedMemoryAllocate(uint64_t bytes, int flags); | ||||||
|   static void SharedMemoryFree(void); |   static void SharedMemoryFree(void); | ||||||
|   static void SharedMemoryCopy(void *dest,void *src,size_t bytes); |   static void SharedMemoryCopy(void *dest,const void *src,size_t bytes); | ||||||
|   static void SharedMemoryZero(void *dest,size_t bytes); |   static void SharedMemoryZero(void *dest,size_t bytes); | ||||||
|  |  | ||||||
| }; | }; | ||||||
|   | |||||||
| @@ -7,7 +7,6 @@ | |||||||
|     Copyright (C) 2015 |     Copyright (C) 2015 | ||||||
|  |  | ||||||
| Author: Peter Boyle <paboyle@ph.ed.ac.uk> | Author: Peter Boyle <paboyle@ph.ed.ac.uk> | ||||||
| Author: Christoph Lehner <christoph@lhnr.de> |  | ||||||
|  |  | ||||||
|     This program is free software; you can redistribute it and/or modify |     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 |     it under the terms of the GNU General Public License as published by | ||||||
| @@ -35,9 +34,6 @@ Author: Christoph Lehner <christoph@lhnr.de> | |||||||
| #endif | #endif | ||||||
| #ifdef GRID_HIP | #ifdef GRID_HIP | ||||||
| #include <hip/hip_runtime_api.h> | #include <hip/hip_runtime_api.h> | ||||||
| #endif |  | ||||||
| #ifdef GRID_SYCl |  | ||||||
|  |  | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
| NAMESPACE_BEGIN(Grid);  | NAMESPACE_BEGIN(Grid);  | ||||||
| @@ -73,7 +69,6 @@ void GlobalSharedMemory::Init(Grid_MPI_Comm comm) | |||||||
|   WorldNodes = WorldSize/WorldShmSize; |   WorldNodes = WorldSize/WorldShmSize; | ||||||
|   assert( (WorldNodes * WorldShmSize) == WorldSize ); |   assert( (WorldNodes * WorldShmSize) == WorldSize ); | ||||||
|  |  | ||||||
|  |  | ||||||
|   // FIXME: Check all WorldShmSize are the same ? |   // FIXME: Check all WorldShmSize are the same ? | ||||||
|  |  | ||||||
|   ///////////////////////////////////////////////////////////////////// |   ///////////////////////////////////////////////////////////////////// | ||||||
| @@ -174,23 +169,6 @@ static inline int divides(int a,int b) | |||||||
| } | } | ||||||
| void GlobalSharedMemory::GetShmDims(const Coordinate &WorldDims,Coordinate &ShmDims) | void GlobalSharedMemory::GetShmDims(const Coordinate &WorldDims,Coordinate &ShmDims) | ||||||
| { | { | ||||||
|   //////////////////////////////////////////////////////////////// |  | ||||||
|   // Allow user to configure through environment variable |  | ||||||
|   //////////////////////////////////////////////////////////////// |  | ||||||
|   char* str = getenv(("GRID_SHM_DIMS_" + std::to_string(ShmDims.size())).c_str()); |  | ||||||
|   if ( str ) { |  | ||||||
|     std::vector<int> IntShmDims; |  | ||||||
|     GridCmdOptionIntVector(std::string(str),IntShmDims); |  | ||||||
|     assert(IntShmDims.size() == WorldDims.size()); |  | ||||||
|     long ShmSize = 1; |  | ||||||
|     for (int dim=0;dim<WorldDims.size();dim++) { |  | ||||||
|       ShmSize *= (ShmDims[dim] = IntShmDims[dim]); |  | ||||||
|       assert(divides(ShmDims[dim],WorldDims[dim])); |  | ||||||
|     } |  | ||||||
|     assert(ShmSize == WorldShmSize); |  | ||||||
|     return; |  | ||||||
|   } |  | ||||||
|    |  | ||||||
|   //////////////////////////////////////////////////////////////// |   //////////////////////////////////////////////////////////////// | ||||||
|   // Powers of 2,3,5 only in prime decomposition for now |   // Powers of 2,3,5 only in prime decomposition for now | ||||||
|   //////////////////////////////////////////////////////////////// |   //////////////////////////////////////////////////////////////// | ||||||
| @@ -450,47 +428,7 @@ void GlobalSharedMemory::SharedMemoryAllocate(uint64_t bytes, int flags) | |||||||
| //////////////////////////////////////////////////////////////////////////////////////////// | //////////////////////////////////////////////////////////////////////////////////////////// | ||||||
| // Hugetlbfs mapping intended | // Hugetlbfs mapping intended | ||||||
| //////////////////////////////////////////////////////////////////////////////////////////// | //////////////////////////////////////////////////////////////////////////////////////////// | ||||||
| #if defined(GRID_CUDA) ||defined(GRID_HIP)  || defined(GRID_SYCL) | #if defined(GRID_CUDA) ||defined(GRID_HIP) | ||||||
|  |  | ||||||
| //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)   |  | ||||||
| void GlobalSharedMemory::SharedMemoryAllocate(uint64_t bytes, int flags) | void GlobalSharedMemory::SharedMemoryAllocate(uint64_t bytes, int flags) | ||||||
| { | { | ||||||
|   void * ShmCommBuf ;  |   void * ShmCommBuf ;  | ||||||
| @@ -514,16 +452,18 @@ void GlobalSharedMemory::SharedMemoryAllocate(uint64_t bytes, int flags) | |||||||
|   // Each MPI rank should allocate our own buffer |   // Each MPI rank should allocate our own buffer | ||||||
|   /////////////////////////////////////////////////////////////////////////////////////////////////////////// |   /////////////////////////////////////////////////////////////////////////////////////////////////////////// | ||||||
|   ShmCommBuf = acceleratorAllocDevice(bytes); |   ShmCommBuf = acceleratorAllocDevice(bytes); | ||||||
|  |  | ||||||
|   if (ShmCommBuf == (void *)NULL ) { |   if (ShmCommBuf == (void *)NULL ) { | ||||||
|     std::cerr << " SharedMemoryMPI.cc acceleratorAllocDevice failed NULL pointer for " << bytes<<" bytes " << std::endl; |     std::cerr << " SharedMemoryMPI.cc acceleratorAllocDevice failed NULL pointer for " << bytes<<" bytes " << std::endl; | ||||||
|     exit(EXIT_FAILURE);   |     exit(EXIT_FAILURE);   | ||||||
|   } |   } | ||||||
|   if ( WorldRank == 0 ){ |   //  if ( WorldRank == 0 ){ | ||||||
|  |   if ( 1 ){ | ||||||
|     std::cout << WorldRank << header " SharedMemoryMPI.cc acceleratorAllocDevice "<< bytes  |     std::cout << WorldRank << header " SharedMemoryMPI.cc acceleratorAllocDevice "<< bytes  | ||||||
| 	      << "bytes at "<< std::hex<< ShmCommBuf <<std::dec<<" for comms buffers " <<std::endl; | 	      << "bytes at "<< std::hex<< ShmCommBuf <<std::dec<<" for comms buffers " <<std::endl; | ||||||
|   } |   } | ||||||
|   SharedMemoryZero(ShmCommBuf,bytes); |   SharedMemoryZero(ShmCommBuf,bytes); | ||||||
|   std::cout<< "Setting up IPC"<<std::endl; |  | ||||||
|   /////////////////////////////////////////////////////////////////////////////////////////////////////////// |   /////////////////////////////////////////////////////////////////////////////////////////////////////////// | ||||||
|   // Loop over ranks/gpu's on our node |   // Loop over ranks/gpu's on our node | ||||||
|   /////////////////////////////////////////////////////////////////////////////////////////////////////////// |   /////////////////////////////////////////////////////////////////////////////////////////////////////////// | ||||||
| @@ -533,29 +473,6 @@ void GlobalSharedMemory::SharedMemoryAllocate(uint64_t bytes, int flags) | |||||||
|     ////////////////////////////////////////////////// |     ////////////////////////////////////////////////// | ||||||
|     // If it is me, pass around the IPC access key |     // 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 | #ifdef GRID_CUDA | ||||||
|     cudaIpcMemHandle_t handle; |     cudaIpcMemHandle_t handle; | ||||||
|     if ( r==WorldShmRank ) {  |     if ( r==WorldShmRank ) {  | ||||||
| @@ -576,7 +493,6 @@ void GlobalSharedMemory::SharedMemoryAllocate(uint64_t bytes, int flags) | |||||||
|       } |       } | ||||||
|     } |     } | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
|     ////////////////////////////////////////////////// |     ////////////////////////////////////////////////// | ||||||
|     // Share this IPC handle across the Shm Comm |     // Share this IPC handle across the Shm Comm | ||||||
|     ////////////////////////////////////////////////// |     ////////////////////////////////////////////////// | ||||||
| @@ -592,35 +508,7 @@ void GlobalSharedMemory::SharedMemoryAllocate(uint64_t bytes, int flags) | |||||||
|     /////////////////////////////////////////////////////////////// |     /////////////////////////////////////////////////////////////// | ||||||
|     // If I am not the source, overwrite thisBuf with remote buffer |     // If I am not the source, overwrite thisBuf with remote buffer | ||||||
|     /////////////////////////////////////////////////////////////// |     /////////////////////////////////////////////////////////////// | ||||||
|  |     void * thisBuf = ShmCommBuf; | ||||||
| #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 |  | ||||||
| #ifdef GRID_CUDA | #ifdef GRID_CUDA | ||||||
|     if ( r!=WorldShmRank ) {  |     if ( r!=WorldShmRank ) {  | ||||||
|       auto err = cudaIpcOpenMemHandle(&thisBuf,handle,cudaIpcMemLazyEnablePeerAccess); |       auto err = cudaIpcOpenMemHandle(&thisBuf,handle,cudaIpcMemLazyEnablePeerAccess); | ||||||
| @@ -642,7 +530,6 @@ void GlobalSharedMemory::SharedMemoryAllocate(uint64_t bytes, int flags) | |||||||
|     /////////////////////////////////////////////////////////////// |     /////////////////////////////////////////////////////////////// | ||||||
|     // Save a copy of the device buffers |     // Save a copy of the device buffers | ||||||
|     /////////////////////////////////////////////////////////////// |     /////////////////////////////////////////////////////////////// | ||||||
|     } |  | ||||||
|     WorldShmCommBufs[r] = thisBuf; |     WorldShmCommBufs[r] = thisBuf; | ||||||
| #else | #else | ||||||
|     WorldShmCommBufs[r] = ShmCommBuf; |     WorldShmCommBufs[r] = ShmCommBuf; | ||||||
| @@ -652,8 +539,6 @@ void GlobalSharedMemory::SharedMemoryAllocate(uint64_t bytes, int flags) | |||||||
|   _ShmAllocBytes=bytes; |   _ShmAllocBytes=bytes; | ||||||
|   _ShmAlloc=1; |   _ShmAlloc=1; | ||||||
| } | } | ||||||
| #endif |  | ||||||
|  |  | ||||||
| #else  | #else  | ||||||
| #ifdef GRID_MPI3_SHMMMAP | #ifdef GRID_MPI3_SHMMMAP | ||||||
| void GlobalSharedMemory::SharedMemoryAllocate(uint64_t bytes, int flags) | void GlobalSharedMemory::SharedMemoryAllocate(uint64_t bytes, int flags) | ||||||
| @@ -824,16 +709,16 @@ void GlobalSharedMemory::SharedMemoryAllocate(uint64_t bytes, int flags) | |||||||
| ///////////////////////////////////////////////////////////////////////// | ///////////////////////////////////////////////////////////////////////// | ||||||
| void GlobalSharedMemory::SharedMemoryZero(void *dest,size_t bytes) | void GlobalSharedMemory::SharedMemoryZero(void *dest,size_t bytes) | ||||||
| { | { | ||||||
| #if defined(GRID_CUDA) || defined(GRID_HIP) || defined(GRID_SYCL) | #ifdef GRID_CUDA | ||||||
|   acceleratorMemSet(dest,0,bytes); |   cudaMemset(dest,0,bytes); | ||||||
| #else | #else | ||||||
|   bzero(dest,bytes); |   bzero(dest,bytes); | ||||||
| #endif | #endif | ||||||
| } | } | ||||||
| void GlobalSharedMemory::SharedMemoryCopy(void *dest,void *src,size_t bytes) | void GlobalSharedMemory::SharedMemoryCopy(void *dest,const void *src,size_t bytes) | ||||||
| { | { | ||||||
| #if defined(GRID_CUDA) || defined(GRID_HIP) || defined(GRID_SYCL) | #ifdef GRID_CUDA | ||||||
|   acceleratorCopyToDevice(src,dest,bytes); |   cudaMemcpy(dest,src,bytes,cudaMemcpyDefault); | ||||||
| #else    | #else    | ||||||
|   bcopy(src,dest,bytes); |   bcopy(src,dest,bytes); | ||||||
| #endif | #endif | ||||||
| @@ -897,7 +782,7 @@ void SharedMemory::SetCommunicator(Grid_MPI_Comm comm) | |||||||
|   } |   } | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
|   //SharedMemoryTest(); |   SharedMemoryTest(); | ||||||
| } | } | ||||||
| ////////////////////////////////////////////////////////////////// | ////////////////////////////////////////////////////////////////// | ||||||
| // On node barrier | // On node barrier | ||||||
|   | |||||||
| @@ -29,7 +29,6 @@ Author: Peter Boyle <paboyle@ph.ed.ac.uk> | |||||||
| #include <Grid/GridCore.h> | #include <Grid/GridCore.h> | ||||||
|  |  | ||||||
| NAMESPACE_BEGIN(Grid);  | NAMESPACE_BEGIN(Grid);  | ||||||
| #define header "SharedMemoryNone: " |  | ||||||
|  |  | ||||||
| /*Construct from an MPI communicator*/ | /*Construct from an MPI communicator*/ | ||||||
| void GlobalSharedMemory::Init(Grid_MPI_Comm comm) | void GlobalSharedMemory::Init(Grid_MPI_Comm comm) | ||||||
| @@ -56,38 +55,6 @@ void GlobalSharedMemory::OptimalCommunicator(const Coordinate &processors,Grid_M | |||||||
| //////////////////////////////////////////////////////////////////////////////////////////// | //////////////////////////////////////////////////////////////////////////////////////////// | ||||||
| // Hugetlbfs mapping intended, use anonymous mmap | // Hugetlbfs mapping intended, use anonymous mmap | ||||||
| //////////////////////////////////////////////////////////////////////////////////////////// | //////////////////////////////////////////////////////////////////////////////////////////// | ||||||
| #if 1 |  | ||||||
| void GlobalSharedMemory::SharedMemoryAllocate(uint64_t bytes, int flags) |  | ||||||
| { |  | ||||||
|   std::cout << header "SharedMemoryAllocate "<< bytes<< " GPU implementation "<<std::endl; |  | ||||||
|   void * ShmCommBuf ;  |  | ||||||
|   assert(_ShmSetup==1); |  | ||||||
|   assert(_ShmAlloc==0); |  | ||||||
|  |  | ||||||
|   /////////////////////////////////////////////////////////////////////////////////////////////////////////// |  | ||||||
|   // Each MPI rank should allocate our own buffer |  | ||||||
|   /////////////////////////////////////////////////////////////////////////////////////////////////////////// |  | ||||||
|   ShmCommBuf = acceleratorAllocDevice(bytes); |  | ||||||
|  |  | ||||||
|   if (ShmCommBuf == (void *)NULL ) { |  | ||||||
|     std::cerr << " SharedMemoryNone.cc acceleratorAllocDevice failed NULL pointer for " << bytes<<" bytes " << std::endl; |  | ||||||
|     exit(EXIT_FAILURE);   |  | ||||||
|   } |  | ||||||
|   if ( WorldRank == 0 ){ |  | ||||||
|     std::cout << WorldRank << header " SharedMemoryNone.cc acceleratorAllocDevice "<< bytes  |  | ||||||
| 	      << "bytes at "<< std::hex<< ShmCommBuf <<std::dec<<" for comms buffers " <<std::endl; |  | ||||||
|   } |  | ||||||
|   SharedMemoryZero(ShmCommBuf,bytes); |  | ||||||
|  |  | ||||||
|   /////////////////////////////////////////////////////////////////////////////////////////////////////////// |  | ||||||
|   // Loop over ranks/gpu's on our node |  | ||||||
|   /////////////////////////////////////////////////////////////////////////////////////////////////////////// |  | ||||||
|   WorldShmCommBufs[0] = ShmCommBuf; |  | ||||||
|  |  | ||||||
|   _ShmAllocBytes=bytes; |  | ||||||
|   _ShmAlloc=1; |  | ||||||
| } |  | ||||||
| #else |  | ||||||
| void GlobalSharedMemory::SharedMemoryAllocate(uint64_t bytes, int flags) | void GlobalSharedMemory::SharedMemoryAllocate(uint64_t bytes, int flags) | ||||||
| { | { | ||||||
|   void * ShmCommBuf ;  |   void * ShmCommBuf ;  | ||||||
| @@ -116,15 +83,7 @@ void GlobalSharedMemory::SharedMemoryAllocate(uint64_t bytes, int flags) | |||||||
|   _ShmAllocBytes=bytes; |   _ShmAllocBytes=bytes; | ||||||
|   _ShmAlloc=1; |   _ShmAlloc=1; | ||||||
| }; | }; | ||||||
| #endif |  | ||||||
| void GlobalSharedMemory::SharedMemoryZero(void *dest,size_t bytes) |  | ||||||
| { |  | ||||||
|   acceleratorMemSet(dest,0,bytes); |  | ||||||
| } |  | ||||||
| void GlobalSharedMemory::SharedMemoryCopy(void *dest,void *src,size_t bytes) |  | ||||||
| { |  | ||||||
|   acceleratorCopyToDevice(src,dest,bytes); |  | ||||||
| } |  | ||||||
| //////////////////////////////////////////////////////// | //////////////////////////////////////////////////////// | ||||||
| // Global shared functionality finished | // Global shared functionality finished | ||||||
| // Now move to per communicator functionality | // Now move to per communicator functionality | ||||||
|   | |||||||
| @@ -112,9 +112,7 @@ Gather_plane_extract(const Lattice<vobj> &rhs, | |||||||
|   if ( cbmask ==0x3){ |   if ( cbmask ==0x3){ | ||||||
| #ifdef ACCELERATOR_CSHIFT     | #ifdef ACCELERATOR_CSHIFT     | ||||||
|     autoView(rhs_v , rhs, AcceleratorRead); |     autoView(rhs_v , rhs, AcceleratorRead); | ||||||
|     accelerator_for(nn,e1*e2,1,{ |     accelerator_for2d(n,e1,b,e2,1,{ | ||||||
| 	int n = nn%e1; |  | ||||||
| 	int b = nn/e1; |  | ||||||
| 	int o      =   n*n1; | 	int o      =   n*n1; | ||||||
| 	int offset = b+n*e2; | 	int offset = b+n*e2; | ||||||
| 	 | 	 | ||||||
| @@ -137,9 +135,7 @@ Gather_plane_extract(const Lattice<vobj> &rhs, | |||||||
|     std::cout << " Dense packed buffer WARNING " <<std::endl; // Does this get called twice once for each cb? |     std::cout << " Dense packed buffer WARNING " <<std::endl; // Does this get called twice once for each cb? | ||||||
| #ifdef ACCELERATOR_CSHIFT     | #ifdef ACCELERATOR_CSHIFT     | ||||||
|     autoView(rhs_v , rhs, AcceleratorRead); |     autoView(rhs_v , rhs, AcceleratorRead); | ||||||
|     accelerator_for(nn,e1*e2,1,{ |     accelerator_for2d(n,e1,b,e2,1,{ | ||||||
| 	int n = nn%e1; |  | ||||||
| 	int b = nn/e1; |  | ||||||
|  |  | ||||||
| 	Coordinate coor; | 	Coordinate coor; | ||||||
|  |  | ||||||
| @@ -261,9 +257,7 @@ template<class vobj> void Scatter_plane_merge(Lattice<vobj> &rhs,ExtractPointerA | |||||||
|     int _slice_block = rhs.Grid()->_slice_block[dimension]; |     int _slice_block = rhs.Grid()->_slice_block[dimension]; | ||||||
| #ifdef ACCELERATOR_CSHIFT     | #ifdef ACCELERATOR_CSHIFT     | ||||||
|     autoView( rhs_v , rhs, AcceleratorWrite); |     autoView( rhs_v , rhs, AcceleratorWrite); | ||||||
|     accelerator_for(nn,e1*e2,1,{ |     accelerator_for2d(n,e1,b,e2,1,{ | ||||||
| 	int n = nn%e1; |  | ||||||
| 	int b = nn/e1; |  | ||||||
| 	int o      = n*_slice_stride; | 	int o      = n*_slice_stride; | ||||||
| 	int offset = b+n*_slice_block; | 	int offset = b+n*_slice_block; | ||||||
| 	merge(rhs_v[so+o+b],pointers,offset); | 	merge(rhs_v[so+o+b],pointers,offset); | ||||||
| @@ -280,7 +274,7 @@ template<class vobj> void Scatter_plane_merge(Lattice<vobj> &rhs,ExtractPointerA | |||||||
|  |  | ||||||
|     // Case of SIMD split AND checker dim cannot currently be hit, except in  |     // Case of SIMD split AND checker dim cannot currently be hit, except in  | ||||||
|     // Test_cshift_red_black code. |     // Test_cshift_red_black code. | ||||||
|     std::cout << "Scatter_plane merge assert(0); think this is buggy FIXME "<< std::endl;// think this is buggy FIXME |     //    std::cout << "Scatter_plane merge assert(0); think this is buggy FIXME "<< std::endl;// think this is buggy FIXME | ||||||
|     std::cout<<" Unthreaded warning -- buffer is not densely packed ??"<<std::endl; |     std::cout<<" Unthreaded warning -- buffer is not densely packed ??"<<std::endl; | ||||||
|     assert(0); // This will fail if hit on GPU |     assert(0); // This will fail if hit on GPU | ||||||
|     autoView( rhs_v, rhs, CpuWrite); |     autoView( rhs_v, rhs, CpuWrite); | ||||||
|   | |||||||
| @@ -122,8 +122,8 @@ template<class vobj> void Cshift_comms(Lattice<vobj> &ret,const Lattice<vobj> &r | |||||||
|   assert(shift<fd); |   assert(shift<fd); | ||||||
|    |    | ||||||
|   int buffer_size = rhs.Grid()->_slice_nblock[dimension]*rhs.Grid()->_slice_block[dimension]; |   int buffer_size = rhs.Grid()->_slice_nblock[dimension]*rhs.Grid()->_slice_block[dimension]; | ||||||
|   static cshiftVector<vobj> send_buf; send_buf.resize(buffer_size); |   cshiftVector<vobj> send_buf(buffer_size); | ||||||
|   static cshiftVector<vobj> recv_buf; recv_buf.resize(buffer_size); |   cshiftVector<vobj> recv_buf(buffer_size); | ||||||
|      |      | ||||||
|   int cb= (cbmask==0x2)? Odd : Even; |   int cb= (cbmask==0x2)? Odd : Even; | ||||||
|   int sshift= rhs.Grid()->CheckerBoardShiftForCB(rhs.Checkerboard(),dimension,shift,cb); |   int sshift= rhs.Grid()->CheckerBoardShiftForCB(rhs.Checkerboard(),dimension,shift,cb); | ||||||
| @@ -198,8 +198,8 @@ template<class vobj> void  Cshift_comms_simd(Lattice<vobj> &ret,const Lattice<vo | |||||||
|   int buffer_size = grid->_slice_nblock[dimension]*grid->_slice_block[dimension]; |   int buffer_size = grid->_slice_nblock[dimension]*grid->_slice_block[dimension]; | ||||||
|   //  int words = sizeof(vobj)/sizeof(vector_type); |   //  int words = sizeof(vobj)/sizeof(vector_type); | ||||||
|  |  | ||||||
|   static std::vector<cshiftVector<scalar_object> >  send_buf_extract; send_buf_extract.resize(Nsimd); |   std::vector<cshiftVector<scalar_object> >  send_buf_extract(Nsimd); | ||||||
|   static std::vector<cshiftVector<scalar_object> >  recv_buf_extract; recv_buf_extract.resize(Nsimd); |   std::vector<cshiftVector<scalar_object> >  recv_buf_extract(Nsimd); | ||||||
|   scalar_object *  recv_buf_extract_mpi; |   scalar_object *  recv_buf_extract_mpi; | ||||||
|   scalar_object *  send_buf_extract_mpi; |   scalar_object *  send_buf_extract_mpi; | ||||||
|   |   | ||||||
| @@ -294,8 +294,8 @@ template<class vobj> void Cshift_comms(Lattice<vobj> &ret,const Lattice<vobj> &r | |||||||
|   assert(shift<fd); |   assert(shift<fd); | ||||||
|    |    | ||||||
|   int buffer_size = rhs.Grid()->_slice_nblock[dimension]*rhs.Grid()->_slice_block[dimension]; |   int buffer_size = rhs.Grid()->_slice_nblock[dimension]*rhs.Grid()->_slice_block[dimension]; | ||||||
|   static cshiftVector<vobj> send_buf_v; send_buf_v.resize(buffer_size); |   cshiftVector<vobj> send_buf_v(buffer_size); | ||||||
|   static cshiftVector<vobj> recv_buf_v; recv_buf_v.resize(buffer_size); |   cshiftVector<vobj> recv_buf_v(buffer_size); | ||||||
|   vobj *send_buf; |   vobj *send_buf; | ||||||
|   vobj *recv_buf; |   vobj *recv_buf; | ||||||
|   { |   { | ||||||
| @@ -381,8 +381,8 @@ template<class vobj> void  Cshift_comms_simd(Lattice<vobj> &ret,const Lattice<vo | |||||||
|   int buffer_size = grid->_slice_nblock[dimension]*grid->_slice_block[dimension]; |   int buffer_size = grid->_slice_nblock[dimension]*grid->_slice_block[dimension]; | ||||||
|   //  int words = sizeof(vobj)/sizeof(vector_type); |   //  int words = sizeof(vobj)/sizeof(vector_type); | ||||||
|  |  | ||||||
|   static std::vector<cshiftVector<scalar_object> >  send_buf_extract; send_buf_extract.resize(Nsimd); |   std::vector<cshiftVector<scalar_object> >  send_buf_extract(Nsimd); | ||||||
|   static std::vector<cshiftVector<scalar_object> >  recv_buf_extract; recv_buf_extract.resize(Nsimd); |   std::vector<cshiftVector<scalar_object> >  recv_buf_extract(Nsimd); | ||||||
|   scalar_object *  recv_buf_extract_mpi; |   scalar_object *  recv_buf_extract_mpi; | ||||||
|   scalar_object *  send_buf_extract_mpi; |   scalar_object *  send_buf_extract_mpi; | ||||||
|   { |   { | ||||||
|   | |||||||
| @@ -225,7 +225,7 @@ void axpy(Lattice<vobj> &ret,sobj a,const Lattice<vobj> &x,const Lattice<vobj> & | |||||||
|   autoView( x_v , x, AcceleratorRead); |   autoView( x_v , x, AcceleratorRead); | ||||||
|   autoView( y_v , y, AcceleratorRead); |   autoView( y_v , y, AcceleratorRead); | ||||||
|   accelerator_for(ss,x_v.size(),vobj::Nsimd(),{ |   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); |     coalescedWrite(ret_v[ss],tmp); | ||||||
|   }); |   }); | ||||||
| } | } | ||||||
|   | |||||||
| @@ -62,7 +62,7 @@ void basisRotate(VField &basis,Matrix& Qt,int j0, int j1, int k0,int k1,int Nm) | |||||||
|     basis_v.push_back(basis[k].View(AcceleratorWrite)); |     basis_v.push_back(basis[k].View(AcceleratorWrite)); | ||||||
|   } |   } | ||||||
|  |  | ||||||
| #if ( (!defined(GRID_CUDA)) ) | #if ( (!defined(GRID_SYCL)) && (!defined(GRID_CUDA)) ) | ||||||
|   int max_threads = thread_max(); |   int max_threads = thread_max(); | ||||||
|   Vector < vobj > Bt(Nm * max_threads); |   Vector < vobj > Bt(Nm * max_threads); | ||||||
|   thread_region |   thread_region | ||||||
| @@ -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){ | 	for(int k=k0; k<k1; ++k){ | ||||||
| 	  auto tmp = coalescedRead(Bp[ss*nrot+j]); | 	  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 jj  =j0+j; | ||||||
| 	int ss =sj/nrot; | 	int ss =sj/nrot; | ||||||
| 	int sss=ss+s; | 	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 | #endif | ||||||
| @@ -164,8 +164,7 @@ void basisRotateJ(Field &result,std::vector<Field> &basis,Eigen::MatrixXd& Qt,in | |||||||
|   auto basis_vp=& basis_v[0]; |   auto basis_vp=& basis_v[0]; | ||||||
|   autoView(result_v,result,AcceleratorWrite); |   autoView(result_v,result,AcceleratorWrite); | ||||||
|   accelerator_for(ss, grid->oSites(),vobj::Nsimd(),{ |   accelerator_for(ss, grid->oSites(),vobj::Nsimd(),{ | ||||||
|     vobj zzz=Zero(); |     auto B=coalescedRead(zz); | ||||||
|     auto B=coalescedRead(zzz); |  | ||||||
|     for(int k=k0; k<k1; ++k){ |     for(int k=k0; k<k1; ++k){ | ||||||
|       B +=Qt_j[k] * coalescedRead(basis_vp[k][ss]); |       B +=Qt_j[k] * coalescedRead(basis_vp[k][ss]); | ||||||
|     } |     } | ||||||
|   | |||||||
| @@ -96,34 +96,8 @@ inline typename vobj::scalar_objectD sumD_cpu(const vobj *arg, Integer osites) | |||||||
|   ssobj ret = ssum; |   ssobj ret = ssum; | ||||||
|   return ret; |   return ret; | ||||||
| } | } | ||||||
| /* |  | ||||||
| Threaded max, don't use for now |  | ||||||
| template<class Double> |  | ||||||
| inline Double max(const Double *arg, Integer osites) |  | ||||||
| { |  | ||||||
|   //  const int Nsimd = vobj::Nsimd(); |  | ||||||
|   const int nthread = GridThread::GetThreads(); |  | ||||||
|  |  | ||||||
|   std::vector<Double> maxarray(nthread); |  | ||||||
|  |  | ||||||
|   thread_for(thr,nthread, { |  | ||||||
|     int nwork, mywork, myoff; |  | ||||||
|     nwork = osites; |  | ||||||
|     GridThread::GetWork(nwork,thr,mywork,myoff); |  | ||||||
|     Double max=arg[0]; |  | ||||||
|     for(int ss=myoff;ss<mywork+myoff; ss++){ |  | ||||||
|       if( arg[ss] > max ) max = arg[ss]; |  | ||||||
|     } |  | ||||||
|     maxarray[thr]=max; |  | ||||||
|   }); |  | ||||||
|    |  | ||||||
|   Double tmax=maxarray[0]; |  | ||||||
|   for(int i=0;i<nthread;i++){ |  | ||||||
|     if (maxarray[i]>tmax) tmax = maxarray[i]; |  | ||||||
|   }  |  | ||||||
|   return tmax; |  | ||||||
| } |  | ||||||
| */ |  | ||||||
| template<class vobj> | template<class vobj> | ||||||
| inline typename vobj::scalar_object sum(const vobj *arg, Integer osites) | inline typename vobj::scalar_object sum(const vobj *arg, Integer osites) | ||||||
| { | { | ||||||
| @@ -167,32 +141,6 @@ template<class vobj> inline RealD norm2(const Lattice<vobj> &arg){ | |||||||
|   return real(nrm);  |   return real(nrm);  | ||||||
| } | } | ||||||
|  |  | ||||||
| //The global maximum of the site norm2 |  | ||||||
| template<class vobj> inline RealD maxLocalNorm2(const Lattice<vobj> &arg) |  | ||||||
| { |  | ||||||
|   typedef typename vobj::tensor_reduced vscalar;  //iScalar<iScalar<.... <vPODtype> > > |  | ||||||
|   typedef typename vscalar::scalar_object  scalar;   //iScalar<iScalar<.... <PODtype> > > |  | ||||||
|  |  | ||||||
|   Lattice<vscalar> inner = localNorm2(arg); |  | ||||||
|  |  | ||||||
|   auto grid = arg.Grid(); |  | ||||||
|  |  | ||||||
|   RealD max; |  | ||||||
|   for(int l=0;l<grid->lSites();l++){ |  | ||||||
|     Coordinate coor; |  | ||||||
|     scalar val; |  | ||||||
|     RealD r; |  | ||||||
|     grid->LocalIndexToLocalCoor(l,coor); |  | ||||||
|     peekLocalSite(val,inner,coor); |  | ||||||
|     r=real(TensorRemove(val)); |  | ||||||
|     if( (l==0) || (r>max)){ |  | ||||||
|       max=r; |  | ||||||
|     } |  | ||||||
|   } |  | ||||||
|   grid->GlobalMax(max); |  | ||||||
|   return max; |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Double inner product | // Double inner product | ||||||
| template<class vobj> | template<class vobj> | ||||||
| inline ComplexD rankInnerProduct(const Lattice<vobj> &left,const Lattice<vobj> &right) | inline ComplexD rankInnerProduct(const Lattice<vobj> &left,const Lattice<vobj> &right) | ||||||
| @@ -361,7 +309,6 @@ template<class vobj> inline void sliceSum(const Lattice<vobj> &Data,std::vector< | |||||||
|   // But easily avoided by using double precision fields |   // But easily avoided by using double precision fields | ||||||
|   /////////////////////////////////////////////////////// |   /////////////////////////////////////////////////////// | ||||||
|   typedef typename vobj::scalar_object sobj; |   typedef typename vobj::scalar_object sobj; | ||||||
|   typedef typename vobj::scalar_object::scalar_type scalar_type; |  | ||||||
|   GridBase  *grid = Data.Grid(); |   GridBase  *grid = Data.Grid(); | ||||||
|   assert(grid!=NULL); |   assert(grid!=NULL); | ||||||
|  |  | ||||||
| @@ -420,19 +367,20 @@ template<class vobj> inline void sliceSum(const Lattice<vobj> &Data,std::vector< | |||||||
|   } |   } | ||||||
|    |    | ||||||
|   // sum over nodes. |   // sum over nodes. | ||||||
|  |   sobj gsum; | ||||||
|   for(int t=0;t<fd;t++){ |   for(int t=0;t<fd;t++){ | ||||||
|     int pt = t/ld; // processor plane |     int pt = t/ld; // processor plane | ||||||
|     int lt = t%ld; |     int lt = t%ld; | ||||||
|     if ( pt == grid->_processor_coor[orthogdim] ) { |     if ( pt == grid->_processor_coor[orthogdim] ) { | ||||||
|       result[t]=lsSum[lt]; |       gsum=lsSum[lt]; | ||||||
|     } else { |     } 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> | template<class vobj> | ||||||
|   | |||||||
| @@ -97,20 +97,6 @@ accelerator_inline void convertType(ComplexF & out, const std::complex<float> & | |||||||
|   out = in; |   out = in; | ||||||
| } | } | ||||||
|  |  | ||||||
| template<typename T> |  | ||||||
| accelerator_inline EnableIf<isGridFundamental<T>> convertType(T & out, const T & in) { |  | ||||||
|   out = in; |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // This would allow for conversions between GridFundamental types, but is not strictly needed as yet |  | ||||||
| /*template<typename T1, typename T2> |  | ||||||
| accelerator_inline typename std::enable_if<isGridFundamental<T1>::value && isGridFundamental<T2>::value>::type |  | ||||||
| // Or to make this very broad, conversions between anything that's not a GridTensor could be allowed |  | ||||||
| //accelerator_inline typename std::enable_if<!isGridTensor<T1>::value && !isGridTensor<T2>::value>::type |  | ||||||
| convertType(T1 & out, const T2 & in) { |  | ||||||
|   out = in; |  | ||||||
| }*/ |  | ||||||
|  |  | ||||||
| #ifdef GRID_SIMT | #ifdef GRID_SIMT | ||||||
| accelerator_inline void convertType(vComplexF & out, const ComplexF & in) { | accelerator_inline void convertType(vComplexF & out, const ComplexF & in) { | ||||||
|   ((ComplexF*)&out)[acceleratorSIMTlane(vComplexF::Nsimd())] = in; |   ((ComplexF*)&out)[acceleratorSIMTlane(vComplexF::Nsimd())] = in; | ||||||
| @@ -131,18 +117,18 @@ accelerator_inline void convertType(vComplexD2 & out, const vComplexF & in) { | |||||||
|   Optimization::PrecisionChange::StoD(in.v,out._internal[0].v,out._internal[1].v); |   Optimization::PrecisionChange::StoD(in.v,out._internal[0].v,out._internal[1].v); | ||||||
| } | } | ||||||
|  |  | ||||||
| template<typename T1,typename T2> | template<typename T1,typename T2,int N> | ||||||
| accelerator_inline void convertType(iScalar<T1> & out, const iScalar<T2> & in) { |   accelerator_inline void convertType(iMatrix<T1,N> & out, const iMatrix<T2,N> & in); | ||||||
|   convertType(out._internal,in._internal); | template<typename T1,typename T2,int N> | ||||||
| } |   accelerator_inline void convertType(iVector<T1,N> & out, const iVector<T2,N> & in); | ||||||
|  |  | ||||||
| template<typename T1,typename T2> | template<typename T1,typename T2, typename std::enable_if<!isGridScalar<T1>::value, T1>::type* = nullptr> | ||||||
| accelerator_inline NotEnableIf<isGridScalar<T1>> convertType(T1 & out, const iScalar<T2> & in) { | accelerator_inline void convertType(T1 & out, const iScalar<T2> & in) { | ||||||
|   convertType(out,in._internal); |   convertType(out,in._internal); | ||||||
| } | } | ||||||
|  |  | ||||||
| template<typename T1,typename T2> | template<typename T1,typename T2> | ||||||
| accelerator_inline NotEnableIf<isGridScalar<T2>> convertType(iScalar<T1> & out, const T2 & in) { | accelerator_inline void convertType(iScalar<T1> & out, const T2 & in) { | ||||||
|   convertType(out._internal,in); |   convertType(out._internal,in); | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -159,6 +145,11 @@ accelerator_inline void convertType(iVector<T1,N> & out, const iVector<T2,N> & i | |||||||
|     convertType(out._internal[i],in._internal[i]); |     convertType(out._internal[i],in._internal[i]); | ||||||
| } | } | ||||||
|  |  | ||||||
|  | template<typename T, typename std::enable_if<isGridFundamental<T>::value, T>::type* = nullptr> | ||||||
|  | accelerator_inline void convertType(T & out, const T & in) { | ||||||
|  |   out = in; | ||||||
|  | } | ||||||
|  |  | ||||||
| template<typename T1,typename T2> | template<typename T1,typename T2> | ||||||
| accelerator_inline void convertType(Lattice<T1> & out, const Lattice<T2> & in) { | accelerator_inline void convertType(Lattice<T1> & out, const Lattice<T2> & in) { | ||||||
|   autoView( out_v , out,AcceleratorWrite); |   autoView( out_v , out,AcceleratorWrite); | ||||||
| @@ -364,21 +355,15 @@ inline void blockSum(Lattice<vobj> &coarseData,const Lattice<vobj> &fineData) | |||||||
|   autoView( coarseData_ , coarseData, AcceleratorWrite); |   autoView( coarseData_ , coarseData, AcceleratorWrite); | ||||||
|   autoView( fineData_   , fineData, AcceleratorRead); |   autoView( fineData_   , fineData, AcceleratorRead); | ||||||
|  |  | ||||||
|   auto coarseData_p = &coarseData_[0]; |  | ||||||
|   auto fineData_p = &fineData_[0]; |  | ||||||
|    |  | ||||||
|   Coordinate fine_rdimensions = fine->_rdimensions; |   Coordinate fine_rdimensions = fine->_rdimensions; | ||||||
|   Coordinate coarse_rdimensions = coarse->_rdimensions; |   Coordinate coarse_rdimensions = coarse->_rdimensions; | ||||||
|    |    | ||||||
|   vobj zz = Zero(); |  | ||||||
|    |  | ||||||
|   accelerator_for(sc,coarse->oSites(),1,{ |   accelerator_for(sc,coarse->oSites(),1,{ | ||||||
|  |  | ||||||
|       // One thread per sub block |       // One thread per sub block | ||||||
|       Coordinate coor_c(_ndimension); |       Coordinate coor_c(_ndimension); | ||||||
|       Lexicographic::CoorFromIndex(coor_c,sc,coarse_rdimensions);  // Block coordinate |       Lexicographic::CoorFromIndex(coor_c,sc,coarse_rdimensions);  // Block coordinate | ||||||
|  |       coarseData_[sc]=Zero(); | ||||||
|       vobj cd = zz; |  | ||||||
|  |  | ||||||
|       for(int sb=0;sb<blockVol;sb++){ |       for(int sb=0;sb<blockVol;sb++){ | ||||||
|  |  | ||||||
| @@ -389,11 +374,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]; | 	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); | 	Lexicographic::IndexFromCoor(coor_f,sf,fine_rdimensions); | ||||||
|  |  | ||||||
| 	cd=cd+fineData_p[sf]; | 	coarseData_[sc]=coarseData_[sc]+fineData_[sf]; | ||||||
|       } |       } | ||||||
|  |  | ||||||
|       coarseData_p[sc] = cd; |  | ||||||
|  |  | ||||||
|     }); |     }); | ||||||
|   return; |   return; | ||||||
| } | } | ||||||
|   | |||||||
| @@ -67,13 +67,8 @@ public: | |||||||
|   accelerator_inline const vobj & operator()(size_t i) const { return this->_odata[i]; } |   accelerator_inline const vobj & operator()(size_t i) const { return this->_odata[i]; } | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
| #if 1 |  | ||||||
|   //  accelerator_inline const vobj & operator[](size_t i) const { return this->_odata[i]; }; |  | ||||||
|   accelerator_inline vobj       & operator[](size_t i) const { return this->_odata[i]; }; |  | ||||||
| #else |  | ||||||
|   accelerator_inline const vobj & operator[](size_t i) const { return this->_odata[i]; }; |   accelerator_inline const vobj & operator[](size_t i) const { return this->_odata[i]; }; | ||||||
|   accelerator_inline vobj       & operator[](size_t i)       { return this->_odata[i]; }; |   accelerator_inline vobj       & operator[](size_t i)       { return this->_odata[i]; }; | ||||||
| #endif |  | ||||||
|  |  | ||||||
|   accelerator_inline uint64_t begin(void) const { return 0;}; |   accelerator_inline uint64_t begin(void) const { return 0;}; | ||||||
|   accelerator_inline uint64_t end(void)   const { return this->_odata_size; }; |   accelerator_inline uint64_t end(void)   const { return this->_odata_size; }; | ||||||
|   | |||||||
| @@ -43,7 +43,7 @@ inline void whereWolf(Lattice<vobj> &ret,const Lattice<iobj> &predicate,Lattice< | |||||||
|   conformable(iftrue,predicate); |   conformable(iftrue,predicate); | ||||||
|   conformable(iftrue,ret); |   conformable(iftrue,ret); | ||||||
|  |  | ||||||
|   GridBase *grid=iftrue.Grid(); |   GridBase *grid=iftrue._grid; | ||||||
|  |  | ||||||
|   typedef typename vobj::scalar_object scalar_object; |   typedef typename vobj::scalar_object scalar_object; | ||||||
|   typedef typename vobj::scalar_type scalar_type; |   typedef typename vobj::scalar_type scalar_type; | ||||||
| @@ -52,23 +52,22 @@ inline void whereWolf(Lattice<vobj> &ret,const Lattice<iobj> &predicate,Lattice< | |||||||
|  |  | ||||||
|   const int Nsimd = grid->Nsimd(); |   const int Nsimd = grid->Nsimd(); | ||||||
|  |  | ||||||
|   autoView(iftrue_v,iftrue,CpuRead); |   std::vector<Integer> mask(Nsimd); | ||||||
|   autoView(iffalse_v,iffalse,CpuRead); |   std::vector<scalar_object> truevals (Nsimd); | ||||||
|   autoView(predicate_v,predicate,CpuRead); |   std::vector<scalar_object> falsevals(Nsimd); | ||||||
|   autoView(ret_v,ret,CpuWrite); |  | ||||||
|   Integer NN= grid->oSites(); |   parallel_for(int ss=0;ss<iftrue._grid->oSites(); ss++){ | ||||||
|   thread_for(ss,NN,{ |  | ||||||
|     Integer mask; |     extract(iftrue._odata[ss]   ,truevals); | ||||||
|     scalar_object trueval; |     extract(iffalse._odata[ss]  ,falsevals); | ||||||
|     scalar_object falseval; |     extract<vInteger,Integer>(TensorRemove(predicate._odata[ss]),mask); | ||||||
|     for(int l=0;l<Nsimd;l++){ |  | ||||||
|       trueval =extractLane(l,iftrue_v[ss]); |     for(int s=0;s<Nsimd;s++){ | ||||||
|       falseval=extractLane(l,iffalse_v[ss]); |       if (mask[s]) falsevals[s]=truevals[s]; | ||||||
|       mask    =extractLane(l,predicate_v[ss]); |  | ||||||
|       if (mask) falseval=trueval; |  | ||||||
|       insertLane(l,ret_v[ss],falseval); |  | ||||||
|     } |     } | ||||||
|   }); |  | ||||||
|  |     merge(ret._odata[ss],falsevals); | ||||||
|  |   } | ||||||
| } | } | ||||||
|  |  | ||||||
| template<class vobj,class iobj> | template<class vobj,class iobj> | ||||||
| @@ -77,9 +76,9 @@ inline Lattice<vobj> whereWolf(const Lattice<iobj> &predicate,Lattice<vobj> &ift | |||||||
|   conformable(iftrue,iffalse); |   conformable(iftrue,iffalse); | ||||||
|   conformable(iftrue,predicate); |   conformable(iftrue,predicate); | ||||||
|  |  | ||||||
|   Lattice<vobj> ret(iftrue.Grid()); |   Lattice<vobj> ret(iftrue._grid); | ||||||
|  |  | ||||||
|   whereWolf(ret,predicate,iftrue,iffalse); |   where(ret,predicate,iftrue,iffalse); | ||||||
|  |  | ||||||
|   return ret; |   return ret; | ||||||
| } | } | ||||||
|   | |||||||
| @@ -123,7 +123,7 @@ assert(GRID_FIELD_NORM_CALC(FieldNormMetaData_, n2ck) < 1.0e-5); | |||||||
|  //////////////////////////////////////////////////////////// |  //////////////////////////////////////////////////////////// | ||||||
|  // Helper to fill out metadata |  // Helper to fill out metadata | ||||||
|  //////////////////////////////////////////////////////////// |  //////////////////////////////////////////////////////////// | ||||||
| template<class vobj> void ScidacMetaData(Lattice<vobj> & field, |  template<class vobj> void ScidacMetaData(Lattice<vobj> & field, | ||||||
| 					  FieldMetaData &header, | 					  FieldMetaData &header, | ||||||
| 					  scidacRecord & _scidacRecord, | 					  scidacRecord & _scidacRecord, | ||||||
| 					  scidacFile   & _scidacFile)  | 					  scidacFile   & _scidacFile)  | ||||||
| @@ -619,12 +619,12 @@ class IldgWriter : public ScidacWriter { | |||||||
|   // Don't require scidac records EXCEPT checksum |   // Don't require scidac records EXCEPT checksum | ||||||
|   // Use Grid MetaData object if present. |   // Use Grid MetaData object if present. | ||||||
|   //////////////////////////////////////////////////////////////// |   //////////////////////////////////////////////////////////////// | ||||||
|   template <class stats = PeriodicGaugeStatistics> |   template <class vsimd> | ||||||
|   void writeConfiguration(Lattice<vLorentzColourMatrixD > &Umu,int sequence,std::string LFN,std::string description)  |   void writeConfiguration(Lattice<iLorentzColourMatrix<vsimd> > &Umu,int sequence,std::string LFN,std::string description)  | ||||||
|   { |   { | ||||||
|     GridBase * grid = Umu.Grid(); |     GridBase * grid = Umu.Grid(); | ||||||
|     typedef Lattice<vLorentzColourMatrixD> GaugeField; |     typedef Lattice<iLorentzColourMatrix<vsimd> > GaugeField; | ||||||
|     typedef vLorentzColourMatrixD vobj; |     typedef iLorentzColourMatrix<vsimd> vobj; | ||||||
|     typedef typename vobj::scalar_object sobj; |     typedef typename vobj::scalar_object sobj; | ||||||
|  |  | ||||||
|     //////////////////////////////////////// |     //////////////////////////////////////// | ||||||
| @@ -636,9 +636,6 @@ class IldgWriter : public ScidacWriter { | |||||||
|  |  | ||||||
|     ScidacMetaData(Umu,header,_scidacRecord,_scidacFile); |     ScidacMetaData(Umu,header,_scidacRecord,_scidacFile); | ||||||
|  |  | ||||||
|     stats Stats; |  | ||||||
|     Stats(Umu,header); |  | ||||||
|      |  | ||||||
|     std::string format = header.floating_point; |     std::string format = header.floating_point; | ||||||
|     header.ensemble_id    = description; |     header.ensemble_id    = description; | ||||||
|     header.ensemble_label = description; |     header.ensemble_label = description; | ||||||
| @@ -708,10 +705,10 @@ class IldgReader : public GridLimeReader { | |||||||
|   // Else use ILDG MetaData object if present. |   // Else use ILDG MetaData object if present. | ||||||
|   // Else use SciDAC MetaData object if present. |   // Else use SciDAC MetaData object if present. | ||||||
|   //////////////////////////////////////////////////////////////// |   //////////////////////////////////////////////////////////////// | ||||||
|   template <class stats = PeriodicGaugeStatistics> |   template <class vsimd> | ||||||
|   void readConfiguration(Lattice<vLorentzColourMatrixD> &Umu, FieldMetaData &FieldMetaData_) { |   void readConfiguration(Lattice<iLorentzColourMatrix<vsimd> > &Umu, FieldMetaData &FieldMetaData_) { | ||||||
|  |  | ||||||
|     typedef Lattice<vLorentzColourMatrixD > GaugeField; |     typedef Lattice<iLorentzColourMatrix<vsimd> > GaugeField; | ||||||
|     typedef typename GaugeField::vector_object  vobj; |     typedef typename GaugeField::vector_object  vobj; | ||||||
|     typedef typename vobj::scalar_object sobj; |     typedef typename vobj::scalar_object sobj; | ||||||
|  |  | ||||||
| @@ -924,8 +921,7 @@ class IldgReader : public GridLimeReader { | |||||||
|  |  | ||||||
|     if ( found_FieldMetaData || found_usqcdInfo ) { |     if ( found_FieldMetaData || found_usqcdInfo ) { | ||||||
|       FieldMetaData checker; |       FieldMetaData checker; | ||||||
|       stats Stats; |       GaugeStatistics(Umu,checker); | ||||||
|       Stats(Umu,checker); |  | ||||||
|       assert(fabs(checker.plaquette  - FieldMetaData_.plaquette )<1.0e-5); |       assert(fabs(checker.plaquette  - FieldMetaData_.plaquette )<1.0e-5); | ||||||
|       assert(fabs(checker.link_trace - FieldMetaData_.link_trace)<1.0e-5); |       assert(fabs(checker.link_trace - FieldMetaData_.link_trace)<1.0e-5); | ||||||
|       std::cout << GridLogMessage<<"Plaquette and link trace match " << std::endl; |       std::cout << GridLogMessage<<"Plaquette and link trace match " << std::endl; | ||||||
|   | |||||||
| @@ -128,7 +128,7 @@ inline void MachineCharacteristics(FieldMetaData &header) | |||||||
|   std::time_t t = std::time(nullptr); |   std::time_t t = std::time(nullptr); | ||||||
|   std::tm tm_ = *std::localtime(&t); |   std::tm tm_ = *std::localtime(&t); | ||||||
|   std::ostringstream oss;  |   std::ostringstream oss;  | ||||||
|   oss << std::put_time(&tm_, "%c %Z"); |   //      oss << std::put_time(&tm_, "%c %Z"); | ||||||
|   header.creation_date = oss.str(); |   header.creation_date = oss.str(); | ||||||
|   header.archive_date  = header.creation_date; |   header.archive_date  = header.creation_date; | ||||||
|  |  | ||||||
| @@ -176,18 +176,29 @@ template<class vobj> inline void PrepareMetaData(Lattice<vobj> & field, FieldMet | |||||||
|   GridMetaData(grid,header);  |   GridMetaData(grid,header);  | ||||||
|   MachineCharacteristics(header); |   MachineCharacteristics(header); | ||||||
| } | } | ||||||
| template<class Impl> | inline void GaugeStatistics(Lattice<vLorentzColourMatrixF> & data,FieldMetaData &header) | ||||||
| class GaugeStatistics |  | ||||||
| { | { | ||||||
| public: |   // How to convert data precision etc... | ||||||
|   void operator()(Lattice<vLorentzColourMatrixD> & data,FieldMetaData &header) |   header.link_trace=WilsonLoops<PeriodicGimplF>::linkTrace(data); | ||||||
|   { |   header.plaquette =WilsonLoops<PeriodicGimplF>::avgPlaquette(data); | ||||||
|     header.link_trace=WilsonLoops<Impl>::linkTrace(data); | } | ||||||
|     header.plaquette =WilsonLoops<Impl>::avgPlaquette(data); | inline void GaugeStatistics(Lattice<vLorentzColourMatrixD> & data,FieldMetaData &header) | ||||||
|   } | { | ||||||
| }; |   // How to convert data precision etc... | ||||||
| typedef GaugeStatistics<PeriodicGimplD> PeriodicGaugeStatistics; |   header.link_trace=WilsonLoops<PeriodicGimplD>::linkTrace(data); | ||||||
| typedef GaugeStatistics<ConjugateGimplD> ConjugateGaugeStatistics; |   header.plaquette =WilsonLoops<PeriodicGimplD>::avgPlaquette(data); | ||||||
|  | } | ||||||
|  | template<> inline void PrepareMetaData<vLorentzColourMatrixF>(Lattice<vLorentzColourMatrixF> & field, FieldMetaData &header) | ||||||
|  | { | ||||||
|  |     | ||||||
|  |   GridBase *grid = field.Grid(); | ||||||
|  |   std::string format = getFormatString<vLorentzColourMatrixF>(); | ||||||
|  |   header.floating_point = format; | ||||||
|  |   header.checksum = 0x0; // Nersc checksum unused in ILDG, Scidac | ||||||
|  |   GridMetaData(grid,header);  | ||||||
|  |   GaugeStatistics(field,header); | ||||||
|  |   MachineCharacteristics(header); | ||||||
|  | } | ||||||
| template<> inline void PrepareMetaData<vLorentzColourMatrixD>(Lattice<vLorentzColourMatrixD> & field, FieldMetaData &header) | template<> inline void PrepareMetaData<vLorentzColourMatrixD>(Lattice<vLorentzColourMatrixD> & field, FieldMetaData &header) | ||||||
| { | { | ||||||
|   GridBase *grid = field.Grid(); |   GridBase *grid = field.Grid(); | ||||||
| @@ -195,6 +206,7 @@ template<> inline void PrepareMetaData<vLorentzColourMatrixD>(Lattice<vLorentzCo | |||||||
|   header.floating_point = format; |   header.floating_point = format; | ||||||
|   header.checksum = 0x0; // Nersc checksum unused in ILDG, Scidac |   header.checksum = 0x0; // Nersc checksum unused in ILDG, Scidac | ||||||
|   GridMetaData(grid,header);  |   GridMetaData(grid,header);  | ||||||
|  |   GaugeStatistics(field,header); | ||||||
|   MachineCharacteristics(header); |   MachineCharacteristics(header); | ||||||
| } | } | ||||||
|  |  | ||||||
|   | |||||||
| @@ -40,8 +40,6 @@ using namespace Grid; | |||||||
| class NerscIO : public BinaryIO {  | class NerscIO : public BinaryIO {  | ||||||
| public: | public: | ||||||
|  |  | ||||||
|   typedef Lattice<vLorentzColourMatrixD> GaugeField; |  | ||||||
|  |  | ||||||
|   static inline void truncate(std::string file){ |   static inline void truncate(std::string file){ | ||||||
|     std::ofstream fout(file,std::ios::out); |     std::ofstream fout(file,std::ios::out); | ||||||
|   } |   } | ||||||
| @@ -131,12 +129,12 @@ public: | |||||||
|   // Now the meat: the object readers |   // Now the meat: the object readers | ||||||
|   ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////// |   ///////////////////////////////////////////////////////////////////////////////////////////////////////////////////// | ||||||
|  |  | ||||||
|   template<class GaugeStats=PeriodicGaugeStatistics> |   template<class vsimd> | ||||||
|   static inline void readConfiguration(GaugeField &Umu, |   static inline void readConfiguration(Lattice<iLorentzColourMatrix<vsimd> > &Umu, | ||||||
| 				       FieldMetaData& header, | 				       FieldMetaData& header, | ||||||
| 				       std::string file, | 				       std::string file) | ||||||
| 				       GaugeStats GaugeStatisticsCalculator=GaugeStats()) |  | ||||||
|   { |   { | ||||||
|  |     typedef Lattice<iLorentzColourMatrix<vsimd> > GaugeField; | ||||||
|  |  | ||||||
|     GridBase *grid = Umu.Grid(); |     GridBase *grid = Umu.Grid(); | ||||||
|     uint64_t offset = readHeader(file,Umu.Grid(),header); |     uint64_t offset = readHeader(file,Umu.Grid(),header); | ||||||
| @@ -155,23 +153,23 @@ public: | |||||||
|     // munger is a function of <floating point, Real, data_type> |     // munger is a function of <floating point, Real, data_type> | ||||||
|     if ( header.data_type == std::string("4D_SU3_GAUGE") ) { |     if ( header.data_type == std::string("4D_SU3_GAUGE") ) { | ||||||
|       if ( ieee32 || ieee32big ) { |       if ( ieee32 || ieee32big ) { | ||||||
| 	BinaryIO::readLatticeObject<vLorentzColourMatrixD, LorentzColour2x3F>  | 	BinaryIO::readLatticeObject<iLorentzColourMatrix<vsimd>, LorentzColour2x3F>  | ||||||
| 	  (Umu,file,Gauge3x2munger<LorentzColour2x3F,LorentzColourMatrix>(), offset,format, | 	  (Umu,file,Gauge3x2munger<LorentzColour2x3F,LorentzColourMatrix>(), offset,format, | ||||||
| 	   nersc_csum,scidac_csuma,scidac_csumb); | 	   nersc_csum,scidac_csuma,scidac_csumb); | ||||||
|       } |       } | ||||||
|       if ( ieee64 || ieee64big ) { |       if ( ieee64 || ieee64big ) { | ||||||
| 	BinaryIO::readLatticeObject<vLorentzColourMatrixD, LorentzColour2x3D>  | 	BinaryIO::readLatticeObject<iLorentzColourMatrix<vsimd>, LorentzColour2x3D>  | ||||||
| 	  (Umu,file,Gauge3x2munger<LorentzColour2x3D,LorentzColourMatrix>(),offset,format, | 	  (Umu,file,Gauge3x2munger<LorentzColour2x3D,LorentzColourMatrix>(),offset,format, | ||||||
| 	   nersc_csum,scidac_csuma,scidac_csumb); | 	   nersc_csum,scidac_csuma,scidac_csumb); | ||||||
|       } |       } | ||||||
|     } else if ( header.data_type == std::string("4D_SU3_GAUGE_3x3") ) { |     } else if ( header.data_type == std::string("4D_SU3_GAUGE_3x3") ) { | ||||||
|       if ( ieee32 || ieee32big ) { |       if ( ieee32 || ieee32big ) { | ||||||
| 	BinaryIO::readLatticeObject<vLorentzColourMatrixD,LorentzColourMatrixF> | 	BinaryIO::readLatticeObject<iLorentzColourMatrix<vsimd>,LorentzColourMatrixF> | ||||||
| 	  (Umu,file,GaugeSimpleMunger<LorentzColourMatrixF,LorentzColourMatrix>(),offset,format, | 	  (Umu,file,GaugeSimpleMunger<LorentzColourMatrixF,LorentzColourMatrix>(),offset,format, | ||||||
| 	   nersc_csum,scidac_csuma,scidac_csumb); | 	   nersc_csum,scidac_csuma,scidac_csumb); | ||||||
|       } |       } | ||||||
|       if ( ieee64 || ieee64big ) { |       if ( ieee64 || ieee64big ) { | ||||||
| 	BinaryIO::readLatticeObject<vLorentzColourMatrixD,LorentzColourMatrixD> | 	BinaryIO::readLatticeObject<iLorentzColourMatrix<vsimd>,LorentzColourMatrixD> | ||||||
| 	  (Umu,file,GaugeSimpleMunger<LorentzColourMatrixD,LorentzColourMatrix>(),offset,format, | 	  (Umu,file,GaugeSimpleMunger<LorentzColourMatrixD,LorentzColourMatrix>(),offset,format, | ||||||
| 	   nersc_csum,scidac_csuma,scidac_csumb); | 	   nersc_csum,scidac_csuma,scidac_csumb); | ||||||
|       } |       } | ||||||
| @@ -179,7 +177,7 @@ public: | |||||||
|       assert(0); |       assert(0); | ||||||
|     } |     } | ||||||
|  |  | ||||||
|     GaugeStats Stats; Stats(Umu,clone); |     GaugeStatistics(Umu,clone); | ||||||
|  |  | ||||||
|     std::cout<<GridLogMessage <<"NERSC Configuration "<<file<<" checksum "<<std::hex<<nersc_csum<< std::dec |     std::cout<<GridLogMessage <<"NERSC Configuration "<<file<<" checksum "<<std::hex<<nersc_csum<< std::dec | ||||||
| 	     <<" header   "<<std::hex<<header.checksum<<std::dec <<std::endl; | 	     <<" header   "<<std::hex<<header.checksum<<std::dec <<std::endl; | ||||||
| @@ -205,22 +203,15 @@ public: | |||||||
|     std::cout<<GridLogMessage <<"NERSC Configuration "<<file<< " and plaquette, link trace, and checksum agree"<<std::endl; |     std::cout<<GridLogMessage <<"NERSC Configuration "<<file<< " and plaquette, link trace, and checksum agree"<<std::endl; | ||||||
|   } |   } | ||||||
|  |  | ||||||
|   // Preferred interface |   template<class vsimd> | ||||||
|   template<class GaugeStats=PeriodicGaugeStatistics> |   static inline void writeConfiguration(Lattice<iLorentzColourMatrix<vsimd> > &Umu, | ||||||
|   static inline void writeConfiguration(Lattice<vLorentzColourMatrixD > &Umu, |  | ||||||
| 					std::string file,  |  | ||||||
| 					std::string ens_label = std::string("DWF")) |  | ||||||
|   { |  | ||||||
|     writeConfiguration(Umu,file,0,1,ens_label); |  | ||||||
|   } |  | ||||||
|   template<class GaugeStats=PeriodicGaugeStatistics> |  | ||||||
|   static inline void writeConfiguration(Lattice<vLorentzColourMatrixD > &Umu, |  | ||||||
| 					std::string file,  | 					std::string file,  | ||||||
| 					int two_row, | 					int two_row, | ||||||
| 					int bits32, | 					int bits32) | ||||||
| 					std::string ens_label = std::string("DWF")) |  | ||||||
|   { |   { | ||||||
|     typedef vLorentzColourMatrixD vobj; |     typedef Lattice<iLorentzColourMatrix<vsimd> > GaugeField; | ||||||
|  |  | ||||||
|  |     typedef iLorentzColourMatrix<vsimd> vobj; | ||||||
|     typedef typename vobj::scalar_object sobj; |     typedef typename vobj::scalar_object sobj; | ||||||
|  |  | ||||||
|     FieldMetaData header; |     FieldMetaData header; | ||||||
| @@ -228,8 +219,8 @@ public: | |||||||
|     // Following should become arguments |     // Following should become arguments | ||||||
|     /////////////////////////////////////////// |     /////////////////////////////////////////// | ||||||
|     header.sequence_number = 1; |     header.sequence_number = 1; | ||||||
|     header.ensemble_id     = std::string("UKQCD"); |     header.ensemble_id     = "UKQCD"; | ||||||
|     header.ensemble_label  = ens_label; |     header.ensemble_label  = "DWF"; | ||||||
|  |  | ||||||
|     typedef LorentzColourMatrixD fobj3D; |     typedef LorentzColourMatrixD fobj3D; | ||||||
|     typedef LorentzColour2x3D    fobj2D; |     typedef LorentzColour2x3D    fobj2D; | ||||||
| @@ -238,28 +229,28 @@ public: | |||||||
|  |  | ||||||
|     GridMetaData(grid,header); |     GridMetaData(grid,header); | ||||||
|     assert(header.nd==4); |     assert(header.nd==4); | ||||||
|     GaugeStats Stats; Stats(Umu,header); |     GaugeStatistics(Umu,header); | ||||||
|     MachineCharacteristics(header); |     MachineCharacteristics(header); | ||||||
|  |  | ||||||
|     uint64_t offset; | 	uint64_t offset; | ||||||
|  |  | ||||||
|     // Sod it -- always write 3x3 double |     // Sod it -- always write 3x3 double | ||||||
|     header.floating_point = std::string("IEEE64BIG"); |     header.floating_point = std::string("IEEE64BIG"); | ||||||
|     header.data_type      = std::string("4D_SU3_GAUGE_3x3"); |     header.data_type      = std::string("4D_SU3_GAUGE_3x3"); | ||||||
|     GaugeSimpleUnmunger<fobj3D,sobj> munge; |     GaugeSimpleUnmunger<fobj3D,sobj> munge; | ||||||
|     if ( grid->IsBoss() ) {  | 	if ( grid->IsBoss() ) {  | ||||||
|       truncate(file); | 	  truncate(file); | ||||||
|       offset = writeHeader(header,file); |     offset = writeHeader(header,file); | ||||||
|     } | 	} | ||||||
|     grid->Broadcast(0,(void *)&offset,sizeof(offset)); | 	grid->Broadcast(0,(void *)&offset,sizeof(offset)); | ||||||
|  |  | ||||||
|     uint32_t nersc_csum,scidac_csuma,scidac_csumb; |     uint32_t nersc_csum,scidac_csuma,scidac_csumb; | ||||||
|     BinaryIO::writeLatticeObject<vobj,fobj3D>(Umu,file,munge,offset,header.floating_point, |     BinaryIO::writeLatticeObject<vobj,fobj3D>(Umu,file,munge,offset,header.floating_point, | ||||||
| 					      nersc_csum,scidac_csuma,scidac_csumb); | 					      nersc_csum,scidac_csuma,scidac_csumb); | ||||||
|     header.checksum = nersc_csum; |     header.checksum = nersc_csum; | ||||||
|     if ( grid->IsBoss() ) {  | 	if ( grid->IsBoss() ) {  | ||||||
|       writeHeader(header,file); |     writeHeader(header,file); | ||||||
|     } | 	} | ||||||
|  |  | ||||||
|     std::cout<<GridLogMessage <<"Written NERSC Configuration on "<< file << " checksum " |     std::cout<<GridLogMessage <<"Written NERSC Configuration on "<< file << " checksum " | ||||||
| 	     <<std::hex<<header.checksum | 	     <<std::hex<<header.checksum | ||||||
|   | |||||||
| @@ -154,7 +154,7 @@ public: | |||||||
|     grid->Barrier(); timer.Stop(); |     grid->Barrier(); timer.Stop(); | ||||||
|     std::cout << Grid::GridLogMessage << "OpenQcdIO::readConfiguration: redistribute overhead " << timer.Elapsed() << std::endl; |     std::cout << Grid::GridLogMessage << "OpenQcdIO::readConfiguration: redistribute overhead " << timer.Elapsed() << std::endl; | ||||||
|  |  | ||||||
|     PeriodicGaugeStatistics Stats; Stats(Umu, clone); |     GaugeStatistics(Umu, clone); | ||||||
|  |  | ||||||
|     RealD plaq_diff = fabs(clone.plaquette - header.plaquette); |     RealD plaq_diff = fabs(clone.plaquette - header.plaquette); | ||||||
|  |  | ||||||
|   | |||||||
| @@ -208,7 +208,7 @@ public: | |||||||
|  |  | ||||||
|     FieldMetaData clone(header); |     FieldMetaData clone(header); | ||||||
|  |  | ||||||
|     PeriodicGaugeStatistics Stats; Stats(Umu, clone); |     GaugeStatistics(Umu, clone); | ||||||
|  |  | ||||||
|     RealD plaq_diff = fabs(clone.plaquette - header.plaquette); |     RealD plaq_diff = fabs(clone.plaquette - header.plaquette); | ||||||
|  |  | ||||||
|   | |||||||
| @@ -80,13 +80,6 @@ template<typename T> struct isSpinor { | |||||||
| template <typename T> using IfSpinor    = Invoke<std::enable_if< isSpinor<T>::value,int> > ; | template <typename T> using IfSpinor    = Invoke<std::enable_if< isSpinor<T>::value,int> > ; | ||||||
| template <typename T> using IfNotSpinor = Invoke<std::enable_if<!isSpinor<T>::value,int> > ; | template <typename T> using IfNotSpinor = Invoke<std::enable_if<!isSpinor<T>::value,int> > ; | ||||||
|  |  | ||||||
| const int CoarseIndex = 4; |  | ||||||
| template<typename T> struct isCoarsened { |  | ||||||
|    static constexpr bool value = (CoarseIndex<=T::TensorLevel); |  | ||||||
| }; |  | ||||||
| template <typename T> using IfCoarsened    = Invoke<std::enable_if< isCoarsened<T>::value,int> > ; |  | ||||||
| template <typename T> using IfNotCoarsened = Invoke<std::enable_if<!isCoarsened<T>::value,int> > ; |  | ||||||
|  |  | ||||||
| // ChrisK very keen to add extra space for Gparity doubling. | // ChrisK very keen to add extra space for Gparity doubling. | ||||||
| // | // | ||||||
| // Also add domain wall index, in a way where Wilson operator  | // Also add domain wall index, in a way where Wilson operator  | ||||||
|   | |||||||
| @@ -41,7 +41,7 @@ class Action | |||||||
| public: | public: | ||||||
|   bool is_smeared = false; |   bool is_smeared = false; | ||||||
|   // Heatbath? |   // Heatbath? | ||||||
|   virtual void refresh(const GaugeField& U, GridSerialRNG &sRNG, GridParallelRNG& pRNG) = 0; // refresh pseudofermions |   virtual void refresh(const GaugeField& U, GridParallelRNG& pRNG) = 0; // refresh pseudofermions | ||||||
|   virtual RealD S(const GaugeField& U) = 0;                             // evaluate the action |   virtual RealD S(const GaugeField& U) = 0;                             // evaluate the action | ||||||
|   virtual void deriv(const GaugeField& U, GaugeField& dSdU) = 0;        // evaluate the action derivative |   virtual void deriv(const GaugeField& U, GaugeField& dSdU) = 0;        // evaluate the action derivative | ||||||
|   virtual std::string action_name()    = 0;                             // return the action name |   virtual std::string action_name()    = 0;                             // return the action name | ||||||
|   | |||||||
| @@ -115,9 +115,9 @@ typedef WilsonFermion<WilsonImplR> WilsonFermionR; | |||||||
| typedef WilsonFermion<WilsonImplF> WilsonFermionF; | typedef WilsonFermion<WilsonImplF> WilsonFermionF; | ||||||
| typedef WilsonFermion<WilsonImplD> WilsonFermionD; | typedef WilsonFermion<WilsonImplD> WilsonFermionD; | ||||||
|  |  | ||||||
| //typedef WilsonFermion<WilsonImplRL> WilsonFermionRL; | typedef WilsonFermion<WilsonImplRL> WilsonFermionRL; | ||||||
| //typedef WilsonFermion<WilsonImplFH> WilsonFermionFH; | typedef WilsonFermion<WilsonImplFH> WilsonFermionFH; | ||||||
| //typedef WilsonFermion<WilsonImplDF> WilsonFermionDF; | typedef WilsonFermion<WilsonImplDF> WilsonFermionDF; | ||||||
|  |  | ||||||
| typedef WilsonFermion<WilsonAdjImplR> WilsonAdjFermionR; | typedef WilsonFermion<WilsonAdjImplR> WilsonAdjFermionR; | ||||||
| typedef WilsonFermion<WilsonAdjImplF> WilsonAdjFermionF; | typedef WilsonFermion<WilsonAdjImplF> WilsonAdjFermionF; | ||||||
| @@ -158,41 +158,41 @@ typedef DomainWallFermion<WilsonImplR> DomainWallFermionR; | |||||||
| typedef DomainWallFermion<WilsonImplF> DomainWallFermionF; | typedef DomainWallFermion<WilsonImplF> DomainWallFermionF; | ||||||
| typedef DomainWallFermion<WilsonImplD> DomainWallFermionD; | typedef DomainWallFermion<WilsonImplD> DomainWallFermionD; | ||||||
|  |  | ||||||
| //typedef DomainWallFermion<WilsonImplRL> DomainWallFermionRL; | typedef DomainWallFermion<WilsonImplRL> DomainWallFermionRL; | ||||||
| //typedef DomainWallFermion<WilsonImplFH> DomainWallFermionFH; | typedef DomainWallFermion<WilsonImplFH> DomainWallFermionFH; | ||||||
| //typedef DomainWallFermion<WilsonImplDF> DomainWallFermionDF; | typedef DomainWallFermion<WilsonImplDF> DomainWallFermionDF; | ||||||
|  |  | ||||||
| typedef DomainWallEOFAFermion<WilsonImplR> DomainWallEOFAFermionR; | typedef DomainWallEOFAFermion<WilsonImplR> DomainWallEOFAFermionR; | ||||||
| typedef DomainWallEOFAFermion<WilsonImplF> DomainWallEOFAFermionF; | typedef DomainWallEOFAFermion<WilsonImplF> DomainWallEOFAFermionF; | ||||||
| typedef DomainWallEOFAFermion<WilsonImplD> DomainWallEOFAFermionD; | typedef DomainWallEOFAFermion<WilsonImplD> DomainWallEOFAFermionD; | ||||||
|  |  | ||||||
| //typedef DomainWallEOFAFermion<WilsonImplRL> DomainWallEOFAFermionRL; | typedef DomainWallEOFAFermion<WilsonImplRL> DomainWallEOFAFermionRL; | ||||||
| //typedef DomainWallEOFAFermion<WilsonImplFH> DomainWallEOFAFermionFH; | typedef DomainWallEOFAFermion<WilsonImplFH> DomainWallEOFAFermionFH; | ||||||
| //typedef DomainWallEOFAFermion<WilsonImplDF> DomainWallEOFAFermionDF; | typedef DomainWallEOFAFermion<WilsonImplDF> DomainWallEOFAFermionDF; | ||||||
|  |  | ||||||
| typedef MobiusFermion<WilsonImplR> MobiusFermionR; | typedef MobiusFermion<WilsonImplR> MobiusFermionR; | ||||||
| typedef MobiusFermion<WilsonImplF> MobiusFermionF; | typedef MobiusFermion<WilsonImplF> MobiusFermionF; | ||||||
| typedef MobiusFermion<WilsonImplD> MobiusFermionD; | typedef MobiusFermion<WilsonImplD> MobiusFermionD; | ||||||
|  |  | ||||||
| //typedef MobiusFermion<WilsonImplRL> MobiusFermionRL; | typedef MobiusFermion<WilsonImplRL> MobiusFermionRL; | ||||||
| //typedef MobiusFermion<WilsonImplFH> MobiusFermionFH; | typedef MobiusFermion<WilsonImplFH> MobiusFermionFH; | ||||||
| //typedef MobiusFermion<WilsonImplDF> MobiusFermionDF; | typedef MobiusFermion<WilsonImplDF> MobiusFermionDF; | ||||||
|  |  | ||||||
| typedef MobiusEOFAFermion<WilsonImplR> MobiusEOFAFermionR; | typedef MobiusEOFAFermion<WilsonImplR> MobiusEOFAFermionR; | ||||||
| typedef MobiusEOFAFermion<WilsonImplF> MobiusEOFAFermionF; | typedef MobiusEOFAFermion<WilsonImplF> MobiusEOFAFermionF; | ||||||
| typedef MobiusEOFAFermion<WilsonImplD> MobiusEOFAFermionD; | typedef MobiusEOFAFermion<WilsonImplD> MobiusEOFAFermionD; | ||||||
|  |  | ||||||
| //typedef MobiusEOFAFermion<WilsonImplRL> MobiusEOFAFermionRL; | typedef MobiusEOFAFermion<WilsonImplRL> MobiusEOFAFermionRL; | ||||||
| //typedef MobiusEOFAFermion<WilsonImplFH> MobiusEOFAFermionFH; | typedef MobiusEOFAFermion<WilsonImplFH> MobiusEOFAFermionFH; | ||||||
| //typedef MobiusEOFAFermion<WilsonImplDF> MobiusEOFAFermionDF; | typedef MobiusEOFAFermion<WilsonImplDF> MobiusEOFAFermionDF; | ||||||
|  |  | ||||||
| typedef ZMobiusFermion<ZWilsonImplR> ZMobiusFermionR; | typedef ZMobiusFermion<ZWilsonImplR> ZMobiusFermionR; | ||||||
| typedef ZMobiusFermion<ZWilsonImplF> ZMobiusFermionF; | typedef ZMobiusFermion<ZWilsonImplF> ZMobiusFermionF; | ||||||
| typedef ZMobiusFermion<ZWilsonImplD> ZMobiusFermionD; | typedef ZMobiusFermion<ZWilsonImplD> ZMobiusFermionD; | ||||||
|  |  | ||||||
| //typedef ZMobiusFermion<ZWilsonImplRL> ZMobiusFermionRL; | typedef ZMobiusFermion<ZWilsonImplRL> ZMobiusFermionRL; | ||||||
| //typedef ZMobiusFermion<ZWilsonImplFH> ZMobiusFermionFH; | typedef ZMobiusFermion<ZWilsonImplFH> ZMobiusFermionFH; | ||||||
| //typedef ZMobiusFermion<ZWilsonImplDF> ZMobiusFermionDF; | typedef ZMobiusFermion<ZWilsonImplDF> ZMobiusFermionDF; | ||||||
|  |  | ||||||
| // Ls vectorised | // Ls vectorised | ||||||
| typedef ScaledShamirFermion<WilsonImplR> ScaledShamirFermionR; | typedef ScaledShamirFermion<WilsonImplR> ScaledShamirFermionR; | ||||||
| @@ -235,49 +235,49 @@ typedef WilsonFermion<GparityWilsonImplR>     GparityWilsonFermionR; | |||||||
| typedef WilsonFermion<GparityWilsonImplF>     GparityWilsonFermionF; | typedef WilsonFermion<GparityWilsonImplF>     GparityWilsonFermionF; | ||||||
| typedef WilsonFermion<GparityWilsonImplD>     GparityWilsonFermionD; | typedef WilsonFermion<GparityWilsonImplD>     GparityWilsonFermionD; | ||||||
|  |  | ||||||
| //typedef WilsonFermion<GparityWilsonImplRL>     GparityWilsonFermionRL; | typedef WilsonFermion<GparityWilsonImplRL>     GparityWilsonFermionRL; | ||||||
| //typedef WilsonFermion<GparityWilsonImplFH>     GparityWilsonFermionFH; | typedef WilsonFermion<GparityWilsonImplFH>     GparityWilsonFermionFH; | ||||||
| //typedef WilsonFermion<GparityWilsonImplDF>     GparityWilsonFermionDF; | typedef WilsonFermion<GparityWilsonImplDF>     GparityWilsonFermionDF; | ||||||
|  |  | ||||||
| typedef DomainWallFermion<GparityWilsonImplR> GparityDomainWallFermionR; | typedef DomainWallFermion<GparityWilsonImplR> GparityDomainWallFermionR; | ||||||
| typedef DomainWallFermion<GparityWilsonImplF> GparityDomainWallFermionF; | typedef DomainWallFermion<GparityWilsonImplF> GparityDomainWallFermionF; | ||||||
| typedef DomainWallFermion<GparityWilsonImplD> GparityDomainWallFermionD; | typedef DomainWallFermion<GparityWilsonImplD> GparityDomainWallFermionD; | ||||||
|  |  | ||||||
| //typedef DomainWallFermion<GparityWilsonImplRL> GparityDomainWallFermionRL; | typedef DomainWallFermion<GparityWilsonImplRL> GparityDomainWallFermionRL; | ||||||
| //typedef DomainWallFermion<GparityWilsonImplFH> GparityDomainWallFermionFH; | typedef DomainWallFermion<GparityWilsonImplFH> GparityDomainWallFermionFH; | ||||||
| //typedef DomainWallFermion<GparityWilsonImplDF> GparityDomainWallFermionDF; | typedef DomainWallFermion<GparityWilsonImplDF> GparityDomainWallFermionDF; | ||||||
|  |  | ||||||
| typedef DomainWallEOFAFermion<GparityWilsonImplR> GparityDomainWallEOFAFermionR; | typedef DomainWallEOFAFermion<GparityWilsonImplR> GparityDomainWallEOFAFermionR; | ||||||
| typedef DomainWallEOFAFermion<GparityWilsonImplF> GparityDomainWallEOFAFermionF; | typedef DomainWallEOFAFermion<GparityWilsonImplF> GparityDomainWallEOFAFermionF; | ||||||
| typedef DomainWallEOFAFermion<GparityWilsonImplD> GparityDomainWallEOFAFermionD; | typedef DomainWallEOFAFermion<GparityWilsonImplD> GparityDomainWallEOFAFermionD; | ||||||
|  |  | ||||||
| //typedef DomainWallEOFAFermion<GparityWilsonImplRL> GparityDomainWallEOFAFermionRL; | typedef DomainWallEOFAFermion<GparityWilsonImplRL> GparityDomainWallEOFAFermionRL; | ||||||
| //typedef DomainWallEOFAFermion<GparityWilsonImplFH> GparityDomainWallEOFAFermionFH; | typedef DomainWallEOFAFermion<GparityWilsonImplFH> GparityDomainWallEOFAFermionFH; | ||||||
| //typedef DomainWallEOFAFermion<GparityWilsonImplDF> GparityDomainWallEOFAFermionDF; | typedef DomainWallEOFAFermion<GparityWilsonImplDF> GparityDomainWallEOFAFermionDF; | ||||||
|  |  | ||||||
| typedef WilsonTMFermion<GparityWilsonImplR> GparityWilsonTMFermionR; | typedef WilsonTMFermion<GparityWilsonImplR> GparityWilsonTMFermionR; | ||||||
| typedef WilsonTMFermion<GparityWilsonImplF> GparityWilsonTMFermionF; | typedef WilsonTMFermion<GparityWilsonImplF> GparityWilsonTMFermionF; | ||||||
| typedef WilsonTMFermion<GparityWilsonImplD> GparityWilsonTMFermionD; | typedef WilsonTMFermion<GparityWilsonImplD> GparityWilsonTMFermionD; | ||||||
|  |  | ||||||
| //typedef WilsonTMFermion<GparityWilsonImplRL> GparityWilsonTMFermionRL; | typedef WilsonTMFermion<GparityWilsonImplRL> GparityWilsonTMFermionRL; | ||||||
| //typedef WilsonTMFermion<GparityWilsonImplFH> GparityWilsonTMFermionFH; | typedef WilsonTMFermion<GparityWilsonImplFH> GparityWilsonTMFermionFH; | ||||||
| //typedef WilsonTMFermion<GparityWilsonImplDF> GparityWilsonTMFermionDF; | typedef WilsonTMFermion<GparityWilsonImplDF> GparityWilsonTMFermionDF; | ||||||
|  |  | ||||||
| typedef MobiusFermion<GparityWilsonImplR> GparityMobiusFermionR; | typedef MobiusFermion<GparityWilsonImplR> GparityMobiusFermionR; | ||||||
| typedef MobiusFermion<GparityWilsonImplF> GparityMobiusFermionF; | typedef MobiusFermion<GparityWilsonImplF> GparityMobiusFermionF; | ||||||
| typedef MobiusFermion<GparityWilsonImplD> GparityMobiusFermionD; | typedef MobiusFermion<GparityWilsonImplD> GparityMobiusFermionD; | ||||||
|  |  | ||||||
| //typedef MobiusFermion<GparityWilsonImplRL> GparityMobiusFermionRL; | typedef MobiusFermion<GparityWilsonImplRL> GparityMobiusFermionRL; | ||||||
| //typedef MobiusFermion<GparityWilsonImplFH> GparityMobiusFermionFH; | typedef MobiusFermion<GparityWilsonImplFH> GparityMobiusFermionFH; | ||||||
| //typedef MobiusFermion<GparityWilsonImplDF> GparityMobiusFermionDF; | typedef MobiusFermion<GparityWilsonImplDF> GparityMobiusFermionDF; | ||||||
|  |  | ||||||
| typedef MobiusEOFAFermion<GparityWilsonImplR> GparityMobiusEOFAFermionR; | typedef MobiusEOFAFermion<GparityWilsonImplR> GparityMobiusEOFAFermionR; | ||||||
| typedef MobiusEOFAFermion<GparityWilsonImplF> GparityMobiusEOFAFermionF; | typedef MobiusEOFAFermion<GparityWilsonImplF> GparityMobiusEOFAFermionF; | ||||||
| typedef MobiusEOFAFermion<GparityWilsonImplD> GparityMobiusEOFAFermionD; | typedef MobiusEOFAFermion<GparityWilsonImplD> GparityMobiusEOFAFermionD; | ||||||
|  |  | ||||||
| //typedef MobiusEOFAFermion<GparityWilsonImplRL> GparityMobiusEOFAFermionRL; | typedef MobiusEOFAFermion<GparityWilsonImplRL> GparityMobiusEOFAFermionRL; | ||||||
| //typedef MobiusEOFAFermion<GparityWilsonImplFH> GparityMobiusEOFAFermionFH; | typedef MobiusEOFAFermion<GparityWilsonImplFH> GparityMobiusEOFAFermionFH; | ||||||
| //typedef MobiusEOFAFermion<GparityWilsonImplDF> GparityMobiusEOFAFermionDF; | typedef MobiusEOFAFermion<GparityWilsonImplDF> GparityMobiusEOFAFermionDF; | ||||||
|  |  | ||||||
| typedef ImprovedStaggeredFermion<StaggeredImplR> ImprovedStaggeredFermionR; | typedef ImprovedStaggeredFermion<StaggeredImplR> ImprovedStaggeredFermionR; | ||||||
| typedef ImprovedStaggeredFermion<StaggeredImplF> ImprovedStaggeredFermionF; | typedef ImprovedStaggeredFermion<StaggeredImplF> ImprovedStaggeredFermionF; | ||||||
| @@ -291,6 +291,12 @@ typedef ImprovedStaggeredFermion5D<StaggeredImplR> ImprovedStaggeredFermion5DR; | |||||||
| typedef ImprovedStaggeredFermion5D<StaggeredImplF> ImprovedStaggeredFermion5DF; | typedef ImprovedStaggeredFermion5D<StaggeredImplF> ImprovedStaggeredFermion5DF; | ||||||
| typedef ImprovedStaggeredFermion5D<StaggeredImplD> ImprovedStaggeredFermion5DD; | typedef ImprovedStaggeredFermion5D<StaggeredImplD> ImprovedStaggeredFermion5DD; | ||||||
|  |  | ||||||
|  | #ifndef GRID_CUDA | ||||||
|  | typedef ImprovedStaggeredFermion5D<StaggeredVec5dImplR> ImprovedStaggeredFermionVec5dR; | ||||||
|  | typedef ImprovedStaggeredFermion5D<StaggeredVec5dImplF> ImprovedStaggeredFermionVec5dF; | ||||||
|  | typedef ImprovedStaggeredFermion5D<StaggeredVec5dImplD> ImprovedStaggeredFermionVec5dD; | ||||||
|  | #endif | ||||||
|  |  | ||||||
| NAMESPACE_END(Grid); | NAMESPACE_END(Grid); | ||||||
|  |  | ||||||
| //////////////////// | //////////////////// | ||||||
|   | |||||||
| @@ -153,8 +153,8 @@ public: | |||||||
|   typedef typename Impl::StencilImpl             StencilImpl;		\ |   typedef typename Impl::StencilImpl             StencilImpl;		\ | ||||||
|   typedef typename Impl::ImplParams               ImplParams;	        \ |   typedef typename Impl::ImplParams               ImplParams;	        \ | ||||||
|   typedef typename Impl::StencilImpl::View_type  StencilView;		\ |   typedef typename Impl::StencilImpl::View_type  StencilView;		\ | ||||||
|   typedef const typename ViewMap<FermionField>::Type      FermionFieldView;	\ |   typedef typename ViewMap<FermionField>::Type      FermionFieldView;	\ | ||||||
|   typedef const typename ViewMap<DoubledGaugeField>::Type DoubledGaugeFieldView; |   typedef typename ViewMap<DoubledGaugeField>::Type DoubledGaugeFieldView; | ||||||
|  |  | ||||||
| #define INHERIT_IMPL_TYPES(Base)		\ | #define INHERIT_IMPL_TYPES(Base)		\ | ||||||
|   INHERIT_GIMPL_TYPES(Base)			\ |   INHERIT_GIMPL_TYPES(Base)			\ | ||||||
| @@ -183,8 +183,7 @@ NAMESPACE_CHECK(ImplStaggered); | |||||||
| ///////////////////////////////////////////////////////////////////////////// | ///////////////////////////////////////////////////////////////////////////// | ||||||
| // Single flavour one component spinors with colour index. 5d vec | // Single flavour one component spinors with colour index. 5d vec | ||||||
| ///////////////////////////////////////////////////////////////////////////// | ///////////////////////////////////////////////////////////////////////////// | ||||||
| // Deprecate Vec5d | #include <Grid/qcd/action/fermion/StaggeredVec5dImpl.h>  | ||||||
| //#include <Grid/qcd/action/fermion/StaggeredVec5dImpl.h>  | NAMESPACE_CHECK(ImplStaggered5dVec);   | ||||||
| //NAMESPACE_CHECK(ImplStaggered5dVec);   |  | ||||||
|  |  | ||||||
|  |  | ||||||
|   | |||||||
| @@ -97,30 +97,42 @@ public: | |||||||
|     Coordinate icoor; |     Coordinate icoor; | ||||||
|  |  | ||||||
| #ifdef GRID_SIMT | #ifdef GRID_SIMT | ||||||
|  |     _Spinor tmp; | ||||||
|  |  | ||||||
|     const int Nsimd =SiteDoubledGaugeField::Nsimd(); |     const int Nsimd =SiteDoubledGaugeField::Nsimd(); | ||||||
|     int s = acceleratorSIMTlane(Nsimd); |     int s = acceleratorSIMTlane(Nsimd); | ||||||
|     St.iCoorFromIindex(icoor,s); |     St.iCoorFromIindex(icoor,s); | ||||||
|  |  | ||||||
|     int mmu = mu % Nd; |     int mmu = mu % Nd; | ||||||
|  |     if ( SE->_around_the_world && St.parameters.twists[mmu] ) { | ||||||
|        |        | ||||||
|     auto UU0=coalescedRead(U(0)(mu)); |       int permute_lane = (sl==1)  | ||||||
|     auto UU1=coalescedRead(U(1)(mu)); |     	|| ((distance== 1)&&(icoor[direction]==1)) | ||||||
|  | 	|| ((distance==-1)&&(icoor[direction]==0)); | ||||||
|  |  | ||||||
|     //Decide whether we do a G-parity flavor twist |       if ( permute_lane ) {  | ||||||
|     //Note: this assumes (but does not check) that sl==1 || sl==2 i.e. max 2 SIMD lanes in G-parity dir | 	tmp(0) = chi(1); | ||||||
|     //It also assumes (but does not check) that abs(distance) == 1 | 	tmp(1) = chi(0); | ||||||
|     int permute_lane = (sl==1)  |       } else { | ||||||
|     || ((distance== 1)&&(icoor[direction]==1)) | 	tmp(0) = chi(0); | ||||||
|     || ((distance==-1)&&(icoor[direction]==0)); | 	tmp(1) = chi(1); | ||||||
|  |       } | ||||||
|  |  | ||||||
|     permute_lane = permute_lane && SE->_around_the_world && St.parameters.twists[mmu]; //only if we are going around the world |       auto UU0=coalescedRead(U(0)(mu)); | ||||||
|  |       auto UU1=coalescedRead(U(1)(mu)); | ||||||
|  |  | ||||||
|     //Apply the links |       mult(&phi(0),&UU0,&tmp(0)); | ||||||
|     int f_upper = permute_lane ? 1 : 0; |       mult(&phi(1),&UU1,&tmp(1)); | ||||||
|     int f_lower = !f_upper; |  | ||||||
|  |  | ||||||
|     mult(&phi(0),&UU0,&chi(f_upper)); |     } else { | ||||||
|     mult(&phi(1),&UU1,&chi(f_lower)); |  | ||||||
|  |       auto UU0=coalescedRead(U(0)(mu)); | ||||||
|  |       auto UU1=coalescedRead(U(1)(mu)); | ||||||
|  |  | ||||||
|  |       mult(&phi(0),&UU0,&chi(0)); | ||||||
|  |       mult(&phi(1),&UU1,&chi(1)); | ||||||
|  |  | ||||||
|  |     } | ||||||
|  |  | ||||||
| #else | #else | ||||||
|     typedef _Spinor vobj; |     typedef _Spinor vobj; | ||||||
| @@ -327,8 +339,8 @@ typedef GparityWilsonImpl<vComplex , FundamentalRepresentation,CoeffReal> Gparit | |||||||
| typedef GparityWilsonImpl<vComplexF, FundamentalRepresentation,CoeffReal> GparityWilsonImplF;  // Float | typedef GparityWilsonImpl<vComplexF, FundamentalRepresentation,CoeffReal> GparityWilsonImplF;  // Float | ||||||
| typedef GparityWilsonImpl<vComplexD, FundamentalRepresentation,CoeffReal> GparityWilsonImplD;  // Double | typedef GparityWilsonImpl<vComplexD, FundamentalRepresentation,CoeffReal> GparityWilsonImplD;  // Double | ||||||
|   |   | ||||||
| //typedef GparityWilsonImpl<vComplex , FundamentalRepresentation,CoeffRealHalfComms> GparityWilsonImplRL;  // Real.. whichever prec | typedef GparityWilsonImpl<vComplex , FundamentalRepresentation,CoeffRealHalfComms> GparityWilsonImplRL;  // Real.. whichever prec | ||||||
| //typedef GparityWilsonImpl<vComplexF, FundamentalRepresentation,CoeffRealHalfComms> GparityWilsonImplFH;  // Float | typedef GparityWilsonImpl<vComplexF, FundamentalRepresentation,CoeffRealHalfComms> GparityWilsonImplFH;  // Float | ||||||
| //typedef GparityWilsonImpl<vComplexD, FundamentalRepresentation,CoeffRealHalfComms> GparityWilsonImplDF;  // Double | typedef GparityWilsonImpl<vComplexD, FundamentalRepresentation,CoeffRealHalfComms> GparityWilsonImplDF;  // Double | ||||||
|  |  | ||||||
| NAMESPACE_END(Grid); | NAMESPACE_END(Grid); | ||||||
|   | |||||||
| @@ -85,7 +85,7 @@ class MADWF | |||||||
|       maxiter     =_maxiter; |       maxiter     =_maxiter; | ||||||
|     }; |     }; | ||||||
|     |     | ||||||
|   void operator() (const FermionFieldo &src,FermionFieldo &sol5) |   void operator() (const FermionFieldo &src4,FermionFieldo &sol5) | ||||||
|   { |   { | ||||||
|     std::cout << GridLogMessage<< " ************************************************" << std::endl; |     std::cout << GridLogMessage<< " ************************************************" << std::endl; | ||||||
|     std::cout << GridLogMessage<< "  MADWF-like algorithm                           " << std::endl; |     std::cout << GridLogMessage<< "  MADWF-like algorithm                           " << std::endl; | ||||||
| @@ -114,16 +114,8 @@ class MADWF | |||||||
|     /////////////////////////////////////// |     /////////////////////////////////////// | ||||||
|     //Import source, include Dminus factors |     //Import source, include Dminus factors | ||||||
|     /////////////////////////////////////// |     /////////////////////////////////////// | ||||||
|     GridBase *src_grid = src.Grid(); |     Mato.ImportPhysicalFermionSource(src4,b);  | ||||||
|  |     std::cout << GridLogMessage << " src4 " <<norm2(src4)<<std::endl; | ||||||
|     assert( (src_grid == Mato.GaugeGrid()) || (src_grid == Mato.FermionGrid())); |  | ||||||
|  |  | ||||||
|     if ( src_grid == Mato.GaugeGrid() ) { |  | ||||||
|       Mato.ImportPhysicalFermionSource(src,b); |  | ||||||
|     } else { |  | ||||||
|       b=src; |  | ||||||
|     } |  | ||||||
|     std::cout << GridLogMessage << " src " <<norm2(src)<<std::endl; |  | ||||||
|     std::cout << GridLogMessage << " b    " <<norm2(b)<<std::endl; |     std::cout << GridLogMessage << " b    " <<norm2(b)<<std::endl; | ||||||
|  |  | ||||||
|     defect = b; |     defect = b; | ||||||
|   | |||||||
| @@ -72,23 +72,19 @@ public: | |||||||
|      |      | ||||||
|   StaggeredImpl(const ImplParams &p = ImplParams()) : Params(p){}; |   StaggeredImpl(const ImplParams &p = ImplParams()) : Params(p){}; | ||||||
|        |        | ||||||
|   template<class _Spinor> |   static accelerator_inline void multLink(SiteSpinor &phi, | ||||||
|   static accelerator_inline void multLink(_Spinor &phi, |  | ||||||
| 		       const SiteDoubledGaugeField &U, | 		       const SiteDoubledGaugeField &U, | ||||||
| 		       const _Spinor &chi, | 		       const SiteSpinor &chi, | ||||||
| 		       int mu) | 		       int mu) | ||||||
|   { |   { | ||||||
|     auto UU = coalescedRead(U(mu)); |     mult(&phi(), &U(mu), &chi()); | ||||||
|     mult(&phi(), &UU, &chi()); |  | ||||||
|   } |   } | ||||||
|   template<class _Spinor> |   static accelerator_inline void multLinkAdd(SiteSpinor &phi, | ||||||
|   static accelerator_inline void multLinkAdd(_Spinor &phi, |  | ||||||
| 			  const SiteDoubledGaugeField &U, | 			  const SiteDoubledGaugeField &U, | ||||||
| 			  const _Spinor &chi, | 			  const SiteSpinor &chi, | ||||||
| 			  int mu) | 			  int mu) | ||||||
|   { |   { | ||||||
|     auto UU = coalescedRead(U(mu)); |     mac(&phi(), &U(mu), &chi()); | ||||||
|     mac(&phi(), &UU, &chi()); |  | ||||||
|   } |   } | ||||||
|        |        | ||||||
|   template <class ref> |   template <class ref> | ||||||
|   | |||||||
| @@ -245,7 +245,7 @@ public: | |||||||
|     return out; |     return out; | ||||||
|   } |   } | ||||||
|  |  | ||||||
| protected: | private: | ||||||
|   // here fixing the 4 dimensions, make it more general? |   // here fixing the 4 dimensions, make it more general? | ||||||
|  |  | ||||||
|   RealD csw_r;                                               // Clover coefficient - spatial |   RealD csw_r;                                               // Clover coefficient - spatial | ||||||
|   | |||||||
| @@ -61,19 +61,18 @@ public: | |||||||
|   typedef typename SiteHalfSpinor::vector_type     vComplexHigh; |   typedef typename SiteHalfSpinor::vector_type     vComplexHigh; | ||||||
|   constexpr static int Nw=sizeof(SiteHalfSpinor)/sizeof(vComplexHigh); |   constexpr static int Nw=sizeof(SiteHalfSpinor)/sizeof(vComplexHigh); | ||||||
|  |  | ||||||
|   accelerator_inline int CommDatumSize(void) const { |   accelerator_inline int CommDatumSize(void) { | ||||||
|     return sizeof(SiteHalfCommSpinor); |     return sizeof(SiteHalfCommSpinor); | ||||||
|   } |   } | ||||||
|  |  | ||||||
|   /*****************************************************/ |   /*****************************************************/ | ||||||
|   /* Compress includes precision change if mpi data is not same */ |   /* Compress includes precision change if mpi data is not same */ | ||||||
|   /*****************************************************/ |   /*****************************************************/ | ||||||
|   accelerator_inline void Compress(SiteHalfSpinor &buf,const SiteSpinor &in) const { |   template<class _SiteHalfSpinor, class _SiteSpinor> | ||||||
|     typedef decltype(coalescedRead(buf)) sobj; |   accelerator_inline void Compress(_SiteHalfSpinor *buf,Integer o,const _SiteSpinor &in) { | ||||||
|     sobj sp; |     _SiteHalfSpinor tmp; | ||||||
|     auto sin = coalescedRead(in); |     projector::Proj(tmp,in,mu,dag); | ||||||
|     projector::Proj(sp,sin,mu,dag); |     vstream(buf[o],tmp); | ||||||
|     coalescedWrite(buf,sp); |  | ||||||
|   } |   } | ||||||
|  |  | ||||||
|   /*****************************************************/ |   /*****************************************************/ | ||||||
| @@ -82,24 +81,19 @@ public: | |||||||
|   accelerator_inline void Exchange(SiteHalfSpinor *mp, |   accelerator_inline void Exchange(SiteHalfSpinor *mp, | ||||||
| 				   const SiteHalfSpinor * __restrict__ vp0, | 				   const SiteHalfSpinor * __restrict__ vp0, | ||||||
| 				   const SiteHalfSpinor * __restrict__ vp1, | 				   const SiteHalfSpinor * __restrict__ vp1, | ||||||
| 				   Integer type,Integer o) const { | 				   Integer type,Integer o){ | ||||||
| #ifdef GRID_SIMT |  | ||||||
|     exchangeSIMT(mp[2*o],mp[2*o+1],vp0[o],vp1[o],type); |  | ||||||
| #else |  | ||||||
|     SiteHalfSpinor tmp1; |     SiteHalfSpinor tmp1; | ||||||
|     SiteHalfSpinor tmp2; |     SiteHalfSpinor tmp2; | ||||||
|     exchange(tmp1,tmp2,vp0[o],vp1[o],type); |     exchange(tmp1,tmp2,vp0[o],vp1[o],type); | ||||||
|     vstream(mp[2*o  ],tmp1); |     vstream(mp[2*o  ],tmp1); | ||||||
|     vstream(mp[2*o+1],tmp2); |     vstream(mp[2*o+1],tmp2); | ||||||
| #endif |  | ||||||
|   } |   } | ||||||
|  |  | ||||||
|  |  | ||||||
|   /*****************************************************/ |   /*****************************************************/ | ||||||
|   /* Have a decompression step if mpi data is not same */ |   /* Have a decompression step if mpi data is not same */ | ||||||
|   /*****************************************************/ |   /*****************************************************/ | ||||||
|   accelerator_inline void Decompress(SiteHalfSpinor * __restrict__ out, |   accelerator_inline void Decompress(SiteHalfSpinor * __restrict__ out, | ||||||
| 				     SiteHalfSpinor * __restrict__ in, Integer o) const {     | 				     SiteHalfSpinor * __restrict__ in, Integer o) {     | ||||||
|     assert(0); |     assert(0); | ||||||
|   } |   } | ||||||
|  |  | ||||||
| @@ -109,30 +103,8 @@ public: | |||||||
|   accelerator_inline void CompressExchange(SiteHalfSpinor * __restrict__ out0, |   accelerator_inline void CompressExchange(SiteHalfSpinor * __restrict__ out0, | ||||||
| 					   SiteHalfSpinor * __restrict__ out1, | 					   SiteHalfSpinor * __restrict__ out1, | ||||||
| 					   const SiteSpinor * __restrict__ in, | 					   const SiteSpinor * __restrict__ in, | ||||||
| 					   Integer j,Integer k, Integer m,Integer type) const | 					   Integer j,Integer k, Integer m,Integer type) | ||||||
|   { |   { | ||||||
| #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 temp1, temp2; | ||||||
|     SiteHalfSpinor temp3, temp4; |     SiteHalfSpinor temp3, temp4; | ||||||
|     projector::Proj(temp1,in[k],mu,dag); |     projector::Proj(temp1,in[k],mu,dag); | ||||||
| @@ -140,17 +112,15 @@ public: | |||||||
|     exchange(temp3,temp4,temp1,temp2,type); |     exchange(temp3,temp4,temp1,temp2,type); | ||||||
|     vstream(out0[j],temp3); |     vstream(out0[j],temp3); | ||||||
|     vstream(out1[j],temp4); |     vstream(out1[j],temp4); | ||||||
| #endif |  | ||||||
|   } |   } | ||||||
|  |  | ||||||
|   /*****************************************************/ |   /*****************************************************/ | ||||||
|   /* Pass the info to the stencil */ |   /* Pass the info to the stencil */ | ||||||
|   /*****************************************************/ |   /*****************************************************/ | ||||||
|   accelerator_inline bool DecompressionStep(void) const { return false; } |   accelerator_inline bool DecompressionStep(void) { return false; } | ||||||
|  |  | ||||||
| }; | }; | ||||||
|  |  | ||||||
| #if 0 |  | ||||||
| template<class _HCspinor,class _Hspinor,class _Spinor, class projector> | template<class _HCspinor,class _Hspinor,class _Spinor, class projector> | ||||||
| class WilsonCompressorTemplate< _HCspinor, _Hspinor, _Spinor, projector, | class WilsonCompressorTemplate< _HCspinor, _Hspinor, _Spinor, projector, | ||||||
| 				typename std::enable_if<!std::is_same<_HCspinor,_Hspinor>::value>::type > | 				typename std::enable_if<!std::is_same<_HCspinor,_Hspinor>::value>::type > | ||||||
| @@ -172,30 +142,20 @@ public: | |||||||
|   typedef typename SiteHalfSpinor::vector_type     vComplexHigh; |   typedef typename SiteHalfSpinor::vector_type     vComplexHigh; | ||||||
|   constexpr static int Nw=sizeof(SiteHalfSpinor)/sizeof(vComplexHigh); |   constexpr static int Nw=sizeof(SiteHalfSpinor)/sizeof(vComplexHigh); | ||||||
|  |  | ||||||
|   accelerator_inline int CommDatumSize(void) const { |   accelerator_inline int CommDatumSize(void) { | ||||||
|     return sizeof(SiteHalfCommSpinor); |     return sizeof(SiteHalfCommSpinor); | ||||||
|   } |   } | ||||||
|  |  | ||||||
|   /*****************************************************/ |   /*****************************************************/ | ||||||
|   /* Compress includes precision change if mpi data is not same */ |   /* Compress includes precision change if mpi data is not same */ | ||||||
|   /*****************************************************/ |   /*****************************************************/ | ||||||
|   accelerator_inline void Compress(SiteHalfSpinor &buf,const SiteSpinor &in) const { |   template<class _SiteHalfSpinor, class _SiteSpinor> | ||||||
|     SiteHalfSpinor hsp; |   accelerator_inline void Compress(_SiteHalfSpinor *buf,Integer o,const _SiteSpinor &in) { | ||||||
|  |     _SiteHalfSpinor hsp; | ||||||
|     SiteHalfCommSpinor *hbuf = (SiteHalfCommSpinor *)buf; |     SiteHalfCommSpinor *hbuf = (SiteHalfCommSpinor *)buf; | ||||||
|     projector::Proj(hsp,in,mu,dag); |     projector::Proj(hsp,in,mu,dag); | ||||||
|     precisionChange((vComplexLow *)&hbuf[o],(vComplexHigh *)&hsp,Nw); |     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 */ |   /* Exchange includes precision change if mpi data is not same */ | ||||||
| @@ -203,7 +163,7 @@ public: | |||||||
|   accelerator_inline void Exchange(SiteHalfSpinor *mp, |   accelerator_inline void Exchange(SiteHalfSpinor *mp, | ||||||
|                        SiteHalfSpinor *vp0, |                        SiteHalfSpinor *vp0, | ||||||
|                        SiteHalfSpinor *vp1, |                        SiteHalfSpinor *vp1, | ||||||
| 		       Integer type,Integer o) const { | 		       Integer type,Integer o){ | ||||||
|     SiteHalfSpinor vt0,vt1; |     SiteHalfSpinor vt0,vt1; | ||||||
|     SiteHalfCommSpinor *vpp0 = (SiteHalfCommSpinor *)vp0; |     SiteHalfCommSpinor *vpp0 = (SiteHalfCommSpinor *)vp0; | ||||||
|     SiteHalfCommSpinor *vpp1 = (SiteHalfCommSpinor *)vp1; |     SiteHalfCommSpinor *vpp1 = (SiteHalfCommSpinor *)vp1; | ||||||
| @@ -215,7 +175,7 @@ public: | |||||||
|   /*****************************************************/ |   /*****************************************************/ | ||||||
|   /* Have a decompression step if mpi data is not same */ |   /* Have a decompression step if mpi data is not same */ | ||||||
|   /*****************************************************/ |   /*****************************************************/ | ||||||
|   accelerator_inline void Decompress(SiteHalfSpinor *out, SiteHalfSpinor *in, Integer o) const { |   accelerator_inline void Decompress(SiteHalfSpinor *out, SiteHalfSpinor *in, Integer o){ | ||||||
|     SiteHalfCommSpinor *hin=(SiteHalfCommSpinor *)in; |     SiteHalfCommSpinor *hin=(SiteHalfCommSpinor *)in; | ||||||
|     precisionChange((vComplexHigh *)&out[o],(vComplexLow *)&hin[o],Nw); |     precisionChange((vComplexHigh *)&out[o],(vComplexLow *)&hin[o],Nw); | ||||||
|   } |   } | ||||||
| @@ -226,7 +186,7 @@ public: | |||||||
|   accelerator_inline void CompressExchange(SiteHalfSpinor *out0, |   accelerator_inline void CompressExchange(SiteHalfSpinor *out0, | ||||||
| 			       SiteHalfSpinor *out1, | 			       SiteHalfSpinor *out1, | ||||||
| 			       const SiteSpinor *in, | 			       const SiteSpinor *in, | ||||||
| 			       Integer j,Integer k, Integer m,Integer type) const { | 			       Integer j,Integer k, Integer m,Integer type){ | ||||||
|     SiteHalfSpinor temp1, temp2,temp3,temp4; |     SiteHalfSpinor temp1, temp2,temp3,temp4; | ||||||
|     SiteHalfCommSpinor *hout0 = (SiteHalfCommSpinor *)out0; |     SiteHalfCommSpinor *hout0 = (SiteHalfCommSpinor *)out0; | ||||||
|     SiteHalfCommSpinor *hout1 = (SiteHalfCommSpinor *)out1; |     SiteHalfCommSpinor *hout1 = (SiteHalfCommSpinor *)out1; | ||||||
| @@ -240,10 +200,9 @@ public: | |||||||
|   /*****************************************************/ |   /*****************************************************/ | ||||||
|   /* Pass the info to the stencil */ |   /* Pass the info to the stencil */ | ||||||
|   /*****************************************************/ |   /*****************************************************/ | ||||||
|   accelerator_inline bool DecompressionStep(void) const { return true; } |   accelerator_inline bool DecompressionStep(void) { return true; } | ||||||
|  |  | ||||||
| }; | }; | ||||||
| #endif |  | ||||||
|  |  | ||||||
| #define DECLARE_PROJ(Projector,Compressor,spProj)			\ | #define DECLARE_PROJ(Projector,Compressor,spProj)			\ | ||||||
|   class Projector {							\ |   class Projector {							\ | ||||||
| @@ -294,8 +253,33 @@ public: | |||||||
|   typedef typename Base::View_type View_type; |   typedef typename Base::View_type View_type; | ||||||
|   typedef typename Base::StencilVector StencilVector; |   typedef typename Base::StencilVector StencilVector; | ||||||
|  |  | ||||||
|   void ZeroCountersi(void)  {  } |   double timer0; | ||||||
|   void Reporti(int calls)  {  } |   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; |   std::vector<int> surface_list; | ||||||
|  |  | ||||||
| @@ -337,18 +321,26 @@ public: | |||||||
|   { |   { | ||||||
|     std::vector<std::vector<CommsRequest_t> > reqs; |     std::vector<std::vector<CommsRequest_t> > reqs; | ||||||
|     this->HaloExchangeOptGather(source,compress); |     this->HaloExchangeOptGather(source,compress); | ||||||
|  |     double t1=usecond(); | ||||||
|     // Asynchronous MPI calls multidirectional, Isend etc... |     // Asynchronous MPI calls multidirectional, Isend etc... | ||||||
|     // Non-overlapped directions within a thread. Asynchronous calls except MPI3, threaded up to comm threads ways. |     // Non-overlapped directions within a thread. Asynchronous calls except MPI3, threaded up to comm threads ways. | ||||||
|     this->Communicate(); |     this->Communicate(); | ||||||
|  |     double t2=usecond(); timer1 += t2-t1; | ||||||
|     this->CommsMerge(compress); |     this->CommsMerge(compress); | ||||||
|  |     double t3=usecond(); timer2 += t3-t2; | ||||||
|     this->CommsMergeSHM(compress); |     this->CommsMergeSHM(compress); | ||||||
|  |     double t4=usecond(); timer3 += t4-t3; | ||||||
|   } |   } | ||||||
|    |    | ||||||
|   template <class compressor> |   template <class compressor> | ||||||
|   void HaloExchangeOptGather(const Lattice<vobj> &source,compressor &compress)  |   void HaloExchangeOptGather(const Lattice<vobj> &source,compressor &compress)  | ||||||
|   { |   { | ||||||
|     this->Prepare(); |     this->Prepare(); | ||||||
|  |     double t0=usecond(); | ||||||
|     this->HaloGatherOpt(source,compress); |     this->HaloGatherOpt(source,compress); | ||||||
|  |     double t1=usecond(); | ||||||
|  |     timer0 += t1-t0; | ||||||
|  |     callsi++; | ||||||
|   } |   } | ||||||
|  |  | ||||||
|   template <class compressor> |   template <class compressor> | ||||||
| @@ -360,9 +352,12 @@ public: | |||||||
|     typedef typename compressor::SiteHalfSpinor     SiteHalfSpinor; |     typedef typename compressor::SiteHalfSpinor     SiteHalfSpinor; | ||||||
|     typedef typename compressor::SiteHalfCommSpinor SiteHalfCommSpinor; |     typedef typename compressor::SiteHalfCommSpinor SiteHalfCommSpinor; | ||||||
|  |  | ||||||
|  |     this->mpi3synctime_g-=usecond(); | ||||||
|     this->_grid->StencilBarrier(); |     this->_grid->StencilBarrier(); | ||||||
|  |     this->mpi3synctime_g+=usecond(); | ||||||
|  |  | ||||||
|     assert(source.Grid()==this->_grid); |     assert(source.Grid()==this->_grid); | ||||||
|  |     this->halogtime-=usecond(); | ||||||
|      |      | ||||||
|     this->u_comm_offset=0; |     this->u_comm_offset=0; | ||||||
|        |        | ||||||
| @@ -398,6 +393,7 @@ public: | |||||||
|     } |     } | ||||||
|     this->face_table_computed=1; |     this->face_table_computed=1; | ||||||
|     assert(this->u_comm_offset==this->_unified_buffer_size); |     assert(this->u_comm_offset==this->_unified_buffer_size); | ||||||
|  |     this->halogtime+=usecond(); | ||||||
|     accelerator_barrier(); |     accelerator_barrier(); | ||||||
|   } |   } | ||||||
|  |  | ||||||
|   | |||||||
| @@ -72,7 +72,7 @@ public: | |||||||
|   typedef WilsonCompressor<SiteHalfCommSpinor,SiteHalfSpinor, SiteSpinor> Compressor; |   typedef WilsonCompressor<SiteHalfCommSpinor,SiteHalfSpinor, SiteSpinor> Compressor; | ||||||
|   typedef WilsonImplParams ImplParams; |   typedef WilsonImplParams ImplParams; | ||||||
|   typedef WilsonStencil<SiteSpinor, SiteHalfSpinor,ImplParams> StencilImpl; |   typedef WilsonStencil<SiteSpinor, SiteHalfSpinor,ImplParams> StencilImpl; | ||||||
|   typedef const typename StencilImpl::View_type StencilView; |   typedef typename StencilImpl::View_type StencilView; | ||||||
|      |      | ||||||
|   ImplParams Params; |   ImplParams Params; | ||||||
|  |  | ||||||
| @@ -106,15 +106,11 @@ public: | |||||||
| 			    const _SpinorField & phi, | 			    const _SpinorField & phi, | ||||||
| 			    int mu) | 			    int mu) | ||||||
|   { |   { | ||||||
|     const int Nsimd = SiteHalfSpinor::Nsimd(); |  | ||||||
|     autoView( out_v, out, AcceleratorWrite); |     autoView( out_v, out, AcceleratorWrite); | ||||||
|     autoView( phi_v, phi, AcceleratorRead); |     autoView( phi_v, phi, AcceleratorRead); | ||||||
|     autoView( Umu_v, Umu, AcceleratorRead); |     autoView( Umu_v, Umu, AcceleratorRead); | ||||||
|     typedef decltype(coalescedRead(out_v[0]))   calcSpinor; |     accelerator_for(sss,out.Grid()->oSites(),1,{ | ||||||
|     accelerator_for(sss,out.Grid()->oSites(),Nsimd,{ | 	multLink(out_v[sss],Umu_v[sss],phi_v[sss],mu); | ||||||
| 	calcSpinor tmp; |  | ||||||
| 	multLink(tmp,Umu_v[sss],phi_v(sss),mu); |  | ||||||
| 	coalescedWrite(out_v[sss],tmp); |  | ||||||
|     }); |     }); | ||||||
|   } |   } | ||||||
| 					    | 					    | ||||||
| @@ -184,22 +180,18 @@ public: | |||||||
|       mat = TraceIndex<SpinIndex>(P);  |       mat = TraceIndex<SpinIndex>(P);  | ||||||
|     } |     } | ||||||
|        |        | ||||||
|     inline void extractLinkField(std::vector<GaugeLinkField> &mat, DoubledGaugeField &Uds) |     inline void extractLinkField(std::vector<GaugeLinkField> &mat, DoubledGaugeField &Uds){ | ||||||
|     { |  | ||||||
|       for (int mu = 0; mu < Nd; mu++) |       for (int mu = 0; mu < Nd; mu++) | ||||||
|       mat[mu] = PeekIndex<LorentzIndex>(Uds, mu); |       mat[mu] = PeekIndex<LorentzIndex>(Uds, mu); | ||||||
|     } |     } | ||||||
|  |  | ||||||
|   inline void InsertForce5D(GaugeField &mat, FermionField &Btilde, FermionField Ã,int mu) |  | ||||||
|   { |   inline void InsertForce5D(GaugeField &mat, FermionField &Btilde, FermionField Ã,int mu){ | ||||||
| #undef USE_OLD_INSERT_FORCE     |        | ||||||
|     int Ls=Btilde.Grid()->_fdimensions[0]; |     int Ls=Btilde.Grid()->_fdimensions[0]; | ||||||
|     autoView( mat_v , mat, AcceleratorWrite); |  | ||||||
| #ifdef USE_OLD_INSERT_FORCE     |  | ||||||
|     GaugeLinkField tmp(mat.Grid()); |     GaugeLinkField tmp(mat.Grid()); | ||||||
|     tmp = Zero(); |     tmp = Zero(); | ||||||
|     { |     { | ||||||
|       const int Nsimd = SiteSpinor::Nsimd(); |  | ||||||
|       autoView( tmp_v , tmp, AcceleratorWrite); |       autoView( tmp_v , tmp, AcceleratorWrite); | ||||||
|       autoView( Btilde_v , Btilde, AcceleratorRead); |       autoView( Btilde_v , Btilde, AcceleratorRead); | ||||||
|       autoView( Atilde_v , Atilde, AcceleratorRead); |       autoView( Atilde_v , Atilde, AcceleratorRead); | ||||||
| @@ -212,29 +204,6 @@ public: | |||||||
| 	}); | 	}); | ||||||
|     } |     } | ||||||
|     PokeIndex<LorentzIndex>(mat,tmp,mu); |     PokeIndex<LorentzIndex>(mat,tmp,mu); | ||||||
| #else |  | ||||||
|     { |  | ||||||
|       const int Nsimd = SiteSpinor::Nsimd(); |  | ||||||
|       autoView( Btilde_v , Btilde, AcceleratorRead); |  | ||||||
|       autoView( Atilde_v , Atilde, AcceleratorRead); |  | ||||||
|       accelerator_for(sss,mat.Grid()->oSites(),Nsimd,{ |  | ||||||
| 	  int sU=sss; |  | ||||||
|   	  typedef decltype(coalescedRead(mat_v[sU](mu)() )) ColorMatrixType; |  | ||||||
|   	  ColorMatrixType sum; |  | ||||||
| 	  zeroit(sum);   |  | ||||||
| 	  for(int s=0;s<Ls;s++){ |  | ||||||
| 	    int sF = s+Ls*sU; |  | ||||||
|   	    for(int spn=0;spn<Ns;spn++){ //sum over spin |  | ||||||
|   	      auto bb = coalescedRead(Btilde_v[sF]()(spn) ); //color vector |  | ||||||
|   	      auto aa = coalescedRead(Atilde_v[sF]()(spn) ); |  | ||||||
| 	      auto op = outerProduct(bb,aa); |  | ||||||
|   	      sum = sum + op; |  | ||||||
| 	    } |  | ||||||
| 	  } |  | ||||||
|   	  coalescedWrite(mat_v[sU](mu)(), sum); |  | ||||||
|       }); |  | ||||||
|     } |  | ||||||
| #endif     |  | ||||||
|   } |   } | ||||||
| }; | }; | ||||||
|  |  | ||||||
| @@ -243,17 +212,17 @@ typedef WilsonImpl<vComplex,  FundamentalRepresentation, CoeffReal > WilsonImplR | |||||||
| typedef WilsonImpl<vComplexF, FundamentalRepresentation, CoeffReal > WilsonImplF;  // Float | typedef WilsonImpl<vComplexF, FundamentalRepresentation, CoeffReal > WilsonImplF;  // Float | ||||||
| typedef WilsonImpl<vComplexD, FundamentalRepresentation, CoeffReal > WilsonImplD;  // Double | typedef WilsonImpl<vComplexD, FundamentalRepresentation, CoeffReal > WilsonImplD;  // Double | ||||||
|  |  | ||||||
| //typedef WilsonImpl<vComplex,  FundamentalRepresentation, CoeffRealHalfComms > WilsonImplRL;  // Real.. whichever prec | typedef WilsonImpl<vComplex,  FundamentalRepresentation, CoeffRealHalfComms > WilsonImplRL;  // Real.. whichever prec | ||||||
| //typedef WilsonImpl<vComplexF, FundamentalRepresentation, CoeffRealHalfComms > WilsonImplFH;  // Float | typedef WilsonImpl<vComplexF, FundamentalRepresentation, CoeffRealHalfComms > WilsonImplFH;  // Float | ||||||
| //typedef WilsonImpl<vComplexD, FundamentalRepresentation, CoeffRealHalfComms > WilsonImplDF;  // Double | typedef WilsonImpl<vComplexD, FundamentalRepresentation, CoeffRealHalfComms > WilsonImplDF;  // Double | ||||||
|  |  | ||||||
| typedef WilsonImpl<vComplex,  FundamentalRepresentation, CoeffComplex > ZWilsonImplR; // Real.. whichever prec | typedef WilsonImpl<vComplex,  FundamentalRepresentation, CoeffComplex > ZWilsonImplR; // Real.. whichever prec | ||||||
| typedef WilsonImpl<vComplexF, FundamentalRepresentation, CoeffComplex > ZWilsonImplF; // Float | typedef WilsonImpl<vComplexF, FundamentalRepresentation, CoeffComplex > ZWilsonImplF; // Float | ||||||
| typedef WilsonImpl<vComplexD, FundamentalRepresentation, CoeffComplex > ZWilsonImplD; // Double | typedef WilsonImpl<vComplexD, FundamentalRepresentation, CoeffComplex > ZWilsonImplD; // Double | ||||||
|  |  | ||||||
| //typedef WilsonImpl<vComplex,  FundamentalRepresentation, CoeffComplexHalfComms > ZWilsonImplRL; // Real.. whichever prec | typedef WilsonImpl<vComplex,  FundamentalRepresentation, CoeffComplexHalfComms > ZWilsonImplRL; // Real.. whichever prec | ||||||
| //typedef WilsonImpl<vComplexF, FundamentalRepresentation, CoeffComplexHalfComms > ZWilsonImplFH; // Float | typedef WilsonImpl<vComplexF, FundamentalRepresentation, CoeffComplexHalfComms > ZWilsonImplFH; // Float | ||||||
| //typedef WilsonImpl<vComplexD, FundamentalRepresentation, CoeffComplexHalfComms > ZWilsonImplDF; // Double | typedef WilsonImpl<vComplexD, FundamentalRepresentation, CoeffComplexHalfComms > ZWilsonImplDF; // Double | ||||||
|   |   | ||||||
| typedef WilsonImpl<vComplex,  AdjointRepresentation, CoeffReal > WilsonAdjImplR;   // Real.. whichever prec | typedef WilsonImpl<vComplex,  AdjointRepresentation, CoeffReal > WilsonAdjImplR;   // Real.. whichever prec | ||||||
| typedef WilsonImpl<vComplexF, AdjointRepresentation, CoeffReal > WilsonAdjImplF;  // Float | typedef WilsonImpl<vComplexF, AdjointRepresentation, CoeffReal > WilsonAdjImplF;  // Float | ||||||
|   | |||||||
| @@ -49,16 +49,8 @@ public: | |||||||
|  |  | ||||||
|   INHERIT_IMPL_TYPES(Impl); |   INHERIT_IMPL_TYPES(Impl); | ||||||
|   typedef FermionOperator<Impl> Base; |   typedef FermionOperator<Impl> Base; | ||||||
|   typedef AcceleratorVector<int,STENCIL_MAX> StencilVector;    |  | ||||||
| public: |  | ||||||
|     |     | ||||||
| #ifdef GRID_SYCL | public: | ||||||
| #define SYCL_HACK |  | ||||||
| #endif   |  | ||||||
| #ifdef SYCL_HACK |  | ||||||
|   static void HandDhopSiteSycl(StencilVector st_perm,StencilEntry *st_p, SiteDoubledGaugeField *U,SiteHalfSpinor  *buf, |  | ||||||
| 			       int ss,int sU,const SiteSpinor *in, SiteSpinor *out); |  | ||||||
| #endif |  | ||||||
|  |  | ||||||
|   static void DhopKernel(int Opt,StencilImpl &st,  DoubledGaugeField &U, SiteHalfSpinor * buf, |   static void DhopKernel(int Opt,StencilImpl &st,  DoubledGaugeField &U, SiteHalfSpinor * buf, | ||||||
| 			 int Ls, int Nsite, const FermionField &in, FermionField &out, | 			 int Ls, int Nsite, const FermionField &in, FermionField &out, | ||||||
|   | |||||||
| @@ -642,7 +642,7 @@ void CayleyFermion5D<Impl>::ContractConservedCurrent( PropagatorField &q_in_1, | |||||||
| 						      Current curr_type, | 						      Current curr_type, | ||||||
| 						      unsigned int mu) | 						      unsigned int mu) | ||||||
| { | { | ||||||
| #if (!defined(GRID_HIP)) | #if (!defined(GRID_CUDA)) && (!defined(GRID_HIP)) | ||||||
|   Gamma::Algebra Gmu [] = { |   Gamma::Algebra Gmu [] = { | ||||||
|     Gamma::Algebra::GammaX, |     Gamma::Algebra::GammaX, | ||||||
|     Gamma::Algebra::GammaY, |     Gamma::Algebra::GammaY, | ||||||
| @@ -826,7 +826,7 @@ void CayleyFermion5D<Impl>::SeqConservedCurrent(PropagatorField &q_in, | |||||||
|   } |   } | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
| #if (!defined(GRID_HIP)) | #if (!defined(GRID_CUDA)) && (!defined(GRID_HIP)) | ||||||
|   int tshift = (mu == Nd-1) ? 1 : 0; |   int tshift = (mu == Nd-1) ? 1 : 0; | ||||||
|   //////////////////////////////////////////////// |   //////////////////////////////////////////////// | ||||||
|   // GENERAL CAYLEY CASE |   // GENERAL CAYLEY CASE | ||||||
| @@ -880,29 +880,17 @@ void CayleyFermion5D<Impl>::SeqConservedCurrent(PropagatorField &q_in, | |||||||
|   } |   } | ||||||
|  |  | ||||||
|   std::vector<RealD> G_s(Ls,1.0); |   std::vector<RealD> G_s(Ls,1.0); | ||||||
|   RealD sign = 1; // sign flip for vector/tadpole |  | ||||||
|   if ( curr_type == Current::Axial ) { |   if ( curr_type == Current::Axial ) { | ||||||
|     for(int s=0;s<Ls/2;s++){ |     for(int s=0;s<Ls/2;s++){ | ||||||
|       G_s[s] = -1.0; |       G_s[s] = -1.0; | ||||||
|     } |     } | ||||||
|   } |   } | ||||||
|   else if ( curr_type == Current::Tadpole ) { |  | ||||||
|     auto b=this->_b; |  | ||||||
|     auto c=this->_c; |  | ||||||
|     if ( b == 1 && c == 0 ) { |  | ||||||
|       sign = -1;     |  | ||||||
|     } |  | ||||||
|     else { |  | ||||||
|       std::cerr << "Error: Tadpole implementation currently unavailable for non-Shamir actions." << std::endl; |  | ||||||
|       assert(b==1 && c==0); |  | ||||||
|     } |  | ||||||
|   } |  | ||||||
|  |  | ||||||
|   for(int s=0;s<Ls;s++){ |   for(int s=0;s<Ls;s++){ | ||||||
|  |  | ||||||
|     int sp = (s+1)%Ls; |     int sp = (s+1)%Ls; | ||||||
|     //    int sr = Ls-1-s; |     int sr = Ls-1-s; | ||||||
|     //    int srp= (sr+1)%Ls; |     int srp= (sr+1)%Ls; | ||||||
|  |  | ||||||
|     // Mobius parameters |     // Mobius parameters | ||||||
|     auto b=this->bs[s]; |     auto b=this->bs[s]; | ||||||
| @@ -919,7 +907,7 @@ void CayleyFermion5D<Impl>::SeqConservedCurrent(PropagatorField &q_in, | |||||||
|  |  | ||||||
|     tmp    = Cshift(tmp,mu,1); |     tmp    = Cshift(tmp,mu,1); | ||||||
|     Impl::multLinkField(Utmp,this->Umu,tmp,mu); |     Impl::multLinkField(Utmp,this->Umu,tmp,mu); | ||||||
|     tmp    = sign*G_s[s]*( Utmp*ph - gmu*Utmp*ph ); // Forward hop |     tmp    = G_s[s]*( Utmp*ph - gmu*Utmp*ph ); // Forward hop | ||||||
|     tmp    = where((lcoor>=tmin),tmp,zz); // Mask the time  |     tmp    = where((lcoor>=tmin),tmp,zz); // Mask the time  | ||||||
|     L_Q    = where((lcoor<=tmax),tmp,zz); // Position of current complicated |     L_Q    = where((lcoor<=tmax),tmp,zz); // Position of current complicated | ||||||
|  |  | ||||||
|   | |||||||
| @@ -680,8 +680,7 @@ void StaggeredKernels<Impl>::DhopSiteAsm(StencilView &st, | |||||||
|   gauge2 =(uint64_t)&UU[sU]( Z );				\ |   gauge2 =(uint64_t)&UU[sU]( Z );				\ | ||||||
|   gauge3 =(uint64_t)&UU[sU]( T );  |   gauge3 =(uint64_t)&UU[sU]( T );  | ||||||
|    |    | ||||||
| #undef STAG_VEC5D |  | ||||||
| #ifdef STAG_VEC5D |  | ||||||
|   // This is the single precision 5th direction vectorised kernel |   // This is the single precision 5th direction vectorised kernel | ||||||
| #include <Grid/simd/Intel512single.h> | #include <Grid/simd/Intel512single.h> | ||||||
| template <> void StaggeredKernels<StaggeredVec5dImplF>::DhopSiteAsm(StencilView &st, | template <> void StaggeredKernels<StaggeredVec5dImplF>::DhopSiteAsm(StencilView &st, | ||||||
| @@ -791,7 +790,7 @@ template <> void StaggeredKernels<StaggeredVec5dImplD>::DhopSiteAsm(StencilView | |||||||
| #endif | #endif | ||||||
| } | } | ||||||
|     |     | ||||||
| #endif    |     | ||||||
|  |  | ||||||
|  |  | ||||||
| #define PERMUTE_DIR3 __asm__ (	\ | #define PERMUTE_DIR3 __asm__ (	\ | ||||||
|   | |||||||
| @@ -32,50 +32,25 @@ Author: paboyle <paboyle@ph.ed.ac.uk> | |||||||
|  |  | ||||||
| NAMESPACE_BEGIN(Grid); | NAMESPACE_BEGIN(Grid); | ||||||
|  |  | ||||||
| #ifdef GRID_SIMT | #define LOAD_CHI(b)		\ | ||||||
|  |  | ||||||
| #define LOAD_CHI(ptype,b)			\ |  | ||||||
|   const SiteSpinor & ref (b[offset]);				\ |  | ||||||
|   Chi_0=coalescedReadPermute<ptype>(ref()()(0),perm,lane);	\ |  | ||||||
|   Chi_1=coalescedReadPermute<ptype>(ref()()(1),perm,lane);	\ |  | ||||||
|   Chi_2=coalescedReadPermute<ptype>(ref()()(2),perm,lane); |  | ||||||
|  |  | ||||||
| #define LOAD_CHI_COMMS(b)		\ |  | ||||||
|   const SiteSpinor & ref (b[offset]);	\ |   const SiteSpinor & ref (b[offset]);	\ | ||||||
|   Chi_0=coalescedRead(ref()()(0),lane);	\ |     Chi_0=ref()()(0);\ | ||||||
|   Chi_1=coalescedRead(ref()()(1),lane);	\ |     Chi_1=ref()()(1);\ | ||||||
|   Chi_2=coalescedRead(ref()()(2),lane); |     Chi_2=ref()()(2); | ||||||
|  |  | ||||||
| #define PERMUTE_DIR(dir)	; |  | ||||||
| #else |  | ||||||
| #define LOAD_CHI(ptype,b)      LOAD_CHI_COMMS(b) |  | ||||||
|  |  | ||||||
| #define LOAD_CHI_COMMS(b)		\ |  | ||||||
|   const SiteSpinor & ref (b[offset]);	\ |  | ||||||
|   Chi_0=ref()()(0);			\ |  | ||||||
|   Chi_1=ref()()(1);			\ |  | ||||||
|   Chi_2=ref()()(2); |  | ||||||
|  |  | ||||||
| #define PERMUTE_DIR(dir)			\ |  | ||||||
|   permute##dir(Chi_0,Chi_0);			\ |  | ||||||
|   permute##dir(Chi_1,Chi_1);			\ |  | ||||||
|   permute##dir(Chi_2,Chi_2); |  | ||||||
|  |  | ||||||
| #endif |  | ||||||
|  |  | ||||||
|  |  | ||||||
| // To splat or not to splat depends on the implementation | // To splat or not to splat depends on the implementation | ||||||
| #define MULT(A,UChi)				\ | #define MULT(A,UChi)				\ | ||||||
|   auto & ref(U[sU](A));			\ |   auto & ref(U[sU](A));			\ | ||||||
|     U_00=coalescedRead(ref()(0,0),lane);				\ |    Impl::loadLinkElement(U_00,ref()(0,0));      \ | ||||||
|     U_10=coalescedRead(ref()(1,0),lane);				\ |    Impl::loadLinkElement(U_10,ref()(1,0));      \ | ||||||
|     U_20=coalescedRead(ref()(2,0),lane);				\ |    Impl::loadLinkElement(U_20,ref()(2,0));      \ | ||||||
|     U_01=coalescedRead(ref()(0,1),lane);				\ |    Impl::loadLinkElement(U_01,ref()(0,1));      \ | ||||||
|     U_11=coalescedRead(ref()(1,1),lane);				\ |    Impl::loadLinkElement(U_11,ref()(1,1));      \ | ||||||
|     U_21=coalescedRead(ref()(2,1),lane);				\ |    Impl::loadLinkElement(U_21,ref()(2,1));      \ | ||||||
|     U_02=coalescedRead(ref()(0,2),lane);				\ |    Impl::loadLinkElement(U_02,ref()(0,2));     \ | ||||||
|     U_12=coalescedRead(ref()(1,2),lane);				\ |    Impl::loadLinkElement(U_12,ref()(1,2));     \ | ||||||
|     U_22=coalescedRead(ref()(2,2),lane);				\ |    Impl::loadLinkElement(U_22,ref()(2,2));     \ | ||||||
|     UChi ## _0  = U_00*Chi_0;	       \ |     UChi ## _0  = U_00*Chi_0;	       \ | ||||||
|     UChi ## _1  = U_10*Chi_0;\ |     UChi ## _1  = U_10*Chi_0;\ | ||||||
|     UChi ## _2  = U_20*Chi_0;\ |     UChi ## _2  = U_20*Chi_0;\ | ||||||
| @@ -88,15 +63,15 @@ NAMESPACE_BEGIN(Grid); | |||||||
|  |  | ||||||
| #define MULT_ADD(U,A,UChi)			\ | #define MULT_ADD(U,A,UChi)			\ | ||||||
|   auto & ref(U[sU](A));			\ |   auto & ref(U[sU](A));			\ | ||||||
|     U_00=coalescedRead(ref()(0,0),lane);				\ |    Impl::loadLinkElement(U_00,ref()(0,0));      \ | ||||||
|     U_10=coalescedRead(ref()(1,0),lane);				\ |    Impl::loadLinkElement(U_10,ref()(1,0));      \ | ||||||
|     U_20=coalescedRead(ref()(2,0),lane);				\ |    Impl::loadLinkElement(U_20,ref()(2,0));      \ | ||||||
|     U_01=coalescedRead(ref()(0,1),lane);				\ |    Impl::loadLinkElement(U_01,ref()(0,1));      \ | ||||||
|     U_11=coalescedRead(ref()(1,1),lane);				\ |    Impl::loadLinkElement(U_11,ref()(1,1));      \ | ||||||
|     U_21=coalescedRead(ref()(2,1),lane);				\ |    Impl::loadLinkElement(U_21,ref()(2,1));      \ | ||||||
|     U_02=coalescedRead(ref()(0,2),lane);				\ |    Impl::loadLinkElement(U_02,ref()(0,2));     \ | ||||||
|     U_12=coalescedRead(ref()(1,2),lane);				\ |    Impl::loadLinkElement(U_12,ref()(1,2));     \ | ||||||
|     U_22=coalescedRead(ref()(2,2),lane);				\ |    Impl::loadLinkElement(U_22,ref()(2,2));     \ | ||||||
|     UChi ## _0 += U_00*Chi_0;	       \ |     UChi ## _0 += U_00*Chi_0;	       \ | ||||||
|     UChi ## _1 += U_10*Chi_0;\ |     UChi ## _1 += U_10*Chi_0;\ | ||||||
|     UChi ## _2 += U_20*Chi_0;\ |     UChi ## _2 += U_20*Chi_0;\ | ||||||
| @@ -108,18 +83,24 @@ NAMESPACE_BEGIN(Grid); | |||||||
|     UChi ## _2 += U_22*Chi_2; |     UChi ## _2 += U_22*Chi_2; | ||||||
|  |  | ||||||
|  |  | ||||||
|  | #define PERMUTE_DIR(dir)			\ | ||||||
|  |   permute##dir(Chi_0,Chi_0);			\ | ||||||
|  |   permute##dir(Chi_1,Chi_1);			\ | ||||||
|  |   permute##dir(Chi_2,Chi_2); | ||||||
|  |  | ||||||
|  |  | ||||||
| #define HAND_STENCIL_LEG_BASE(Dir,Perm,skew)	\ | #define HAND_STENCIL_LEG_BASE(Dir,Perm,skew)	\ | ||||||
|   SE=st.GetEntry(ptype,Dir+skew,sF);	\ |   SE=st.GetEntry(ptype,Dir+skew,sF);	\ | ||||||
|   offset = SE->_offset;			\ |   offset = SE->_offset;			\ | ||||||
|   local  = SE->_is_local;		\ |   local  = SE->_is_local;		\ | ||||||
|   perm   = SE->_permute;		\ |   perm   = SE->_permute;		\ | ||||||
|   if ( local ) {						\ |   if ( local ) {						\ | ||||||
|     LOAD_CHI(Perm,in);						\ |     LOAD_CHI(in);					\ | ||||||
|     if ( perm) {						\ |     if ( perm) {						\ | ||||||
|       PERMUTE_DIR(Perm);					\ |       PERMUTE_DIR(Perm);					\ | ||||||
|     }								\ |     }								\ | ||||||
|   } else {							\ |   } else {							\ | ||||||
|     LOAD_CHI_COMMS(buf);					\ |     LOAD_CHI(buf);						\ | ||||||
|   }								 |   }								 | ||||||
|  |  | ||||||
| #define HAND_STENCIL_LEG_BEGIN(Dir,Perm,skew,even)		\ | #define HAND_STENCIL_LEG_BEGIN(Dir,Perm,skew,even)		\ | ||||||
| @@ -135,18 +116,19 @@ NAMESPACE_BEGIN(Grid); | |||||||
|   } |   } | ||||||
|  |  | ||||||
|  |  | ||||||
|  |  | ||||||
| #define HAND_STENCIL_LEG_INT(U,Dir,Perm,skew,even)	\ | #define HAND_STENCIL_LEG_INT(U,Dir,Perm,skew,even)	\ | ||||||
|   SE=st.GetEntry(ptype,Dir+skew,sF);			\ |   SE=st.GetEntry(ptype,Dir+skew,sF);			\ | ||||||
|   offset = SE->_offset;					\ |   offset = SE->_offset;					\ | ||||||
|   local  = SE->_is_local;				\ |   local  = SE->_is_local;				\ | ||||||
|   perm   = SE->_permute;				\ |   perm   = SE->_permute;				\ | ||||||
|   if ( local ) {					\ |   if ( local ) {					\ | ||||||
|     LOAD_CHI(Perm,in);					\ |     LOAD_CHI(in);				\ | ||||||
|     if ( perm) {					\ |     if ( perm) {					\ | ||||||
|       PERMUTE_DIR(Perm);				\ |       PERMUTE_DIR(Perm);				\ | ||||||
|     }							\ |     }							\ | ||||||
|   } else if ( st.same_node[Dir] ) {			\ |   } else if ( st.same_node[Dir] ) {			\ | ||||||
|     LOAD_CHI_COMMS(buf);				\ |     LOAD_CHI(buf);					\ | ||||||
|   }							\ |   }							\ | ||||||
|   if (local || st.same_node[Dir] ) {		\ |   if (local || st.same_node[Dir] ) {		\ | ||||||
|     MULT_ADD(U,Dir,even);				\ |     MULT_ADD(U,Dir,even);				\ | ||||||
| @@ -158,32 +140,10 @@ NAMESPACE_BEGIN(Grid); | |||||||
|   local  = SE->_is_local;				\ |   local  = SE->_is_local;				\ | ||||||
|   if ((!local) && (!st.same_node[Dir]) ) {		\ |   if ((!local) && (!st.same_node[Dir]) ) {		\ | ||||||
|     nmu++;							\ |     nmu++;							\ | ||||||
|     { LOAD_CHI_COMMS(buf);	  }				\ |     { LOAD_CHI(buf);	  }					\ | ||||||
|     { MULT_ADD(U,Dir,even); }					\ |     { MULT_ADD(U,Dir,even); }					\ | ||||||
|   }								 |   }								 | ||||||
|  |  | ||||||
| #define HAND_DECLARATIONS(Simd) \ |  | ||||||
|   Simd even_0;			\ |  | ||||||
|   Simd even_1;			\ |  | ||||||
|   Simd even_2;			\ |  | ||||||
|   Simd odd_0;			\ |  | ||||||
|   Simd odd_1;			\ |  | ||||||
|   Simd odd_2;		        \ |  | ||||||
| 		      		\ |  | ||||||
|   Simd Chi_0;			\ |  | ||||||
|   Simd Chi_1;			\ |  | ||||||
|   Simd Chi_2;			\ |  | ||||||
| 				\ |  | ||||||
|   Simd U_00;			\ |  | ||||||
|   Simd U_10;			\ |  | ||||||
|   Simd U_20;			\ |  | ||||||
|   Simd U_01;			\ |  | ||||||
|   Simd U_11;			\ |  | ||||||
|   Simd U_21;			\ |  | ||||||
|   Simd U_02;			\ |  | ||||||
|   Simd U_12;			\ |  | ||||||
|   Simd U_22;			 |  | ||||||
|    |  | ||||||
|  |  | ||||||
| template <class Impl> | template <class Impl> | ||||||
| template <int Naik> accelerator_inline | template <int Naik> accelerator_inline | ||||||
| @@ -195,14 +155,28 @@ void StaggeredKernels<Impl>::DhopSiteHand(StencilView &st, | |||||||
|   typedef typename Simd::scalar_type S; |   typedef typename Simd::scalar_type S; | ||||||
|   typedef typename Simd::vector_type V; |   typedef typename Simd::vector_type V; | ||||||
|  |  | ||||||
|  |   Simd even_0; // 12 regs on knc | ||||||
|  |   Simd even_1; | ||||||
|  |   Simd even_2; | ||||||
|  |   Simd odd_0; // 12 regs on knc | ||||||
|  |   Simd odd_1; | ||||||
|  |   Simd odd_2; | ||||||
|  |  | ||||||
|   const int Nsimd = SiteHalfSpinor::Nsimd(); |   Simd Chi_0;    // two spinor; 6 regs | ||||||
|   const int lane=acceleratorSIMTlane(Nsimd); |   Simd Chi_1; | ||||||
|   typedef decltype( coalescedRead( in[0]()()(0) )) Simt; |   Simd Chi_2; | ||||||
|   HAND_DECLARATIONS(Simt); |  | ||||||
|    |    | ||||||
|   typedef decltype( coalescedRead( in[0] )) calcSiteSpinor; |   Simd U_00;  // two rows of U matrix | ||||||
|   calcSiteSpinor result; |   Simd U_10; | ||||||
|  |   Simd U_20;   | ||||||
|  |   Simd U_01; | ||||||
|  |   Simd U_11; | ||||||
|  |   Simd U_21;  // 2 reg left. | ||||||
|  |   Simd U_02; | ||||||
|  |   Simd U_12; | ||||||
|  |   Simd U_22;  | ||||||
|  |  | ||||||
|  |   SiteSpinor result; | ||||||
|   int offset,local,perm, ptype; |   int offset,local,perm, ptype; | ||||||
|  |  | ||||||
|   StencilEntry *SE; |   StencilEntry *SE; | ||||||
| @@ -241,7 +215,7 @@ void StaggeredKernels<Impl>::DhopSiteHand(StencilView &st, | |||||||
|       result()()(1) = even_1 + odd_1; |       result()()(1) = even_1 + odd_1; | ||||||
|       result()()(2) = even_2 + odd_2; |       result()()(2) = even_2 + odd_2; | ||||||
|     } |     } | ||||||
|     coalescedWrite(out[sF],result); |     vstream(out[sF],result); | ||||||
|   } |   } | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -256,13 +230,28 @@ void StaggeredKernels<Impl>::DhopSiteHandInt(StencilView &st, | |||||||
|   typedef typename Simd::scalar_type S; |   typedef typename Simd::scalar_type S; | ||||||
|   typedef typename Simd::vector_type V; |   typedef typename Simd::vector_type V; | ||||||
|  |  | ||||||
|   const int Nsimd = SiteHalfSpinor::Nsimd(); |   Simd even_0; // 12 regs on knc | ||||||
|   const int lane=acceleratorSIMTlane(Nsimd); |   Simd even_1; | ||||||
|   typedef decltype( coalescedRead( in[0]()()(0) )) Simt; |   Simd even_2; | ||||||
|   HAND_DECLARATIONS(Simt); |   Simd odd_0; // 12 regs on knc | ||||||
|  |   Simd odd_1; | ||||||
|  |   Simd odd_2; | ||||||
|  |  | ||||||
|   typedef decltype( coalescedRead( in[0] )) calcSiteSpinor; |   Simd Chi_0;    // two spinor; 6 regs | ||||||
|   calcSiteSpinor result; |   Simd Chi_1; | ||||||
|  |   Simd Chi_2; | ||||||
|  |    | ||||||
|  |   Simd U_00;  // two rows of U matrix | ||||||
|  |   Simd U_10; | ||||||
|  |   Simd U_20;   | ||||||
|  |   Simd U_01; | ||||||
|  |   Simd U_11; | ||||||
|  |   Simd U_21;  // 2 reg left. | ||||||
|  |   Simd U_02; | ||||||
|  |   Simd U_12; | ||||||
|  |   Simd U_22;  | ||||||
|  |  | ||||||
|  |   SiteSpinor result; | ||||||
|   int offset, ptype, local, perm; |   int offset, ptype, local, perm; | ||||||
|  |  | ||||||
|   StencilEntry *SE; |   StencilEntry *SE; | ||||||
| @@ -272,8 +261,8 @@ void StaggeredKernels<Impl>::DhopSiteHandInt(StencilView &st, | |||||||
|   //    int sF=s+LLs*sU; |   //    int sF=s+LLs*sU; | ||||||
|   { |   { | ||||||
|  |  | ||||||
|     zeroit(even_0);    zeroit(even_1);    zeroit(even_2); |     even_0 = Zero();    even_1 = Zero();    even_2 = Zero(); | ||||||
|     zeroit(odd_0);    zeroit(odd_1);    zeroit(odd_2); |      odd_0 = Zero();     odd_1 = Zero();     odd_2 = Zero(); | ||||||
|  |  | ||||||
|     skew = 0; |     skew = 0; | ||||||
|     HAND_STENCIL_LEG_INT(U,Xp,3,skew,even);   |     HAND_STENCIL_LEG_INT(U,Xp,3,skew,even);   | ||||||
| @@ -305,7 +294,7 @@ void StaggeredKernels<Impl>::DhopSiteHandInt(StencilView &st, | |||||||
|       result()()(1) = even_1 + odd_1; |       result()()(1) = even_1 + odd_1; | ||||||
|       result()()(2) = even_2 + odd_2; |       result()()(2) = even_2 + odd_2; | ||||||
|     } |     } | ||||||
|     coalescedWrite(out[sF],result); |     vstream(out[sF],result); | ||||||
|   } |   } | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -320,13 +309,28 @@ void StaggeredKernels<Impl>::DhopSiteHandExt(StencilView &st, | |||||||
|   typedef typename Simd::scalar_type S; |   typedef typename Simd::scalar_type S; | ||||||
|   typedef typename Simd::vector_type V; |   typedef typename Simd::vector_type V; | ||||||
|  |  | ||||||
|   const int Nsimd = SiteHalfSpinor::Nsimd(); |   Simd even_0; // 12 regs on knc | ||||||
|   const int lane=acceleratorSIMTlane(Nsimd); |   Simd even_1; | ||||||
|   typedef decltype( coalescedRead( in[0]()()(0) )) Simt; |   Simd even_2; | ||||||
|   HAND_DECLARATIONS(Simt); |   Simd odd_0; // 12 regs on knc | ||||||
|  |   Simd odd_1; | ||||||
|  |   Simd odd_2; | ||||||
|  |  | ||||||
|   typedef decltype( coalescedRead( in[0] )) calcSiteSpinor; |   Simd Chi_0;    // two spinor; 6 regs | ||||||
|   calcSiteSpinor result; |   Simd Chi_1; | ||||||
|  |   Simd Chi_2; | ||||||
|  |    | ||||||
|  |   Simd U_00;  // two rows of U matrix | ||||||
|  |   Simd U_10; | ||||||
|  |   Simd U_20;   | ||||||
|  |   Simd U_01; | ||||||
|  |   Simd U_11; | ||||||
|  |   Simd U_21;  // 2 reg left. | ||||||
|  |   Simd U_02; | ||||||
|  |   Simd U_12; | ||||||
|  |   Simd U_22;  | ||||||
|  |  | ||||||
|  |   SiteSpinor result; | ||||||
|   int offset, ptype, local; |   int offset, ptype, local; | ||||||
|  |  | ||||||
|   StencilEntry *SE; |   StencilEntry *SE; | ||||||
| @@ -336,8 +340,8 @@ void StaggeredKernels<Impl>::DhopSiteHandExt(StencilView &st, | |||||||
|   //    int sF=s+LLs*sU; |   //    int sF=s+LLs*sU; | ||||||
|   { |   { | ||||||
|  |  | ||||||
|     zeroit(even_0);    zeroit(even_1);    zeroit(even_2); |     even_0 = Zero();    even_1 = Zero();    even_2 = Zero(); | ||||||
|     zeroit(odd_0);    zeroit(odd_1);    zeroit(odd_2); |      odd_0 = Zero();     odd_1 = Zero();     odd_2 = Zero(); | ||||||
|     int nmu=0; |     int nmu=0; | ||||||
|     skew = 0; |     skew = 0; | ||||||
|     HAND_STENCIL_LEG_EXT(U,Xp,3,skew,even);   |     HAND_STENCIL_LEG_EXT(U,Xp,3,skew,even);   | ||||||
| @@ -370,7 +374,7 @@ void StaggeredKernels<Impl>::DhopSiteHandExt(StencilView &st, | |||||||
| 	result()()(1) = even_1 + odd_1; | 	result()()(1) = even_1 + odd_1; | ||||||
| 	result()()(2) = even_2 + odd_2; | 	result()()(2) = even_2 + odd_2; | ||||||
|       } |       } | ||||||
|       coalescedWrite(out[sF] , out(sF)+ result); |       out[sF] = out[sF] + result; | ||||||
|     } |     } | ||||||
|   } |   } | ||||||
| } | } | ||||||
| @@ -393,7 +397,6 @@ void StaggeredKernels<Impl>::DhopSiteHandExt(StencilView &st, | |||||||
| 						     const FermionFieldView &in, FermionFieldView &out, int dag); \ | 						     const FermionFieldView &in, FermionFieldView &out, int dag); \ | ||||||
| */ | */ | ||||||
| #undef LOAD_CHI | #undef LOAD_CHI | ||||||
| #undef HAND_DECLARATIONS |  | ||||||
|  |  | ||||||
| NAMESPACE_END(Grid); | NAMESPACE_END(Grid); | ||||||
|  |  | ||||||
|   | |||||||
| @@ -35,32 +35,39 @@ NAMESPACE_BEGIN(Grid); | |||||||
| #define GENERIC_STENCIL_LEG(U,Dir,skew,multLink)		\ | #define GENERIC_STENCIL_LEG(U,Dir,skew,multLink)		\ | ||||||
|   SE = st.GetEntry(ptype, Dir+skew, sF);			\ |   SE = st.GetEntry(ptype, Dir+skew, sF);			\ | ||||||
|   if (SE->_is_local ) {						\ |   if (SE->_is_local ) {						\ | ||||||
|     int perm= SE->_permute;						\ |     if (SE->_permute) {						\ | ||||||
|     chi = coalescedReadPermute(in[SE->_offset],ptype,perm,lane);\ |       chi_p = χ						\ | ||||||
|  |       permute(chi,  in[SE->_offset], ptype);			\ | ||||||
|  |     } else {							\ | ||||||
|  |       chi_p = &in[SE->_offset];					\ | ||||||
|  |     }								\ | ||||||
|   } else {							\ |   } else {							\ | ||||||
|     chi = coalescedRead(buf[SE->_offset],lane);			\ |     chi_p = &buf[SE->_offset];					\ | ||||||
|   }								\ |   }								\ | ||||||
|   acceleratorSynchronise();					\ |   multLink(Uchi, U[sU], *chi_p, Dir);			 | ||||||
|   multLink(Uchi, U[sU], chi, Dir);			 |  | ||||||
|  |  | ||||||
| #define GENERIC_STENCIL_LEG_INT(U,Dir,skew,multLink)		\ | #define GENERIC_STENCIL_LEG_INT(U,Dir,skew,multLink)		\ | ||||||
|   SE = st.GetEntry(ptype, Dir+skew, sF);			\ |   SE = st.GetEntry(ptype, Dir+skew, sF);			\ | ||||||
|   if (SE->_is_local ) {						\ |   if (SE->_is_local ) {						\ | ||||||
|     int perm= SE->_permute;						\ |     if (SE->_permute) {						\ | ||||||
|     chi = coalescedReadPermute(in[SE->_offset],ptype,perm,lane);\ |       chi_p = χ						\ | ||||||
|  |       permute(chi,  in[SE->_offset], ptype);			\ | ||||||
|  |     } else {							\ | ||||||
|  |       chi_p = &in[SE->_offset];					\ | ||||||
|  |     }								\ | ||||||
|   } else if ( st.same_node[Dir] ) {				\ |   } else if ( st.same_node[Dir] ) {				\ | ||||||
|     chi = coalescedRead(buf[SE->_offset],lane);                 \ |     chi_p = &buf[SE->_offset];					\ | ||||||
|   }								\ |   }								\ | ||||||
|   if (SE->_is_local || st.same_node[Dir] ) {			\ |   if (SE->_is_local || st.same_node[Dir] ) {			\ | ||||||
|     multLink(Uchi, U[sU], chi, Dir);				\ |     multLink(Uchi, U[sU], *chi_p, Dir);				\ | ||||||
|   } |   } | ||||||
|  |  | ||||||
| #define GENERIC_STENCIL_LEG_EXT(U,Dir,skew,multLink)		\ | #define GENERIC_STENCIL_LEG_EXT(U,Dir,skew,multLink)		\ | ||||||
|   SE = st.GetEntry(ptype, Dir+skew, sF);			\ |   SE = st.GetEntry(ptype, Dir+skew, sF);			\ | ||||||
|   if ((!SE->_is_local) && (!st.same_node[Dir]) ) {		\ |   if ((!SE->_is_local) && (!st.same_node[Dir]) ) {		\ | ||||||
|     nmu++;							\ |     nmu++;							\ | ||||||
|     chi = coalescedRead(buf[SE->_offset],lane);			\ |     chi_p = &buf[SE->_offset];					\ | ||||||
|     multLink(Uchi, U[sU], chi, Dir);				\ |     multLink(Uchi, U[sU], *chi_p, Dir);				\ | ||||||
|   } |   } | ||||||
|  |  | ||||||
| template <class Impl> | template <class Impl> | ||||||
| @@ -77,14 +84,12 @@ void StaggeredKernels<Impl>::DhopSiteGeneric(StencilView &st, | |||||||
| 					     SiteSpinor *buf, int sF, int sU,  | 					     SiteSpinor *buf, int sF, int sU,  | ||||||
| 					     const FermionFieldView &in, FermionFieldView &out, int dag)  | 					     const FermionFieldView &in, FermionFieldView &out, int dag)  | ||||||
| { | { | ||||||
|   typedef decltype(coalescedRead(in[0])) calcSpinor; |   const SiteSpinor *chi_p; | ||||||
|   calcSpinor chi; |   SiteSpinor chi; | ||||||
|   calcSpinor Uchi; |   SiteSpinor Uchi; | ||||||
|   StencilEntry *SE; |   StencilEntry *SE; | ||||||
|   int ptype; |   int ptype; | ||||||
|   int skew; |   int skew; | ||||||
|   const int Nsimd = SiteHalfSpinor::Nsimd(); |  | ||||||
|   const int lane=acceleratorSIMTlane(Nsimd); |  | ||||||
|  |  | ||||||
|   //  for(int s=0;s<LLs;s++){ |   //  for(int s=0;s<LLs;s++){ | ||||||
|   // |   // | ||||||
| @@ -113,7 +118,7 @@ void StaggeredKernels<Impl>::DhopSiteGeneric(StencilView &st, | |||||||
|     if ( dag ) {  |     if ( dag ) {  | ||||||
|       Uchi = - Uchi; |       Uchi = - Uchi; | ||||||
|     }  |     }  | ||||||
|     coalescedWrite(out[sF], Uchi,lane); |     vstream(out[sF], Uchi); | ||||||
|   } |   } | ||||||
| }; | }; | ||||||
|  |  | ||||||
| @@ -125,16 +130,13 @@ template <int Naik> accelerator_inline | |||||||
| void StaggeredKernels<Impl>::DhopSiteGenericInt(StencilView &st,  | void StaggeredKernels<Impl>::DhopSiteGenericInt(StencilView &st,  | ||||||
| 						DoubledGaugeFieldView &U, DoubledGaugeFieldView &UUU, | 						DoubledGaugeFieldView &U, DoubledGaugeFieldView &UUU, | ||||||
| 						SiteSpinor *buf, int sF, int sU,  | 						SiteSpinor *buf, int sF, int sU,  | ||||||
| 						const FermionFieldView &in, FermionFieldView &out,int dag) | 						const FermionFieldView &in, FermionFieldView &out,int dag) { | ||||||
| { |   const SiteSpinor *chi_p; | ||||||
|   typedef decltype(coalescedRead(in[0])) calcSpinor; |   SiteSpinor chi; | ||||||
|   calcSpinor chi; |   SiteSpinor Uchi; | ||||||
|   calcSpinor Uchi; |  | ||||||
|   StencilEntry *SE; |   StencilEntry *SE; | ||||||
|   int ptype; |   int ptype; | ||||||
|   int skew ; |   int skew ; | ||||||
|   const int Nsimd = SiteHalfSpinor::Nsimd(); |  | ||||||
|   const int lane=acceleratorSIMTlane(Nsimd); |  | ||||||
|  |  | ||||||
|   //  for(int s=0;s<LLs;s++){ |   //  for(int s=0;s<LLs;s++){ | ||||||
|   //    int sF=LLs*sU+s; |   //    int sF=LLs*sU+s; | ||||||
| @@ -163,7 +165,7 @@ void StaggeredKernels<Impl>::DhopSiteGenericInt(StencilView &st, | |||||||
|     if ( dag ) { |     if ( dag ) { | ||||||
|       Uchi = - Uchi; |       Uchi = - Uchi; | ||||||
|     } |     } | ||||||
|     coalescedWrite(out[sF], Uchi,lane); |     vstream(out[sF], Uchi); | ||||||
|   } |   } | ||||||
| }; | }; | ||||||
|  |  | ||||||
| @@ -176,17 +178,14 @@ template <int Naik> accelerator_inline | |||||||
| void StaggeredKernels<Impl>::DhopSiteGenericExt(StencilView &st,  | void StaggeredKernels<Impl>::DhopSiteGenericExt(StencilView &st,  | ||||||
| 						DoubledGaugeFieldView &U, DoubledGaugeFieldView &UUU, | 						DoubledGaugeFieldView &U, DoubledGaugeFieldView &UUU, | ||||||
| 						SiteSpinor *buf, int sF, int sU, | 						SiteSpinor *buf, int sF, int sU, | ||||||
| 						const FermionFieldView &in, FermionFieldView &out,int dag) | 						const FermionFieldView &in, FermionFieldView &out,int dag) { | ||||||
| { |   const SiteSpinor *chi_p; | ||||||
|   typedef decltype(coalescedRead(in[0])) calcSpinor; |   //  SiteSpinor chi; | ||||||
|   calcSpinor chi; |   SiteSpinor Uchi; | ||||||
|   calcSpinor Uchi; |  | ||||||
|   StencilEntry *SE; |   StencilEntry *SE; | ||||||
|   int ptype; |   int ptype; | ||||||
|   int nmu=0; |   int nmu=0; | ||||||
|   int skew ; |   int skew ; | ||||||
|   const int Nsimd = SiteHalfSpinor::Nsimd(); |  | ||||||
|   const int lane=acceleratorSIMTlane(Nsimd); |  | ||||||
|  |  | ||||||
|   //  for(int s=0;s<LLs;s++){ |   //  for(int s=0;s<LLs;s++){ | ||||||
|   //    int sF=LLs*sU+s; |   //    int sF=LLs*sU+s; | ||||||
| @@ -213,11 +212,10 @@ void StaggeredKernels<Impl>::DhopSiteGenericExt(StencilView &st, | |||||||
|     GENERIC_STENCIL_LEG_EXT(UUU,Tm,skew,Impl::multLinkAdd); |     GENERIC_STENCIL_LEG_EXT(UUU,Tm,skew,Impl::multLinkAdd); | ||||||
|     } |     } | ||||||
|     if ( nmu ) {  |     if ( nmu ) {  | ||||||
|       auto _out = coalescedRead(out[sF],lane); |  | ||||||
|       if ( dag ) {  |       if ( dag ) {  | ||||||
| 	coalescedWrite(out[sF], _out-Uchi,lane); | 	out[sF] = out[sF] - Uchi; | ||||||
|       } else {  |       } else {  | ||||||
| 	coalescedWrite(out[sF], _out+Uchi,lane); | 	out[sF] = out[sF] + Uchi; | ||||||
|       } |       } | ||||||
|     } |     } | ||||||
|   } |   } | ||||||
| @@ -263,8 +261,6 @@ void StaggeredKernels<Impl>::DhopImproved(StencilImpl &st, LebesgueOrder &lo, | |||||||
|   GridBase *FGrid=in.Grid();   |   GridBase *FGrid=in.Grid();   | ||||||
|   GridBase *UGrid=U.Grid();   |   GridBase *UGrid=U.Grid();   | ||||||
|   typedef StaggeredKernels<Impl> ThisKernel; |   typedef StaggeredKernels<Impl> ThisKernel; | ||||||
|   const int Nsimd = SiteHalfSpinor::Nsimd(); |  | ||||||
|   const int lane=acceleratorSIMTlane(Nsimd); |  | ||||||
|   autoView( UUU_v , UUU, AcceleratorRead); |   autoView( UUU_v , UUU, AcceleratorRead); | ||||||
|   autoView( U_v   ,   U, AcceleratorRead); |   autoView( U_v   ,   U, AcceleratorRead); | ||||||
|   autoView( in_v  ,  in, AcceleratorRead); |   autoView( in_v  ,  in, AcceleratorRead); | ||||||
| @@ -305,8 +301,6 @@ void StaggeredKernels<Impl>::DhopNaive(StencilImpl &st, LebesgueOrder &lo, | |||||||
|   GridBase *FGrid=in.Grid();   |   GridBase *FGrid=in.Grid();   | ||||||
|   GridBase *UGrid=U.Grid();   |   GridBase *UGrid=U.Grid();   | ||||||
|   typedef StaggeredKernels<Impl> ThisKernel; |   typedef StaggeredKernels<Impl> ThisKernel; | ||||||
|   const int Nsimd = SiteHalfSpinor::Nsimd(); |  | ||||||
|   const int lane=acceleratorSIMTlane(Nsimd); |  | ||||||
|   autoView( UUU_v ,   U, AcceleratorRead); |   autoView( UUU_v ,   U, AcceleratorRead); | ||||||
|   autoView( U_v   ,   U, AcceleratorRead); |   autoView( U_v   ,   U, AcceleratorRead); | ||||||
|   autoView( in_v  ,  in, AcceleratorRead); |   autoView( in_v  ,  in, AcceleratorRead); | ||||||
|   | |||||||
| @@ -92,16 +92,20 @@ void WilsonCloverFermion<Impl>::ImportGauge(const GaugeField &_Umu) | |||||||
|   int lvol = _Umu.Grid()->lSites(); |   int lvol = _Umu.Grid()->lSites(); | ||||||
|   int DimRep = Impl::Dimension; |   int DimRep = Impl::Dimension; | ||||||
|  |  | ||||||
|  |   Eigen::MatrixXcd EigenCloverOp = Eigen::MatrixXcd::Zero(Ns * DimRep, Ns * DimRep); | ||||||
|  |   Eigen::MatrixXcd EigenInvCloverOp = Eigen::MatrixXcd::Zero(Ns * DimRep, Ns * DimRep); | ||||||
|  |  | ||||||
|  |   Coordinate lcoor; | ||||||
|  |   typename SiteCloverType::scalar_object Qx = Zero(), Qxinv = Zero(); | ||||||
|  |  | ||||||
|   { |   { | ||||||
|     autoView(CTv,CloverTerm,CpuRead); |     autoView(CTv,CloverTerm,CpuRead); | ||||||
|     autoView(CTIv,CloverTermInv,CpuWrite); |     autoView(CTIv,CloverTermInv,CpuWrite); | ||||||
|     thread_for(site, lvol, { |     for (int site = 0; site < lvol; site++) { | ||||||
|       Coordinate lcoor; |  | ||||||
|       grid->LocalIndexToLocalCoor(site, lcoor); |       grid->LocalIndexToLocalCoor(site, lcoor); | ||||||
|       Eigen::MatrixXcd EigenCloverOp = Eigen::MatrixXcd::Zero(Ns * DimRep, Ns * DimRep); |       EigenCloverOp = Eigen::MatrixXcd::Zero(Ns * DimRep, Ns * DimRep); | ||||||
|       Eigen::MatrixXcd EigenInvCloverOp = Eigen::MatrixXcd::Zero(Ns * DimRep, Ns * DimRep); |  | ||||||
|       typename SiteCloverType::scalar_object Qx = Zero(), Qxinv = Zero(); |  | ||||||
|       peekLocalSite(Qx, CTv, lcoor); |       peekLocalSite(Qx, CTv, lcoor); | ||||||
|  |       Qxinv = Zero(); | ||||||
|       //if (csw!=0){ |       //if (csw!=0){ | ||||||
|       for (int j = 0; j < Ns; j++) |       for (int j = 0; j < Ns; j++) | ||||||
| 	for (int k = 0; k < Ns; k++) | 	for (int k = 0; k < Ns; k++) | ||||||
| @@ -122,7 +126,7 @@ void WilsonCloverFermion<Impl>::ImportGauge(const GaugeField &_Umu) | |||||||
|       //    if (site==0) std::cout << "site =" << site << "\n" << EigenInvCloverOp << std::endl; |       //    if (site==0) std::cout << "site =" << site << "\n" << EigenInvCloverOp << std::endl; | ||||||
|       //  } |       //  } | ||||||
|       pokeLocalSite(Qxinv, CTIv, lcoor); |       pokeLocalSite(Qxinv, CTIv, lcoor); | ||||||
|     }); |     } | ||||||
|   } |   } | ||||||
|  |  | ||||||
|   // Separate the even and odd parts |   // Separate the even and odd parts | ||||||
|   | |||||||
| @@ -397,7 +397,6 @@ void WilsonFermion<Impl>::DhopDerivEO(GaugeField &mat, const FermionField &U, co | |||||||
| template <class Impl> | template <class Impl> | ||||||
| void WilsonFermion<Impl>::Dhop(const FermionField &in, FermionField &out, int dag) | void WilsonFermion<Impl>::Dhop(const FermionField &in, FermionField &out, int dag) | ||||||
| { | { | ||||||
|   DhopCalls+=2; |  | ||||||
|   conformable(in.Grid(), _grid);  // verifies full grid |   conformable(in.Grid(), _grid);  // verifies full grid | ||||||
|   conformable(in.Grid(), out.Grid()); |   conformable(in.Grid(), out.Grid()); | ||||||
|  |  | ||||||
| @@ -409,7 +408,6 @@ void WilsonFermion<Impl>::Dhop(const FermionField &in, FermionField &out, int da | |||||||
| template <class Impl> | template <class Impl> | ||||||
| void WilsonFermion<Impl>::DhopOE(const FermionField &in, FermionField &out, int dag) | void WilsonFermion<Impl>::DhopOE(const FermionField &in, FermionField &out, int dag) | ||||||
| { | { | ||||||
|   DhopCalls++; |  | ||||||
|   conformable(in.Grid(), _cbgrid);    // verifies half grid |   conformable(in.Grid(), _cbgrid);    // verifies half grid | ||||||
|   conformable(in.Grid(), out.Grid());  // drops the cb check |   conformable(in.Grid(), out.Grid());  // drops the cb check | ||||||
|  |  | ||||||
| @@ -422,7 +420,6 @@ void WilsonFermion<Impl>::DhopOE(const FermionField &in, FermionField &out, int | |||||||
| template <class Impl> | template <class Impl> | ||||||
| void WilsonFermion<Impl>::DhopEO(const FermionField &in, FermionField &out,int dag) | void WilsonFermion<Impl>::DhopEO(const FermionField &in, FermionField &out,int dag) | ||||||
| { | { | ||||||
|   DhopCalls++; |  | ||||||
|   conformable(in.Grid(), _cbgrid);    // verifies half grid |   conformable(in.Grid(), _cbgrid);    // verifies half grid | ||||||
|   conformable(in.Grid(), out.Grid());  // drops the cb check |   conformable(in.Grid(), out.Grid());  // drops the cb check | ||||||
|  |  | ||||||
|   | |||||||
| @@ -38,6 +38,9 @@ Author: Nils Meyer  <nils.meyer@ur.de>  Regensburg University | |||||||
| // undefine everything related to kernels | // undefine everything related to kernels | ||||||
| #include <simd/Fujitsu_A64FX_undef.h> | #include <simd/Fujitsu_A64FX_undef.h> | ||||||
|  |  | ||||||
|  | // enable A64FX body | ||||||
|  | #define WILSONKERNELSASMBODYA64FX | ||||||
|  | //#pragma message("A64FX Dslash: WilsonKernelsAsmBodyA64FX.h") | ||||||
|  |  | ||||||
|     /////////////////////////////////////////////////////////// |     /////////////////////////////////////////////////////////// | ||||||
|     // If we are A64FX specialise the single precision routine |     // If we are A64FX specialise the single precision routine | ||||||
| @@ -60,89 +63,119 @@ Author: Nils Meyer  <nils.meyer@ur.de>  Regensburg University | |||||||
| #define INTERIOR_AND_EXTERIOR | #define INTERIOR_AND_EXTERIOR | ||||||
| #undef INTERIOR | #undef INTERIOR | ||||||
| #undef EXTERIOR | #undef EXTERIOR | ||||||
|  |  | ||||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") |  | ||||||
| template<> void | template<> void | ||||||
| WilsonKernels<WilsonImplF>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<WilsonImplF>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") |  | ||||||
| template<> void | template<> void | ||||||
| WilsonKernels<ZWilsonImplF>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<ZWilsonImplF>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
| //#pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | template<> void | ||||||
| //template<> void | WilsonKernels<WilsonImplFH>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //WilsonKernels<WilsonImplFH>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
| //#pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| //template<> void | #endif | ||||||
| //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> |  | ||||||
|  |  | ||||||
|  | template<> void | ||||||
|  | WilsonKernels<ZWilsonImplFH>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
|  | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
| #undef INTERIOR_AND_EXTERIOR | #undef INTERIOR_AND_EXTERIOR | ||||||
| #define INTERIOR | #define INTERIOR | ||||||
| #undef EXTERIOR | #undef EXTERIOR | ||||||
|  |  | ||||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") |  | ||||||
| template<> void | template<> void | ||||||
| WilsonKernels<WilsonImplF>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<WilsonImplF>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") |  | ||||||
| template<> void | template<> void | ||||||
| WilsonKernels<ZWilsonImplF>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<ZWilsonImplF>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
| //#pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | template<> void | ||||||
| //template<> void | WilsonKernels<WilsonImplFH>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //WilsonKernels<WilsonImplFH>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
| //#pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| //template<> void | #endif | ||||||
| //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> |  | ||||||
|  |  | ||||||
|  | template<> void | ||||||
|  | WilsonKernels<ZWilsonImplFH>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
|  | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
| #undef INTERIOR_AND_EXTERIOR | #undef INTERIOR_AND_EXTERIOR | ||||||
| #undef INTERIOR | #undef INTERIOR | ||||||
| #define EXTERIOR | #define EXTERIOR | ||||||
|  |  | ||||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") |  | ||||||
| template<> void | template<> void | ||||||
| WilsonKernels<WilsonImplF>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<WilsonImplF>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") |  | ||||||
| template<> void | template<> void | ||||||
| WilsonKernels<ZWilsonImplF>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<ZWilsonImplF>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
| //#pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | template<> void | ||||||
| //template<> void | WilsonKernels<WilsonImplFH>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //WilsonKernels<WilsonImplFH>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
| //#pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| //template<> void | #endif | ||||||
| //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> |  | ||||||
|  |  | ||||||
|  | template<> void | ||||||
|  | WilsonKernels<ZWilsonImplFH>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
|  | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
|  |  | ||||||
| ///////////////////////////////////////////////////////////////// | ///////////////////////////////////////////////////////////////// | ||||||
| @@ -152,89 +185,119 @@ WilsonKernels<ZWilsonImplF>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldVi | |||||||
| #define INTERIOR_AND_EXTERIOR | #define INTERIOR_AND_EXTERIOR | ||||||
| #undef INTERIOR | #undef INTERIOR | ||||||
| #undef EXTERIOR | #undef EXTERIOR | ||||||
|  |  | ||||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") |  | ||||||
| template<> void | template<> void | ||||||
| WilsonKernels<WilsonImplF>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<WilsonImplF>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") |  | ||||||
| template<> void | template<> void | ||||||
| WilsonKernels<ZWilsonImplF>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<ZWilsonImplF>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
| //#pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | template<> void | ||||||
| //template<> void | WilsonKernels<WilsonImplFH>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //WilsonKernels<WilsonImplFH>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
| //#pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| //template<> void | #endif | ||||||
| //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> |  | ||||||
|  |  | ||||||
|  | template<> void | ||||||
|  | WilsonKernels<ZWilsonImplFH>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
|  | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
| #undef INTERIOR_AND_EXTERIOR | #undef INTERIOR_AND_EXTERIOR | ||||||
| #define INTERIOR | #define INTERIOR | ||||||
| #undef EXTERIOR | #undef EXTERIOR | ||||||
|  |  | ||||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") |  | ||||||
| template<> void | template<> void | ||||||
| WilsonKernels<WilsonImplF>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<WilsonImplF>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") |  | ||||||
| template<> void | template<> void | ||||||
| WilsonKernels<ZWilsonImplF>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<ZWilsonImplF>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
| //#pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | template<> void | ||||||
| //template<> void | WilsonKernels<WilsonImplFH>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //WilsonKernels<WilsonImplFH>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
| //#pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| //template<> void | #endif | ||||||
| //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> |  | ||||||
|  |  | ||||||
|  | template<> void | ||||||
|  | WilsonKernels<ZWilsonImplFH>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
|  | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
| #undef INTERIOR_AND_EXTERIOR | #undef INTERIOR_AND_EXTERIOR | ||||||
| #undef INTERIOR | #undef INTERIOR | ||||||
| #define EXTERIOR | #define EXTERIOR | ||||||
|  |  | ||||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") |  | ||||||
| template<> void | template<> void | ||||||
| WilsonKernels<WilsonImplF>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<WilsonImplF>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") |  | ||||||
| template<> void | template<> void | ||||||
| WilsonKernels<ZWilsonImplF>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<ZWilsonImplF>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
| //#pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | template<> void | ||||||
| //template<> void | WilsonKernels<WilsonImplFH>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //WilsonKernels<WilsonImplFH>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
| //#pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| //template<> void | #endif | ||||||
| //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> |  | ||||||
|  |  | ||||||
|  | template<> void | ||||||
|  | WilsonKernels<ZWilsonImplFH>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
|  | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
|  |  | ||||||
| // undefine | // undefine | ||||||
| @@ -267,89 +330,119 @@ WilsonKernels<ZWilsonImplF>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFiel | |||||||
| #define INTERIOR_AND_EXTERIOR | #define INTERIOR_AND_EXTERIOR | ||||||
| #undef INTERIOR | #undef INTERIOR | ||||||
| #undef EXTERIOR | #undef EXTERIOR | ||||||
|  |  | ||||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") |  | ||||||
| template<> void | template<> void | ||||||
| WilsonKernels<WilsonImplD>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<WilsonImplD>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") |  | ||||||
| template<> void | template<> void | ||||||
| WilsonKernels<ZWilsonImplD>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<ZWilsonImplD>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
| // #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | template<> void | ||||||
| // template<> void | WilsonKernels<WilsonImplDF>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| // WilsonKernels<WilsonImplDF>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| // 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| // #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
| // #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| // template<> void | #endif | ||||||
| // 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> |  | ||||||
|  |  | ||||||
|  | template<> void | ||||||
|  | WilsonKernels<ZWilsonImplDF>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
|  | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
| #undef INTERIOR_AND_EXTERIOR | #undef INTERIOR_AND_EXTERIOR | ||||||
| #define INTERIOR | #define INTERIOR | ||||||
| #undef EXTERIOR | #undef EXTERIOR | ||||||
|  |  | ||||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") |  | ||||||
| template<> void | template<> void | ||||||
| WilsonKernels<WilsonImplD>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<WilsonImplD>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") |  | ||||||
| template<> void | template<> void | ||||||
| WilsonKernels<ZWilsonImplD>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<ZWilsonImplD>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
| // #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | template<> void | ||||||
| // template<> void | WilsonKernels<WilsonImplDF>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| // WilsonKernels<WilsonImplDF>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| // 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| // #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
| // #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| // template<> void | #endif | ||||||
| // 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> |  | ||||||
|  |  | ||||||
|  | template<> void | ||||||
|  | WilsonKernels<ZWilsonImplDF>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
|  | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
| #undef INTERIOR_AND_EXTERIOR | #undef INTERIOR_AND_EXTERIOR | ||||||
| #undef INTERIOR | #undef INTERIOR | ||||||
| #define EXTERIOR | #define EXTERIOR | ||||||
|  |  | ||||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") |  | ||||||
| template<> void | template<> void | ||||||
| WilsonKernels<WilsonImplD>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<WilsonImplD>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") |  | ||||||
| template<> void | template<> void | ||||||
| WilsonKernels<ZWilsonImplD>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<ZWilsonImplD>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
| // #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | template<> void | ||||||
| // template<> void | WilsonKernels<WilsonImplDF>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| // WilsonKernels<WilsonImplDF>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| // 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| // #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
| // #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| // template<> void | #endif | ||||||
| // 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> |  | ||||||
|  |  | ||||||
|  | template<> void | ||||||
|  | WilsonKernels<ZWilsonImplDF>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
|  | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
| ///////////////////////////////////////////////////////////////// | ///////////////////////////////////////////////////////////////// | ||||||
| // XYZT vectorised, dag Kernel, double | // XYZT vectorised, dag Kernel, double | ||||||
| @@ -358,93 +451,124 @@ WilsonKernels<ZWilsonImplD>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldVi | |||||||
| #define INTERIOR_AND_EXTERIOR | #define INTERIOR_AND_EXTERIOR | ||||||
| #undef INTERIOR | #undef INTERIOR | ||||||
| #undef EXTERIOR | #undef EXTERIOR | ||||||
|  |  | ||||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") |  | ||||||
| template<> void | template<> void | ||||||
| WilsonKernels<WilsonImplD>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<WilsonImplD>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") |  | ||||||
| template<> void | template<> void | ||||||
| WilsonKernels<ZWilsonImplD>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<ZWilsonImplD>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
| // #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | template<> void | ||||||
| // template<> void | WilsonKernels<WilsonImplDF>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| // WilsonKernels<WilsonImplDF>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| // 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| // #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
| // #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| // template<> void | #endif | ||||||
| // 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> |  | ||||||
|  |  | ||||||
|  | template<> void | ||||||
|  | WilsonKernels<ZWilsonImplDF>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
|  | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
| #undef INTERIOR_AND_EXTERIOR | #undef INTERIOR_AND_EXTERIOR | ||||||
| #define INTERIOR | #define INTERIOR | ||||||
| #undef EXTERIOR | #undef EXTERIOR | ||||||
|  |  | ||||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") |  | ||||||
| template<> void | template<> void | ||||||
| WilsonKernels<WilsonImplD>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<WilsonImplD>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") |  | ||||||
| template<> void | template<> void | ||||||
| WilsonKernels<ZWilsonImplD>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<ZWilsonImplD>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
| // #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | template<> void | ||||||
| // template<> void | WilsonKernels<WilsonImplDF>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| // WilsonKernels<WilsonImplDF>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| // 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| // #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
| // #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| // template<> void | #endif | ||||||
| // 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> |  | ||||||
|  |  | ||||||
|  | template<> void | ||||||
|  | WilsonKernels<ZWilsonImplDF>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
|  | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
| #undef INTERIOR_AND_EXTERIOR | #undef INTERIOR_AND_EXTERIOR | ||||||
| #undef INTERIOR | #undef INTERIOR | ||||||
| #define EXTERIOR | #define EXTERIOR | ||||||
|  |  | ||||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") |  | ||||||
| template<> void | template<> void | ||||||
| WilsonKernels<WilsonImplD>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<WilsonImplD>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
| #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") |  | ||||||
| template<> void | template<> void | ||||||
| WilsonKernels<ZWilsonImplD>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<ZWilsonImplD>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
| // #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | template<> void | ||||||
| // template<> void | WilsonKernels<WilsonImplDF>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| // WilsonKernels<WilsonImplDF>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| // 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
| // #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
| // #pragma GCC optimize ("-O3", "-fno-schedule-insns", "-fno-schedule-insns2") | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| // template<> void | #endif | ||||||
| // 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> |  | ||||||
|  |  | ||||||
|  | template<> void | ||||||
|  | WilsonKernels<ZWilsonImplDF>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
|  | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
|  | #if defined (WILSONKERNELSASMBODYA64FX) | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBodyA64FX.h> | ||||||
|  | #else | ||||||
|  | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  | #endif | ||||||
|  |  | ||||||
|  |  | ||||||
|  |  | ||||||
| // undefs | // undefs | ||||||
|  | #undef WILSONKERNELSASMBODYA64FX | ||||||
| #include <simd/Fujitsu_A64FX_undef.h> | #include <simd/Fujitsu_A64FX_undef.h> | ||||||
|  |  | ||||||
| #endif //A64FXASM | #endif //A64FXASM | ||||||
|   | |||||||
| @@ -74,15 +74,15 @@ WilsonKernels<ZWilsonImplF>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView | |||||||
| 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<WilsonImplFH>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<WilsonImplFH>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| // |  | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<ZWilsonImplFH>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<ZWilsonImplFH>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| #undef INTERIOR_AND_EXTERIOR | #undef INTERIOR_AND_EXTERIOR | ||||||
| #define INTERIOR | #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) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<WilsonImplFH>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<WilsonImplFH>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| // |  | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<ZWilsonImplFH>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<ZWilsonImplFH>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
|  |  | ||||||
| #undef INTERIOR_AND_EXTERIOR | #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) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<WilsonImplFH>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<WilsonImplFH>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| // |  | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<ZWilsonImplFH>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<ZWilsonImplFH>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|        |        | ||||||
| ///////////////////////////////////////////////////////////////// | ///////////////////////////////////////////////////////////////// | ||||||
| // XYZT vectorised, dag Kernel, single | // 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) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<WilsonImplFH>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<WilsonImplFH>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| // |  | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<ZWilsonImplFH>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<ZWilsonImplFH>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| #undef INTERIOR_AND_EXTERIOR | #undef INTERIOR_AND_EXTERIOR | ||||||
| #define INTERIOR | #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) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<WilsonImplFH>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<WilsonImplFH>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| // |  | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<ZWilsonImplFH>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<ZWilsonImplFH>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| #undef INTERIOR_AND_EXTERIOR | #undef INTERIOR_AND_EXTERIOR | ||||||
| #undef INTERIOR | #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) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| 				     | 				     | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<WilsonImplFH>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<WilsonImplFH>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| // | 				     | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<ZWilsonImplFH>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<ZWilsonImplFH>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| 				     | 				     | ||||||
| #undef MAYBEPERM | #undef MAYBEPERM | ||||||
| #undef MULT_2SPIN | #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) | 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<DomainWallVec5dImplFH>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<DomainWallVec5dImplFH>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<ZDomainWallVec5dImplFH>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<ZDomainWallVec5dImplFH>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| #undef INTERIOR_AND_EXTERIOR | #undef INTERIOR_AND_EXTERIOR | ||||||
| #define INTERIOR | #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) | 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<DomainWallVec5dImplFH>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<DomainWallVec5dImplFH>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<ZDomainWallVec5dImplFH>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<ZDomainWallVec5dImplFH>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| #undef INTERIOR_AND_EXTERIOR | #undef INTERIOR_AND_EXTERIOR | ||||||
| #undef INTERIOR | #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) | 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| 				     | 				     | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<DomainWallVec5dImplFH>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<DomainWallVec5dImplFH>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| // | 				     | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<ZDomainWallVec5dImplFH>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<ZDomainWallVec5dImplFH>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| 				     | 				     | ||||||
| ///////////////////////////////////////////////////////////////// | ///////////////////////////////////////////////////////////////// | ||||||
| // Ls vectorised, dag Kernel, single | // 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) | 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<DomainWallVec5dImplFH>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | WilsonKernels<DomainWallVec5dImplFH>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||||
| //							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<ZDomainWallVec5dImplFH>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | WilsonKernels<ZDomainWallVec5dImplFH>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||||
| //							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| #undef INTERIOR_AND_EXTERIOR | #undef INTERIOR_AND_EXTERIOR | ||||||
| #define INTERIOR | #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) | 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<DomainWallVec5dImplFH>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | WilsonKernels<DomainWallVec5dImplFH>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||||
| //							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<ZDomainWallVec5dImplFH>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | WilsonKernels<ZDomainWallVec5dImplFH>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||||
| //							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| #undef INTERIOR_AND_EXTERIOR | #undef INTERIOR_AND_EXTERIOR | ||||||
| #undef INTERIOR | #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) | 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<DomainWallVec5dImplFH>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | WilsonKernels<DomainWallVec5dImplFH>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||||
| //							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<ZDomainWallVec5dImplFH>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | WilsonKernels<ZDomainWallVec5dImplFH>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||||
| //							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| #endif  // VEC 5D | #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) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<WilsonImplDF>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<WilsonImplDF>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<ZWilsonImplDF>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<ZWilsonImplDF>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| #undef INTERIOR_AND_EXTERIOR | #undef INTERIOR_AND_EXTERIOR | ||||||
| #define INTERIOR | #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) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<WilsonImplDF>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<WilsonImplDF>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<ZWilsonImplDF>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<ZWilsonImplDF>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| #undef INTERIOR_AND_EXTERIOR | #undef INTERIOR_AND_EXTERIOR | ||||||
| #undef INTERIOR | #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) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|        |        | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<WilsonImplDF>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<WilsonImplDF>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<ZWilsonImplDF>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<ZWilsonImplDF>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|        |        | ||||||
| ///////////////////////////////////////////////////////////////// | ///////////////////////////////////////////////////////////////// | ||||||
| // XYZT vectorised, dag Kernel, single | // 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) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<WilsonImplDF>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<WilsonImplDF>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<ZWilsonImplDF>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<ZWilsonImplDF>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| #undef INTERIOR_AND_EXTERIOR | #undef INTERIOR_AND_EXTERIOR | ||||||
| #define INTERIOR | #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) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<WilsonImplDF>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<WilsonImplDF>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<ZWilsonImplDF>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<ZWilsonImplDF>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| #undef INTERIOR_AND_EXTERIOR | #undef INTERIOR_AND_EXTERIOR | ||||||
| #undef INTERIOR | #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) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| 				     | 				     | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<WilsonImplDF>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<WilsonImplDF>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<ZWilsonImplDF>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<ZWilsonImplDF>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 						int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| 				     | 				     | ||||||
| #undef MAYBEPERM | #undef MAYBEPERM | ||||||
| #undef MULT_2SPIN | #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) | 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<DomainWallVec5dImplDF>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<DomainWallVec5dImplDF>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<ZDomainWallVec5dImplDF>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<ZDomainWallVec5dImplDF>::AsmDhopSite(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| #undef INTERIOR_AND_EXTERIOR | #undef INTERIOR_AND_EXTERIOR | ||||||
| #define INTERIOR | #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) | 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<DomainWallVec5dImplDF>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<DomainWallVec5dImplDF>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<ZDomainWallVec5dImplDF>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<ZDomainWallVec5dImplDF>::AsmDhopSiteInt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| #undef INTERIOR_AND_EXTERIOR | #undef INTERIOR_AND_EXTERIOR | ||||||
| #undef INTERIOR | #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) | 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| 				     | 				     | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<DomainWallVec5dImplDF>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<DomainWallVec5dImplDF>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<ZDomainWallVec5dImplDF>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | WilsonKernels<ZDomainWallVec5dImplDF>::AsmDhopSiteExt(StencilView &st, DoubledGaugeFieldView &U, SiteHalfSpinor *buf, | ||||||
| //							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 							 int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| 				     | 				     | ||||||
| ///////////////////////////////////////////////////////////////// | ///////////////////////////////////////////////////////////////// | ||||||
| // Ls vectorised, dag Kernel, single | // 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) | 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<DomainWallVec5dImplDF>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | WilsonKernels<DomainWallVec5dImplDF>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||||
| //							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<ZDomainWallVec5dImplDF>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | WilsonKernels<ZDomainWallVec5dImplDF>::AsmDhopSiteDag(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||||
| //							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| #undef INTERIOR_AND_EXTERIOR | #undef INTERIOR_AND_EXTERIOR | ||||||
| #define INTERIOR | #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) | 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<DomainWallVec5dImplDF>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | WilsonKernels<DomainWallVec5dImplDF>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||||
| //							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<ZDomainWallVec5dImplDF>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | WilsonKernels<ZDomainWallVec5dImplDF>::AsmDhopSiteDagInt(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||||
| //							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| #undef INTERIOR_AND_EXTERIOR | #undef INTERIOR_AND_EXTERIOR | ||||||
| #undef INTERIOR | #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) | 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<DomainWallVec5dImplDF>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | WilsonKernels<DomainWallVec5dImplDF>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||||
| //							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
| //template<> void | template<> void  | ||||||
| //WilsonKernels<ZDomainWallVec5dImplDF>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | WilsonKernels<ZDomainWallVec5dImplDF>::AsmDhopSiteDagExt(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||||
| //							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | 							    int ss,int ssU,int Ls,int Ns,const FermionFieldView &in, FermionFieldView &out) | ||||||
| //#include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | #include <qcd/action/fermion/implementation/WilsonKernelsAsmBody.h> | ||||||
|  |  | ||||||
| #endif  // VEC 5D | #endif  // VEC 5D | ||||||
|  |  | ||||||
|   | |||||||
| @@ -25,11 +25,6 @@ Author:  Nils Meyer  <nils.meyer@ur.de>  Regensburg University | |||||||
|     See the full license in the file "LICENSE" in the top level distribution directory |     See the full license in the file "LICENSE" in the top level distribution directory | ||||||
| *************************************************************************************/ | *************************************************************************************/ | ||||||
| /*  END LEGAL */ | /*  END LEGAL */ | ||||||
|  |  | ||||||
| // GCC 10 messes up SVE instruction scheduling using -O3, but |  | ||||||
| // -O3 -fno-schedule-insns -fno-schedule-insns2 does wonders |  | ||||||
| // performance now is better than armclang 20.2 |  | ||||||
|  |  | ||||||
| #ifdef KERNEL_DAG | #ifdef KERNEL_DAG | ||||||
| #define DIR0_PROJ    XP_PROJ | #define DIR0_PROJ    XP_PROJ | ||||||
| #define DIR1_PROJ    YP_PROJ | #define DIR1_PROJ    YP_PROJ | ||||||
| @@ -102,7 +97,7 @@ Author:  Nils Meyer  <nils.meyer@ur.de>  Regensburg University | |||||||
|     PROJ;							                        \ |     PROJ;							                        \ | ||||||
|     MAYBEPERM(PERMUTE_DIR,perm);					        \ |     MAYBEPERM(PERMUTE_DIR,perm);					        \ | ||||||
|       } else {								                \ |       } else {								                \ | ||||||
| 	  LOAD_CHI(base);							                \ | 	LOAD_CHI(base);							                \ | ||||||
|       }									                    \ |       }									                    \ | ||||||
|       base = st.GetInfo(ptype,local,perm,NxtDir,ent,plocal); ent++;	\ |       base = st.GetInfo(ptype,local,perm,NxtDir,ent,plocal); ent++;	\ | ||||||
|     MULT_2SPIN_1(Dir);					                    \ |     MULT_2SPIN_1(Dir);					                    \ | ||||||
| @@ -115,11 +110,6 @@ Author:  Nils Meyer  <nils.meyer@ur.de>  Regensburg University | |||||||
|     }                                                       \ |     }                                                       \ | ||||||
|     RECON;								                    \ |     RECON;								                    \ | ||||||
|  |  | ||||||
| /* |  | ||||||
| NB: picking PREFETCH_GAUGE_L2(Dir+4); here results in performance penalty |  | ||||||
|     though I expected that it would improve on performance |  | ||||||
| */ |  | ||||||
|  |  | ||||||
| #define ASM_LEG_XP(Dir,NxtDir,PERMUTE_DIR,PROJ,RECON)	    \ | #define ASM_LEG_XP(Dir,NxtDir,PERMUTE_DIR,PROJ,RECON)	    \ | ||||||
|   base = st.GetInfo(ptype,local,perm,Dir,ent,plocal); ent++; \ |   base = st.GetInfo(ptype,local,perm,Dir,ent,plocal); ent++; \ | ||||||
|   PREFETCH1_CHIMU(base);						            \ |   PREFETCH1_CHIMU(base);						            \ | ||||||
| @@ -136,63 +126,73 @@ NB: picking PREFETCH_GAUGE_L2(Dir+4); here results in performance penalty | |||||||
|  |  | ||||||
| #define ASM_LEG(Dir,NxtDir,PERMUTE_DIR,PROJ,RECON)			\ | #define ASM_LEG(Dir,NxtDir,PERMUTE_DIR,PROJ,RECON)			\ | ||||||
|       basep = st.GetPFInfo(nent,plocal); nent++;			\ |       basep = st.GetPFInfo(nent,plocal); nent++;			\ | ||||||
|       if ( local ) {							\ |       if ( local ) {							            \ | ||||||
|   LOAD_CHIMU(base);                                       \ |     LOAD_CHIMU(base);                                       \ | ||||||
|   LOAD_TABLE(PERMUTE_DIR);                                \ |     LOAD_TABLE(PERMUTE_DIR);                                \ | ||||||
|   PROJ;							                        \ |     PROJ;							                        \ | ||||||
|   MAYBEPERM(PERMUTE_DIR,perm);					        \ |     MAYBEPERM(PERMUTE_DIR,perm);					        \ | ||||||
|       }else if ( st.same_node[Dir] ) {LOAD_CHI(base);}			\ |       }else if ( st.same_node[Dir] ) {LOAD_CHI(base);}	    \ | ||||||
|       if ( local || st.same_node[Dir] ) {				\ |       base = st.GetInfo(ptype,local,perm,NxtDir,ent,plocal); ent++;	\ | ||||||
|   MULT_2SPIN_1(Dir);					                    \ |       if ( local || st.same_node[Dir] ) {				    \ | ||||||
|   MULT_2SPIN_2;					                        \ |     MULT_2SPIN_1(Dir);					                    \ | ||||||
|   RECON;								\ |     PREFETCH_CHIMU(base);                                   \ | ||||||
|       }									\ |     /* PREFETCH_GAUGE_L1(NxtDir); */                        \ | ||||||
|   base = st.GetInfo(ptype,local,perm,NxtDir,ent,plocal); ent++;	\ |     MULT_2SPIN_2;					                        \ | ||||||
|   PREFETCH_CHIMU(base);						\ |     if (s == 0) {                                           \ | ||||||
|   PREFETCH_CHIMU_L2(basep);                               \ |        if ((Dir == 0) || (Dir == 4)) { PREFETCH_GAUGE_L2(Dir); } \ | ||||||
|  |     }                                                       \ | ||||||
|  |     RECON;								                    \ | ||||||
|  |     PREFETCH_CHIMU_L2(basep);                               \ | ||||||
|  |       } else { PREFETCH_CHIMU(base); }								                    \ | ||||||
|  |  | ||||||
| #define ASM_LEG_XP(Dir,NxtDir,PERMUTE_DIR,PROJ,RECON)			\ | #define ASM_LEG_XP(Dir,NxtDir,PERMUTE_DIR,PROJ,RECON)			\ | ||||||
|   base = st.GetInfo(ptype,local,perm,Dir,ent,plocal); ent++;		\ |   base = st.GetInfo(ptype,local,perm,Dir,ent,plocal); ent++;		\ | ||||||
|   PREFETCH1_CHIMU(base);						\ |   PREFETCH1_CHIMU(base);						\ | ||||||
|   { ZERO_PSI; }								\ |  | ||||||
|   ASM_LEG(Dir,NxtDir,PERMUTE_DIR,PROJ,RECON) |   ASM_LEG(Dir,NxtDir,PERMUTE_DIR,PROJ,RECON) | ||||||
|  |  | ||||||
| #define RESULT(base,basep) SAVE_RESULT(base,basep); | #define RESULT(base,basep) SAVE_RESULT(base,basep); | ||||||
|  |  | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
| //////////////////////////////////////////////////////////////////////////////// | //////////////////////////////////////////////////////////////////////////////// | ||||||
| // Post comms kernel | // Post comms kernel | ||||||
| //////////////////////////////////////////////////////////////////////////////// | //////////////////////////////////////////////////////////////////////////////// | ||||||
| #ifdef EXTERIOR | #ifdef EXTERIOR | ||||||
|  |  | ||||||
|  |  | ||||||
| #define ASM_LEG(Dir,NxtDir,PERMUTE_DIR,PROJ,RECON)			\ | #define ASM_LEG(Dir,NxtDir,PERMUTE_DIR,PROJ,RECON)			\ | ||||||
|   base = st.GetInfo(ptype,local,perm,Dir,ent,plocal); ent++;		\ |   base = st.GetInfo(ptype,local,perm,Dir,ent,plocal); ent++; \ | ||||||
|   if((!local)&&(!st.same_node[Dir]) ) {					\ |   if((!local)&&(!st.same_node[Dir]) ) {					    \ | ||||||
|     LOAD_CHI(base);							\ |     LOAD_CHI(base);							                \ | ||||||
|     MULT_2SPIN_1(Dir);					                    \ |     MULT_2SPIN_1(Dir);					                    \ | ||||||
|  |     PREFETCH_CHIMU(base);                                   \ | ||||||
|  |     /* PREFETCH_GAUGE_L1(NxtDir); */                        \ | ||||||
|     MULT_2SPIN_2;					                        \ |     MULT_2SPIN_2;					                        \ | ||||||
|     RECON;								\ |     if (s == 0) {                                           \ | ||||||
|     nmu++;								\ |       if ((Dir == 0) || (Dir == 4)) { PREFETCH_GAUGE_L2(Dir); } \ | ||||||
|  |     }                                                       \ | ||||||
|  |     RECON;								                    \ | ||||||
|  |     nmu++;								                    \ | ||||||
|   } |   } | ||||||
|  |  | ||||||
| #define ASM_LEG_XP(Dir,NxtDir,PERMUTE_DIR,PROJ,RECON)			\ | #define ASM_LEG_XP(Dir,NxtDir,PERMUTE_DIR,PROJ,RECON)	    \ | ||||||
|   nmu=0;								\ |   nmu=0;								                    \ | ||||||
|   { ZERO_PSI;}								\ |   base = st.GetInfo(ptype,local,perm,Dir,ent,plocal); ent++;\ | ||||||
|   base = st.GetInfo(ptype,local,perm,Dir,ent,plocal); ent++;		\ |   if((!local)&&(!st.same_node[Dir]) ) {					    \ | ||||||
|   if((!local)&&(!st.same_node[Dir]) ) {					\ |     LOAD_CHI(base);							                \ | ||||||
|     LOAD_CHI(base);							\ |  | ||||||
|     MULT_2SPIN_1(Dir);					                    \ |     MULT_2SPIN_1(Dir);					                    \ | ||||||
|  |     PREFETCH_CHIMU(base);                                   \ | ||||||
|  |     /* PREFETCH_GAUGE_L1(NxtDir); */                        \ | ||||||
|     MULT_2SPIN_2;					                        \ |     MULT_2SPIN_2;					                        \ | ||||||
|     RECON;								\ |     if (s == 0) {                                           \ | ||||||
|     nmu++;								\ |       if ((Dir == 0) || (Dir == 4)) { PREFETCH_GAUGE_L2(Dir); } \ | ||||||
|  |     }                                                       \ | ||||||
|  |     RECON;								                    \ | ||||||
|  |     nmu++;								                    \ | ||||||
|   } |   } | ||||||
|  |  | ||||||
| #define RESULT(base,basep) if (nmu){ ADD_RESULT(base,base);} | #define RESULT(base,basep) if (nmu){ ADD_RESULT(base,base);} | ||||||
|  |  | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
|  |  | ||||||
| { | { | ||||||
|   int nmu; |   int nmu; | ||||||
|   int local,perm, ptype; |   int local,perm, ptype; | ||||||
| @@ -209,6 +209,7 @@ NB: picking PREFETCH_GAUGE_L2(Dir+4); here results in performance penalty | |||||||
|     int ssn=ssU+1;     if(ssn>=nmax) ssn=0; |     int ssn=ssU+1;     if(ssn>=nmax) ssn=0; | ||||||
|     //    int sUn=lo.Reorder(ssn); |     //    int sUn=lo.Reorder(ssn); | ||||||
|     int sUn=ssn; |     int sUn=ssn; | ||||||
|  |     LOCK_GAUGE(0); | ||||||
| #else | #else | ||||||
|     int sU =ssU; |     int sU =ssU; | ||||||
|     int ssn=ssU+1;     if(ssn>=nmax) ssn=0; |     int ssn=ssU+1;     if(ssn>=nmax) ssn=0; | ||||||
| @@ -294,11 +295,6 @@ NB: picking PREFETCH_GAUGE_L2(Dir+4); here results in performance penalty | |||||||
|       std::cout << "----------------------------------------------------" << std::endl; |       std::cout << "----------------------------------------------------" << std::endl; | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
|       // DC ZVA test |  | ||||||
|       // { uint64_t basestore = (uint64_t)&out[ss]; |  | ||||||
|       //   PREFETCH_RESULT_L2_STORE(basestore); } |  | ||||||
|  |  | ||||||
|  |  | ||||||
|       ASM_LEG(Ym,Zm,PERMUTE_DIR2,DIR5_PROJ,DIR5_RECON); |       ASM_LEG(Ym,Zm,PERMUTE_DIR2,DIR5_PROJ,DIR5_RECON); | ||||||
|  |  | ||||||
| #ifdef SHOW | #ifdef SHOW | ||||||
| @@ -312,11 +308,6 @@ NB: picking PREFETCH_GAUGE_L2(Dir+4); here results in performance penalty | |||||||
|       std::cout << "----------------------------------------------------" << std::endl; |       std::cout << "----------------------------------------------------" << std::endl; | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
|       // DC ZVA test |  | ||||||
|       //{ uint64_t basestore = (uint64_t)&out[ss]; |  | ||||||
|       //  PREFETCH_RESULT_L2_STORE(basestore); } |  | ||||||
|  |  | ||||||
|  |  | ||||||
|       ASM_LEG(Zm,Tm,PERMUTE_DIR1,DIR6_PROJ,DIR6_RECON); |       ASM_LEG(Zm,Tm,PERMUTE_DIR1,DIR6_PROJ,DIR6_RECON); | ||||||
|  |  | ||||||
| #ifdef SHOW | #ifdef SHOW | ||||||
| @@ -330,11 +321,6 @@ NB: picking PREFETCH_GAUGE_L2(Dir+4); here results in performance penalty | |||||||
|       std::cout << "----------------------------------------------------" << std::endl; |       std::cout << "----------------------------------------------------" << std::endl; | ||||||
| #endif | #endif | ||||||
|  |  | ||||||
|       // DC ZVA test |  | ||||||
|       //{ uint64_t basestore = (uint64_t)&out[ss]; |  | ||||||
|       //  PREFETCH_RESULT_L2_STORE(basestore); } |  | ||||||
|  |  | ||||||
|  |  | ||||||
|       ASM_LEG(Tm,Xp,PERMUTE_DIR0,DIR7_PROJ,DIR7_RECON); |       ASM_LEG(Tm,Xp,PERMUTE_DIR0,DIR7_PROJ,DIR7_RECON); | ||||||
|  |  | ||||||
| #ifdef SHOW | #ifdef SHOW | ||||||
| @@ -355,7 +341,6 @@ NB: picking PREFETCH_GAUGE_L2(Dir+4); here results in performance penalty | |||||||
|       base = (uint64_t) &out[ss]; |       base = (uint64_t) &out[ss]; | ||||||
|       basep= st.GetPFInfo(nent,plocal); ent++; |       basep= st.GetPFInfo(nent,plocal); ent++; | ||||||
|       basep = (uint64_t) &out[ssn]; |       basep = (uint64_t) &out[ssn]; | ||||||
|       //PREFETCH_RESULT_L1_STORE(base); |  | ||||||
|       RESULT(base,basep); |       RESULT(base,basep); | ||||||
|  |  | ||||||
| #ifdef SHOW | #ifdef SHOW | ||||||
|   | |||||||
| @@ -76,24 +76,7 @@ Author: paboyle <paboyle@ph.ed.ac.uk> | |||||||
|  |  | ||||||
| #define REGISTER | #define REGISTER | ||||||
|  |  | ||||||
| #ifdef GRID_SIMT | #define LOAD_CHIMU \ | ||||||
| #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);	} |  | ||||||
| #define PERMUTE_DIR(dir) ; |  | ||||||
| #else |  | ||||||
| #define LOAD_CHIMU(ptype)		\ |  | ||||||
|   {const SiteSpinor & ref (in[offset]);	\ |   {const SiteSpinor & ref (in[offset]);	\ | ||||||
|     Chimu_00=ref()(0)(0);\ |     Chimu_00=ref()(0)(0);\ | ||||||
|     Chimu_01=ref()(0)(1);\ |     Chimu_01=ref()(0)(1);\ | ||||||
| @@ -108,54 +91,54 @@ Author: paboyle <paboyle@ph.ed.ac.uk> | |||||||
|     Chimu_31=ref()(3)(1);\ |     Chimu_31=ref()(3)(1);\ | ||||||
|     Chimu_32=ref()(3)(2);} |     Chimu_32=ref()(3)(2);} | ||||||
|  |  | ||||||
| #define PERMUTE_DIR(dir)			\ | #define LOAD_CHI\ | ||||||
|   permute##dir(Chi_00,Chi_00);	\ |   {const SiteHalfSpinor &ref(buf[offset]);	\ | ||||||
|       permute##dir(Chi_01,Chi_01);\ |     Chi_00 = ref()(0)(0);\ | ||||||
|       permute##dir(Chi_02,Chi_02);\ |     Chi_01 = ref()(0)(1);\ | ||||||
|       permute##dir(Chi_10,Chi_10);	\ |     Chi_02 = ref()(0)(2);\ | ||||||
|       permute##dir(Chi_11,Chi_11);\ |     Chi_10 = ref()(1)(0);\ | ||||||
|       permute##dir(Chi_12,Chi_12); |     Chi_11 = ref()(1)(1);\ | ||||||
|  |     Chi_12 = ref()(1)(2);} | ||||||
| #endif |  | ||||||
|  |  | ||||||
|  | // To splat or not to splat depends on the implementation | ||||||
| #define MULT_2SPIN(A)\ | #define MULT_2SPIN(A)\ | ||||||
|   {auto & ref(U[sU](A));						\ |   {auto & ref(U[sU](A));			\ | ||||||
|     U_00=coalescedRead(ref()(0,0),lane);				\ |    Impl::loadLinkElement(U_00,ref()(0,0));	\ | ||||||
|     U_10=coalescedRead(ref()(1,0),lane);				\ |    Impl::loadLinkElement(U_10,ref()(1,0));	\ | ||||||
|     U_20=coalescedRead(ref()(2,0),lane);				\ |    Impl::loadLinkElement(U_20,ref()(2,0));	\ | ||||||
|     U_01=coalescedRead(ref()(0,1),lane);				\ |    Impl::loadLinkElement(U_01,ref()(0,1));	\ | ||||||
|     U_11=coalescedRead(ref()(1,1),lane);				\ |    Impl::loadLinkElement(U_11,ref()(1,1));	\ | ||||||
|     U_21=coalescedRead(ref()(2,1),lane);				\ |    Impl::loadLinkElement(U_21,ref()(2,1));	\ | ||||||
|     UChi_00 = U_00*Chi_00;						\ |     UChi_00 = U_00*Chi_00;\ | ||||||
|     UChi_10 = U_00*Chi_10;						\ |     UChi_10 = U_00*Chi_10;\ | ||||||
|     UChi_01 = U_10*Chi_00;						\ |     UChi_01 = U_10*Chi_00;\ | ||||||
|     UChi_11 = U_10*Chi_10;						\ |     UChi_11 = U_10*Chi_10;\ | ||||||
|     UChi_02 = U_20*Chi_00;						\ |     UChi_02 = U_20*Chi_00;\ | ||||||
|     UChi_12 = U_20*Chi_10;						\ |     UChi_12 = U_20*Chi_10;\ | ||||||
|     UChi_00+= U_01*Chi_01;						\ |     UChi_00+= U_01*Chi_01;\ | ||||||
|     UChi_10+= U_01*Chi_11;						\ |     UChi_10+= U_01*Chi_11;\ | ||||||
|     UChi_01+= U_11*Chi_01;						\ |     UChi_01+= U_11*Chi_01;\ | ||||||
|     UChi_11+= U_11*Chi_11;						\ |     UChi_11+= U_11*Chi_11;\ | ||||||
|     UChi_02+= U_21*Chi_01;						\ |     UChi_02+= U_21*Chi_01;\ | ||||||
|     UChi_12+= U_21*Chi_11;						\ |     UChi_12+= U_21*Chi_11;\ | ||||||
|     U_00=coalescedRead(ref()(0,2),lane);				\ |     Impl::loadLinkElement(U_00,ref()(0,2));	\ | ||||||
|     U_10=coalescedRead(ref()(1,2),lane);				\ |     Impl::loadLinkElement(U_10,ref()(1,2));	\ | ||||||
|     U_20=coalescedRead(ref()(2,2),lane);				\ |     Impl::loadLinkElement(U_20,ref()(2,2));	\ | ||||||
|     UChi_00+= U_00*Chi_02;						\ |     UChi_00+= U_00*Chi_02;\ | ||||||
|     UChi_10+= U_00*Chi_12;						\ |     UChi_10+= U_00*Chi_12;\ | ||||||
|     UChi_01+= U_10*Chi_02;						\ |     UChi_01+= U_10*Chi_02;\ | ||||||
|     UChi_11+= U_10*Chi_12;						\ |     UChi_11+= U_10*Chi_12;\ | ||||||
|     UChi_02+= U_20*Chi_02;						\ |     UChi_02+= U_20*Chi_02;\ | ||||||
|     UChi_12+= U_20*Chi_12;} |     UChi_12+= U_20*Chi_12;} | ||||||
|  |  | ||||||
| #define LOAD_CHI				\ |  | ||||||
|   {const SiteHalfSpinor &ref(buf[offset]);	\ | #define PERMUTE_DIR(dir)			\ | ||||||
|     Chi_00 = coalescedRead(ref()(0)(0),lane);	\ |       permute##dir(Chi_00,Chi_00);\ | ||||||
|     Chi_01 = coalescedRead(ref()(0)(1),lane);	\ |       permute##dir(Chi_01,Chi_01);\ | ||||||
|     Chi_02 = coalescedRead(ref()(0)(2),lane);	\ |       permute##dir(Chi_02,Chi_02);\ | ||||||
|     Chi_10 = coalescedRead(ref()(1)(0),lane);	\ |       permute##dir(Chi_10,Chi_10);\ | ||||||
|     Chi_11 = coalescedRead(ref()(1)(1),lane);	\ |       permute##dir(Chi_11,Chi_11);\ | ||||||
|     Chi_12 = coalescedRead(ref()(1)(2),lane);} |       permute##dir(Chi_12,Chi_12); | ||||||
|  |  | ||||||
| //      hspin(0)=fspin(0)+timesI(fspin(3)); | //      hspin(0)=fspin(0)+timesI(fspin(3)); | ||||||
| //      hspin(1)=fspin(1)+timesI(fspin(2)); | //      hspin(1)=fspin(1)+timesI(fspin(2)); | ||||||
| @@ -370,13 +353,13 @@ Author: paboyle <paboyle@ph.ed.ac.uk> | |||||||
|   result_31-= UChi_11;	\ |   result_31-= UChi_11;	\ | ||||||
|   result_32-= UChi_12; |   result_32-= UChi_12; | ||||||
|  |  | ||||||
| #define HAND_STENCIL_LEGB(PROJ,PERM,DIR,RECON)	\ | #define HAND_STENCIL_LEG(PROJ,PERM,DIR,RECON)	\ | ||||||
|   SE=st.GetEntry(ptype,DIR,ss);			\ |   SE=st.GetEntry(ptype,DIR,ss);			\ | ||||||
|   offset = SE->_offset;				\ |   offset = SE->_offset;				\ | ||||||
|   local  = SE->_is_local;			\ |   local  = SE->_is_local;			\ | ||||||
|   perm   = SE->_permute;			\ |   perm   = SE->_permute;			\ | ||||||
|   if ( local ) {				\ |   if ( local ) {				\ | ||||||
|     LOAD_CHIMU(PERM);				\ |     LOAD_CHIMU;					\ | ||||||
|     PROJ;					\ |     PROJ;					\ | ||||||
|     if ( perm) {				\ |     if ( perm) {				\ | ||||||
|       PERMUTE_DIR(PERM);			\ |       PERMUTE_DIR(PERM);			\ | ||||||
| @@ -384,37 +367,6 @@ Author: paboyle <paboyle@ph.ed.ac.uk> | |||||||
|   } else {					\ |   } else {					\ | ||||||
|     LOAD_CHI;					\ |     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];							\ |  | ||||||
|   ptype=st_perm[DIR];							\ |  | ||||||
|  /*SE=st.GetEntry(ptype,DIR,ss);*/					\ |  | ||||||
|   offset = SE->_offset;				\ |  | ||||||
|   perm   = SE->_permute;			\ |  | ||||||
|   LOAD_CHIMU(PERM);				\ |  | ||||||
|   PROJ;						\ |  | ||||||
|   MULT_2SPIN(DIR);				\ |   MULT_2SPIN(DIR);				\ | ||||||
|   RECON;					 |   RECON;					 | ||||||
|  |  | ||||||
| @@ -424,7 +376,7 @@ Author: paboyle <paboyle@ph.ed.ac.uk> | |||||||
|   local  = SE->_is_local;			\ |   local  = SE->_is_local;			\ | ||||||
|   perm   = SE->_permute;			\ |   perm   = SE->_permute;			\ | ||||||
|   if ( local ) {				\ |   if ( local ) {				\ | ||||||
|     LOAD_CHIMU(PERM);				\ |     LOAD_CHIMU;					\ | ||||||
|     PROJ;					\ |     PROJ;					\ | ||||||
|     if ( perm) {				\ |     if ( perm) {				\ | ||||||
|       PERMUTE_DIR(PERM);			\ |       PERMUTE_DIR(PERM);			\ | ||||||
| @@ -432,12 +384,10 @@ Author: paboyle <paboyle@ph.ed.ac.uk> | |||||||
|   } else if ( st.same_node[DIR] ) {		\ |   } else if ( st.same_node[DIR] ) {		\ | ||||||
|     LOAD_CHI;					\ |     LOAD_CHI;					\ | ||||||
|   }						\ |   }						\ | ||||||
|   acceleratorSynchronise();			\ |  | ||||||
|   if (local || st.same_node[DIR] ) {		\ |   if (local || st.same_node[DIR] ) {		\ | ||||||
|     MULT_2SPIN(DIR);				\ |     MULT_2SPIN(DIR);				\ | ||||||
|     RECON;					\ |     RECON;					\ | ||||||
|   }						\ |   } | ||||||
|   acceleratorSynchronise();			 |  | ||||||
|  |  | ||||||
| #define HAND_STENCIL_LEG_EXT(PROJ,PERM,DIR,RECON)	\ | #define HAND_STENCIL_LEG_EXT(PROJ,PERM,DIR,RECON)	\ | ||||||
|   SE=st.GetEntry(ptype,DIR,ss);			\ |   SE=st.GetEntry(ptype,DIR,ss);			\ | ||||||
| @@ -447,44 +397,44 @@ Author: paboyle <paboyle@ph.ed.ac.uk> | |||||||
|     MULT_2SPIN(DIR);				\ |     MULT_2SPIN(DIR);				\ | ||||||
|     RECON;					\ |     RECON;					\ | ||||||
|     nmu++;					\ |     nmu++;					\ | ||||||
|   }						\ |   } | ||||||
|   acceleratorSynchronise();			 |  | ||||||
|  |  | ||||||
| #define HAND_RESULT(ss)				\ | #define HAND_RESULT(ss)				\ | ||||||
|   {						\ |   {						\ | ||||||
|     SiteSpinor & ref (out[ss]);			\ |     SiteSpinor & ref (out[ss]);		\ | ||||||
|     coalescedWrite(ref()(0)(0),result_00,lane);		\ |     vstream(ref()(0)(0),result_00);		\ | ||||||
|     coalescedWrite(ref()(0)(1),result_01,lane);		\ |     vstream(ref()(0)(1),result_01);		\ | ||||||
|     coalescedWrite(ref()(0)(2),result_02,lane);		\ |     vstream(ref()(0)(2),result_02);		\ | ||||||
|     coalescedWrite(ref()(1)(0),result_10,lane);		\ |     vstream(ref()(1)(0),result_10);		\ | ||||||
|     coalescedWrite(ref()(1)(1),result_11,lane);		\ |     vstream(ref()(1)(1),result_11);		\ | ||||||
|     coalescedWrite(ref()(1)(2),result_12,lane);		\ |     vstream(ref()(1)(2),result_12);		\ | ||||||
|     coalescedWrite(ref()(2)(0),result_20,lane);		\ |     vstream(ref()(2)(0),result_20);		\ | ||||||
|     coalescedWrite(ref()(2)(1),result_21,lane);		\ |     vstream(ref()(2)(1),result_21);		\ | ||||||
|     coalescedWrite(ref()(2)(2),result_22,lane);		\ |     vstream(ref()(2)(2),result_22);		\ | ||||||
|     coalescedWrite(ref()(3)(0),result_30,lane);		\ |     vstream(ref()(3)(0),result_30);		\ | ||||||
|     coalescedWrite(ref()(3)(1),result_31,lane);		\ |     vstream(ref()(3)(1),result_31);		\ | ||||||
|     coalescedWrite(ref()(3)(2),result_32,lane);		\ |     vstream(ref()(3)(2),result_32);		\ | ||||||
|   } |   } | ||||||
|  |  | ||||||
| #define HAND_RESULT_EXT(ss)				\ | #define HAND_RESULT_EXT(ss)			\ | ||||||
|   {							\ |   if (nmu){					\ | ||||||
|     SiteSpinor & ref (out[ss]);				\ |     SiteSpinor & ref (out[ss]);		\ | ||||||
|     coalescedWrite(ref()(0)(0),coalescedRead(ref()(0)(0))+result_00,lane);	\ |     ref()(0)(0)+=result_00;		\ | ||||||
|     coalescedWrite(ref()(0)(1),coalescedRead(ref()(0)(1))+result_01,lane);	\ |     ref()(0)(1)+=result_01;		\ | ||||||
|     coalescedWrite(ref()(0)(2),coalescedRead(ref()(0)(2))+result_02,lane);	\ |     ref()(0)(2)+=result_02;		\ | ||||||
|     coalescedWrite(ref()(1)(0),coalescedRead(ref()(1)(0))+result_10,lane);	\ |     ref()(1)(0)+=result_10;		\ | ||||||
|     coalescedWrite(ref()(1)(1),coalescedRead(ref()(1)(1))+result_11,lane);	\ |     ref()(1)(1)+=result_11;		\ | ||||||
|     coalescedWrite(ref()(1)(2),coalescedRead(ref()(1)(2))+result_12,lane);	\ |     ref()(1)(2)+=result_12;		\ | ||||||
|     coalescedWrite(ref()(2)(0),coalescedRead(ref()(2)(0))+result_20,lane);	\ |     ref()(2)(0)+=result_20;		\ | ||||||
|     coalescedWrite(ref()(2)(1),coalescedRead(ref()(2)(1))+result_21,lane);	\ |     ref()(2)(1)+=result_21;		\ | ||||||
|     coalescedWrite(ref()(2)(2),coalescedRead(ref()(2)(2))+result_22,lane);	\ |     ref()(2)(2)+=result_22;		\ | ||||||
|     coalescedWrite(ref()(3)(0),coalescedRead(ref()(3)(0))+result_30,lane);	\ |     ref()(3)(0)+=result_30;		\ | ||||||
|     coalescedWrite(ref()(3)(1),coalescedRead(ref()(3)(1))+result_31,lane);	\ |     ref()(3)(1)+=result_31;		\ | ||||||
|     coalescedWrite(ref()(3)(2),coalescedRead(ref()(3)(2))+result_32,lane);	\ |     ref()(3)(2)+=result_32;		\ | ||||||
|   } |   } | ||||||
|  |  | ||||||
| #define HAND_DECLARATIONS(Simd)			\ |  | ||||||
|  | #define HAND_DECLARATIONS(a)			\ | ||||||
|   Simd result_00;				\ |   Simd result_00;				\ | ||||||
|   Simd result_01;				\ |   Simd result_01;				\ | ||||||
|   Simd result_02;				\ |   Simd result_02;				\ | ||||||
| @@ -516,19 +466,19 @@ Author: paboyle <paboyle@ph.ed.ac.uk> | |||||||
|   Simd U_11;					\ |   Simd U_11;					\ | ||||||
|   Simd U_21; |   Simd U_21; | ||||||
|  |  | ||||||
| #define ZERO_RESULT							\ | #define ZERO_RESULT				\ | ||||||
|   zeroit(result_00);							\ |   result_00=Zero();				\ | ||||||
|   zeroit(result_01);							\ |   result_01=Zero();				\ | ||||||
|   zeroit(result_02);							\ |   result_02=Zero();				\ | ||||||
|   zeroit(result_10);							\ |   result_10=Zero();				\ | ||||||
|   zeroit(result_11);							\ |   result_11=Zero();				\ | ||||||
|   zeroit(result_12);							\ |   result_12=Zero();				\ | ||||||
|   zeroit(result_20);							\ |   result_20=Zero();				\ | ||||||
|   zeroit(result_21);							\ |   result_21=Zero();				\ | ||||||
|   zeroit(result_22);							\ |   result_22=Zero();				\ | ||||||
|   zeroit(result_30);							\ |   result_30=Zero();				\ | ||||||
|   zeroit(result_31);							\ |   result_31=Zero();				\ | ||||||
|   zeroit(result_32);			 |   result_32=Zero();			 | ||||||
|  |  | ||||||
| #define Chimu_00 Chi_00 | #define Chimu_00 Chi_00 | ||||||
| #define Chimu_01 Chi_01 | #define Chimu_01 Chi_01 | ||||||
| @@ -545,53 +495,15 @@ Author: paboyle <paboyle@ph.ed.ac.uk> | |||||||
|  |  | ||||||
| NAMESPACE_BEGIN(Grid); | NAMESPACE_BEGIN(Grid); | ||||||
|  |  | ||||||
|  |  | ||||||
| #ifdef SYCL_HACK |  | ||||||
| template<class Impl> accelerator_inline void  |  | ||||||
| WilsonKernels<Impl>::HandDhopSiteSycl(StencilVector st_perm,StencilEntry *st_p, SiteDoubledGaugeField *U,SiteHalfSpinor  *buf, |  | ||||||
| 				      int ss,int sU,const SiteSpinor *in, SiteSpinor *out) |  | ||||||
| { |  | ||||||
| // 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; |  | ||||||
|   typedef iSinglet<Simd> vCplx; |  | ||||||
|   //  typedef decltype( coalescedRead( vCplx()()() )) Simt; |  | ||||||
|   typedef decltype( coalescedRead( in[0]()(0)(0) )) Simt; |  | ||||||
|  |  | ||||||
|   const int Nsimd = SiteHalfSpinor::Nsimd(); |  | ||||||
|   const int lane=acceleratorSIMTlane(Nsimd); |  | ||||||
|  |  | ||||||
|   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); |  | ||||||
|   HAND_STENCIL_LEG(TM_PROJ,0,Tp,TM_RECON_ACCUM); |  | ||||||
|   HAND_STENCIL_LEG(XP_PROJ,3,Xm,XP_RECON_ACCUM); |  | ||||||
|   HAND_STENCIL_LEG(YP_PROJ,2,Ym,YP_RECON_ACCUM); |  | ||||||
|   HAND_STENCIL_LEG(ZP_PROJ,1,Zm,ZP_RECON_ACCUM); |  | ||||||
|   HAND_STENCIL_LEG(TP_PROJ,0,Tm,TP_RECON_ACCUM); |  | ||||||
|   HAND_RESULT(ss); |  | ||||||
| } |  | ||||||
| #endif |  | ||||||
|  |  | ||||||
| template<class Impl> accelerator_inline void  | template<class Impl> accelerator_inline void  | ||||||
| WilsonKernels<Impl>::HandDhopSite(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor  *buf, | WilsonKernels<Impl>::HandDhopSite(StencilView &st, DoubledGaugeFieldView &U,SiteHalfSpinor  *buf, | ||||||
| 				  int ss,int sU,const FermionFieldView &in, FermionFieldView &out) | 				  int ss,int sU,const FermionFieldView &in, FermionFieldView &out) | ||||||
| { | { | ||||||
|   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... | // 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::scalar_type S; | ||||||
|   typedef typename Simd::vector_type V; |   typedef typename Simd::vector_type V; | ||||||
|   typedef decltype( coalescedRead( in[0]()(0)(0) )) Simt; |  | ||||||
|  |  | ||||||
|   const int Nsimd = SiteHalfSpinor::Nsimd(); |   HAND_DECLARATIONS(ignore); | ||||||
|   const int lane=acceleratorSIMTlane(Nsimd); |  | ||||||
|  |  | ||||||
|   HAND_DECLARATIONS(Simt); |  | ||||||
|  |  | ||||||
|   int offset,local,perm, ptype; |   int offset,local,perm, ptype; | ||||||
|   StencilEntry *SE; |   StencilEntry *SE; | ||||||
| @@ -611,16 +523,10 @@ template<class Impl>  accelerator_inline | |||||||
| void WilsonKernels<Impl>::HandDhopSiteDag(StencilView &st,DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | void WilsonKernels<Impl>::HandDhopSiteDag(StencilView &st,DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||||
| 					  int ss,int sU,const FermionFieldView &in, FermionFieldView &out) | 					  int ss,int sU,const FermionFieldView &in, FermionFieldView &out) | ||||||
| { | { | ||||||
|   auto st_p = st._entries_p;						 |  | ||||||
|   auto st_perm = st._permute_type;					 |  | ||||||
|   typedef typename Simd::scalar_type S; |   typedef typename Simd::scalar_type S; | ||||||
|   typedef typename Simd::vector_type V; |   typedef typename Simd::vector_type V; | ||||||
|   typedef decltype( coalescedRead( in[0]()(0)(0) )) Simt; |  | ||||||
|  |  | ||||||
|   const int Nsimd = SiteHalfSpinor::Nsimd(); |   HAND_DECLARATIONS(ignore); | ||||||
|   const int lane=acceleratorSIMTlane(Nsimd); |  | ||||||
|  |  | ||||||
|   HAND_DECLARATIONS(Simt); |  | ||||||
|  |  | ||||||
|   StencilEntry *SE; |   StencilEntry *SE; | ||||||
|   int offset,local,perm, ptype; |   int offset,local,perm, ptype; | ||||||
| @@ -640,17 +546,11 @@ template<class Impl>  accelerator_inline void | |||||||
| WilsonKernels<Impl>::HandDhopSiteInt(StencilView &st,DoubledGaugeFieldView &U,SiteHalfSpinor  *buf, | WilsonKernels<Impl>::HandDhopSiteInt(StencilView &st,DoubledGaugeFieldView &U,SiteHalfSpinor  *buf, | ||||||
| 					  int ss,int sU,const FermionFieldView &in, FermionFieldView &out) | 					  int ss,int sU,const FermionFieldView &in, FermionFieldView &out) | ||||||
| { | { | ||||||
|   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... | // 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::scalar_type S; | ||||||
|   typedef typename Simd::vector_type V; |   typedef typename Simd::vector_type V; | ||||||
|   typedef decltype( coalescedRead( in[0]()(0)(0) )) Simt; |  | ||||||
|  |  | ||||||
|   const int Nsimd = SiteHalfSpinor::Nsimd(); |   HAND_DECLARATIONS(ignore); | ||||||
|   const int lane=acceleratorSIMTlane(Nsimd); |  | ||||||
|  |  | ||||||
|   HAND_DECLARATIONS(Simt); |  | ||||||
|  |  | ||||||
|   int offset,local,perm, ptype; |   int offset,local,perm, ptype; | ||||||
|   StencilEntry *SE; |   StencilEntry *SE; | ||||||
| @@ -670,16 +570,10 @@ template<class Impl> accelerator_inline | |||||||
| void WilsonKernels<Impl>::HandDhopSiteDagInt(StencilView &st,DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | void WilsonKernels<Impl>::HandDhopSiteDagInt(StencilView &st,DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||||
| 						  int ss,int sU,const FermionFieldView &in, FermionFieldView &out) | 						  int ss,int sU,const FermionFieldView &in, FermionFieldView &out) | ||||||
| { | { | ||||||
|   auto st_p = st._entries_p;						 |  | ||||||
|   auto st_perm = st._permute_type;					 |  | ||||||
|   typedef typename Simd::scalar_type S; |   typedef typename Simd::scalar_type S; | ||||||
|   typedef typename Simd::vector_type V; |   typedef typename Simd::vector_type V; | ||||||
|   typedef decltype( coalescedRead( in[0]()(0)(0) )) Simt; |  | ||||||
|  |  | ||||||
|   const int Nsimd = SiteHalfSpinor::Nsimd(); |   HAND_DECLARATIONS(ignore); | ||||||
|   const int lane=acceleratorSIMTlane(Nsimd); |  | ||||||
|  |  | ||||||
|   HAND_DECLARATIONS(Simt); |  | ||||||
|  |  | ||||||
|   StencilEntry *SE; |   StencilEntry *SE; | ||||||
|   int offset,local,perm, ptype; |   int offset,local,perm, ptype; | ||||||
| @@ -699,17 +593,11 @@ template<class Impl>  accelerator_inline void | |||||||
| WilsonKernels<Impl>::HandDhopSiteExt(StencilView &st,DoubledGaugeFieldView &U,SiteHalfSpinor  *buf, | WilsonKernels<Impl>::HandDhopSiteExt(StencilView &st,DoubledGaugeFieldView &U,SiteHalfSpinor  *buf, | ||||||
| 					  int ss,int sU,const FermionFieldView &in, FermionFieldView &out) | 					  int ss,int sU,const FermionFieldView &in, FermionFieldView &out) | ||||||
| { | { | ||||||
|   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... | // 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::scalar_type S; | ||||||
|   typedef typename Simd::vector_type V; |   typedef typename Simd::vector_type V; | ||||||
|   typedef decltype( coalescedRead( in[0]()(0)(0) )) Simt; |  | ||||||
|  |  | ||||||
|   const int Nsimd = SiteHalfSpinor::Nsimd(); |   HAND_DECLARATIONS(ignore); | ||||||
|   const int lane=acceleratorSIMTlane(Nsimd); |  | ||||||
|  |  | ||||||
|   HAND_DECLARATIONS(Simt); |  | ||||||
|  |  | ||||||
|   int offset, ptype; |   int offset, ptype; | ||||||
|   StencilEntry *SE; |   StencilEntry *SE; | ||||||
| @@ -730,16 +618,10 @@ template<class Impl>  accelerator_inline | |||||||
| void WilsonKernels<Impl>::HandDhopSiteDagExt(StencilView &st,DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | void WilsonKernels<Impl>::HandDhopSiteDagExt(StencilView &st,DoubledGaugeFieldView &U,SiteHalfSpinor *buf, | ||||||
| 						  int ss,int sU,const FermionFieldView &in, FermionFieldView &out) | 						  int ss,int sU,const FermionFieldView &in, FermionFieldView &out) | ||||||
| { | { | ||||||
|   auto st_p = st._entries_p;						 |  | ||||||
|   auto st_perm = st._permute_type;					 |  | ||||||
|   typedef typename Simd::scalar_type S; |   typedef typename Simd::scalar_type S; | ||||||
|   typedef typename Simd::vector_type V; |   typedef typename Simd::vector_type V; | ||||||
|   typedef decltype( coalescedRead( in[0]()(0)(0) )) Simt; |  | ||||||
|  |  | ||||||
|   const int Nsimd = SiteHalfSpinor::Nsimd(); |   HAND_DECLARATIONS(ignore); | ||||||
|   const int lane=acceleratorSIMTlane(Nsimd); |  | ||||||
|  |  | ||||||
|   HAND_DECLARATIONS(Simt); |  | ||||||
|  |  | ||||||
|   StencilEntry *SE; |   StencilEntry *SE; | ||||||
|   int offset, ptype; |   int offset, ptype; | ||||||
| @@ -800,4 +682,3 @@ NAMESPACE_END(Grid); | |||||||
| #undef HAND_RESULT | #undef HAND_RESULT | ||||||
| #undef HAND_RESULT_INT | #undef HAND_RESULT_INT | ||||||
| #undef HAND_RESULT_EXT | #undef HAND_RESULT_EXT | ||||||
| #undef HAND_DECLARATIONS |  | ||||||
|   | |||||||
| @@ -416,21 +416,7 @@ void WilsonKernels<Impl>::DhopDirKernel( StencilImpl &st, DoubledGaugeField &U,S | |||||||
| #undef LoopBody | #undef LoopBody | ||||||
| } | } | ||||||
|  |  | ||||||
| #define KERNEL_CALL_TMP(A) \ | #define KERNEL_CALLNB(A) \ | ||||||
|   const uint64_t    NN = Nsite*Ls;					\ |  | ||||||
|   auto U_p = & U_v[0];							\ |  | ||||||
|   auto in_p = & in_v[0];						\ |  | ||||||
|   auto out_p = & out_v[0];						\ |  | ||||||
|   auto st_p = st_v._entries_p;						\ |  | ||||||
|   auto st_perm = st_v._permute_type;					\ |  | ||||||
|   accelerator_forNB( ss, NN, Simd::Nsimd(), {				\ |  | ||||||
|       int sF = ss;							\ |  | ||||||
|       int sU = ss/Ls;							\ |  | ||||||
|       WilsonKernels<Impl>::A(st_perm,st_p,U_p,buf,sF,sU,in_p,out_p);	\ |  | ||||||
|     });									\ |  | ||||||
|   accelerator_barrier(); |  | ||||||
|  |  | ||||||
| #define KERNEL_CALLNB(A)						\ |  | ||||||
|   const uint64_t    NN = Nsite*Ls;					\ |   const uint64_t    NN = Nsite*Ls;					\ | ||||||
|   accelerator_forNB( ss, NN, Simd::Nsimd(), {				\ |   accelerator_forNB( ss, NN, Simd::Nsimd(), {				\ | ||||||
|       int sF = ss;							\ |       int sF = ss;							\ | ||||||
| @@ -459,24 +445,20 @@ void WilsonKernels<Impl>::DhopKernel(int Opt,StencilImpl &st,  DoubledGaugeField | |||||||
|  |  | ||||||
|    if( interior && exterior ) { |    if( interior && exterior ) { | ||||||
|      if (Opt == WilsonKernelsStatic::OptGeneric    ) { KERNEL_CALL(GenericDhopSite); return;} |      if (Opt == WilsonKernelsStatic::OptGeneric    ) { KERNEL_CALL(GenericDhopSite); return;} | ||||||
| #ifdef SYCL_HACK      |  | ||||||
|      if (Opt == WilsonKernelsStatic::OptHandUnroll ) { KERNEL_CALL_TMP(HandDhopSiteSycl);    return; } |  | ||||||
| #else |  | ||||||
|      if (Opt == WilsonKernelsStatic::OptHandUnroll ) { KERNEL_CALL(HandDhopSite);    return;} |  | ||||||
| #endif      |  | ||||||
| #ifndef GRID_CUDA | #ifndef GRID_CUDA | ||||||
|  |      if (Opt == WilsonKernelsStatic::OptHandUnroll ) { KERNEL_CALL(HandDhopSite);    return;} | ||||||
|      if (Opt == WilsonKernelsStatic::OptInlineAsm  ) {  ASM_CALL(AsmDhopSite);    return;} |      if (Opt == WilsonKernelsStatic::OptInlineAsm  ) {  ASM_CALL(AsmDhopSite);    return;} | ||||||
| #endif | #endif | ||||||
|    } else if( interior ) { |    } else if( interior ) { | ||||||
|      if (Opt == WilsonKernelsStatic::OptGeneric    ) { KERNEL_CALLNB(GenericDhopSiteInt); return;} |      if (Opt == WilsonKernelsStatic::OptGeneric    ) { KERNEL_CALLNB(GenericDhopSiteInt); return;} | ||||||
|      if (Opt == WilsonKernelsStatic::OptHandUnroll ) { KERNEL_CALLNB(HandDhopSiteInt);    return;} |  | ||||||
| #ifndef GRID_CUDA | #ifndef GRID_CUDA | ||||||
|  |      if (Opt == WilsonKernelsStatic::OptHandUnroll ) { KERNEL_CALLNB(HandDhopSiteInt);    return;} | ||||||
|      if (Opt == WilsonKernelsStatic::OptInlineAsm  ) {  ASM_CALL(AsmDhopSiteInt);    return;} |      if (Opt == WilsonKernelsStatic::OptInlineAsm  ) {  ASM_CALL(AsmDhopSiteInt);    return;} | ||||||
| #endif | #endif | ||||||
|    } else if( exterior ) { |    } else if( exterior ) { | ||||||
|      if (Opt == WilsonKernelsStatic::OptGeneric    ) { KERNEL_CALL(GenericDhopSiteExt); return;} |      if (Opt == WilsonKernelsStatic::OptGeneric    ) { KERNEL_CALL(GenericDhopSiteExt); return;} | ||||||
|      if (Opt == WilsonKernelsStatic::OptHandUnroll ) { KERNEL_CALL(HandDhopSiteExt);    return;} |  | ||||||
| #ifndef GRID_CUDA | #ifndef GRID_CUDA | ||||||
|  |      if (Opt == WilsonKernelsStatic::OptHandUnroll ) { KERNEL_CALL(HandDhopSiteExt);    return;} | ||||||
|      if (Opt == WilsonKernelsStatic::OptInlineAsm  ) {  ASM_CALL(AsmDhopSiteExt);    return;} |      if (Opt == WilsonKernelsStatic::OptInlineAsm  ) {  ASM_CALL(AsmDhopSiteExt);    return;} | ||||||
| #endif | #endif | ||||||
|    } |    } | ||||||
| @@ -494,20 +476,20 @@ void WilsonKernels<Impl>::DhopKernel(int Opt,StencilImpl &st,  DoubledGaugeField | |||||||
|  |  | ||||||
|    if( interior && exterior ) { |    if( interior && exterior ) { | ||||||
|      if (Opt == WilsonKernelsStatic::OptGeneric    ) { KERNEL_CALL(GenericDhopSiteDag); return;} |      if (Opt == WilsonKernelsStatic::OptGeneric    ) { KERNEL_CALL(GenericDhopSiteDag); return;} | ||||||
|      if (Opt == WilsonKernelsStatic::OptHandUnroll ) { KERNEL_CALL(HandDhopSiteDag);    return;} |  | ||||||
| #ifndef GRID_CUDA | #ifndef GRID_CUDA | ||||||
|  |      if (Opt == WilsonKernelsStatic::OptHandUnroll ) { KERNEL_CALL(HandDhopSiteDag);    return;} | ||||||
|      if (Opt == WilsonKernelsStatic::OptInlineAsm  ) {  ASM_CALL(AsmDhopSiteDag);     return;} |      if (Opt == WilsonKernelsStatic::OptInlineAsm  ) {  ASM_CALL(AsmDhopSiteDag);     return;} | ||||||
| #endif | #endif | ||||||
|    } else if( interior ) { |    } else if( interior ) { | ||||||
|      if (Opt == WilsonKernelsStatic::OptGeneric    ) { KERNEL_CALL(GenericDhopSiteDagInt); return;} |      if (Opt == WilsonKernelsStatic::OptGeneric    ) { KERNEL_CALL(GenericDhopSiteDagInt); return;} | ||||||
|      if (Opt == WilsonKernelsStatic::OptHandUnroll ) { KERNEL_CALL(HandDhopSiteDagInt);    return;} |  | ||||||
| #ifndef GRID_CUDA | #ifndef GRID_CUDA | ||||||
|  |      if (Opt == WilsonKernelsStatic::OptHandUnroll ) { KERNEL_CALL(HandDhopSiteDagInt);    return;} | ||||||
|      if (Opt == WilsonKernelsStatic::OptInlineAsm  ) {  ASM_CALL(AsmDhopSiteDagInt);     return;} |      if (Opt == WilsonKernelsStatic::OptInlineAsm  ) {  ASM_CALL(AsmDhopSiteDagInt);     return;} | ||||||
| #endif | #endif | ||||||
|    } else if( exterior ) { |    } else if( exterior ) { | ||||||
|      if (Opt == WilsonKernelsStatic::OptGeneric    ) { KERNEL_CALL(GenericDhopSiteDagExt); return;} |      if (Opt == WilsonKernelsStatic::OptGeneric    ) { KERNEL_CALL(GenericDhopSiteDagExt); return;} | ||||||
|      if (Opt == WilsonKernelsStatic::OptHandUnroll ) { KERNEL_CALL(HandDhopSiteDagExt);    return;} |  | ||||||
| #ifndef GRID_CUDA | #ifndef GRID_CUDA | ||||||
|  |      if (Opt == WilsonKernelsStatic::OptHandUnroll ) { KERNEL_CALL(HandDhopSiteDagExt);    return;} | ||||||
|      if (Opt == WilsonKernelsStatic::OptInlineAsm  ) {  ASM_CALL(AsmDhopSiteDagExt);     return;} |      if (Opt == WilsonKernelsStatic::OptInlineAsm  ) {  ASM_CALL(AsmDhopSiteDagExt);     return;} | ||||||
| #endif | #endif | ||||||
|    } |    } | ||||||
|   | |||||||
| @@ -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 | ||||||
| @@ -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,13 +2,14 @@ | |||||||
| 
 | 
 | ||||||
| Grid physics library, www.github.com/paboyle/Grid | Grid physics library, www.github.com/paboyle/Grid | ||||||
| 
 | 
 | ||||||
| Source file: ./lib/qcd/action/gauge/Gauge.cc | Source file: ./lib/qcd/action/fermion/WilsonKernels.cc | ||||||
| 
 | 
 | ||||||
| Copyright (C) 2020 | Copyright (C) 2015, 2020 | ||||||
| 
 | 
 | ||||||
| Author: Peter Boyle <paboyle@ph.ed.ac.uk> | Author: Peter Boyle <paboyle@ph.ed.ac.uk> | ||||||
| Author: Peter Boyle <peterboyle@Peters-MacBook-Pro-2.local> | Author: Peter Boyle <peterboyle@Peters-MacBook-Pro-2.local> | ||||||
| Author: paboyle <paboyle@ph.ed.ac.uk> | 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 | 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 | it under the terms of the GNU General Public License as published by | ||||||
| @@ -29,10 +30,22 @@ directory | |||||||
| *************************************************************************************/ | *************************************************************************************/ | ||||||
| /*  END LEGAL */ | /*  END LEGAL */ | ||||||
| #include <Grid/qcd/action/fermion/FermionCore.h> | #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); | NAMESPACE_BEGIN(Grid); | ||||||
| 
 | 
 | ||||||
| std::vector<int> ConjugateGaugeImplBase::_conjDirs; | #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 | ||||||
| @@ -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 | 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 | 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 | 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., | with this program; if not, write to the Free Software Foundation, Inc., | ||||||
| 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. | 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 */ | /*  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 WilsonImplFH | ||||||
Some files were not shown because too many files have changed in this diff Show More
		Reference in New Issue
	
	Block a user