From 31cbbfc07e42b1740658fd101b21f780bd2e8309 Mon Sep 17 00:00:00 2001 From: paboyle Date: Sun, 14 Jan 2018 23:58:44 +0000 Subject: [PATCH] Namespace, indent --- lib/lattice/Lattice_peekpoke.h | 334 ++++++++++++++++----------------- 1 file changed, 167 insertions(+), 167 deletions(-) diff --git a/lib/lattice/Lattice_peekpoke.h b/lib/lattice/Lattice_peekpoke.h index 3d6268d2..a5dc808a 100644 --- a/lib/lattice/Lattice_peekpoke.h +++ b/lib/lattice/Lattice_peekpoke.h @@ -1,4 +1,4 @@ - /************************************************************************************* +/************************************************************************************* Grid physics library, www.github.com/paboyle/Grid @@ -25,8 +25,8 @@ Author: Peter Boyle 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. See the full license in the file "LICENSE" in the top level distribution directory - *************************************************************************************/ - /* END LEGAL */ +*************************************************************************************/ +/* END LEGAL */ #ifndef GRID_LATTICE_PEEK_H #define GRID_LATTICE_PEEK_H @@ -34,172 +34,172 @@ Author: Peter Boyle // Peeking and poking around /////////////////////////////////////////////// -namespace Grid { +NAMESPACE_BEGIN(Grid); - //////////////////////////////////////////////////////////////////////////////////////////////////// - // Peek internal indices of a Lattice object - //////////////////////////////////////////////////////////////////////////////////////////////////// - template - auto PeekIndex(const Lattice &lhs,int i) -> Lattice(lhs._odata[0],i))> - { - Lattice(lhs._odata[0],i))> ret(lhs._grid); - ret.checkerboard=lhs.checkerboard; - parallel_for(int ss=0;ssoSites();ss++){ - ret._odata[ss] = peekIndex(lhs._odata[ss],i); - } - return ret; - }; - template - auto PeekIndex(const Lattice &lhs,int i,int j) -> Lattice(lhs._odata[0],i,j))> - { - Lattice(lhs._odata[0],i,j))> ret(lhs._grid); - ret.checkerboard=lhs.checkerboard; - parallel_for(int ss=0;ssoSites();ss++){ - ret._odata[ss] = peekIndex(lhs._odata[ss],i,j); - } - return ret; - }; - - //////////////////////////////////////////////////////////////////////////////////////////////////// - // Poke internal indices of a Lattice object - //////////////////////////////////////////////////////////////////////////////////////////////////// - template - void PokeIndex(Lattice &lhs,const Lattice(lhs._odata[0],0))> & rhs,int i) - { - parallel_for(int ss=0;ssoSites();ss++){ - pokeIndex(lhs._odata[ss],rhs._odata[ss],i); - } - } - template - void PokeIndex(Lattice &lhs,const Lattice(lhs._odata[0],0,0))> & rhs,int i,int j) - { - parallel_for(int ss=0;ssoSites();ss++){ - pokeIndex(lhs._odata[ss],rhs._odata[ss],i,j); - } - } - - ////////////////////////////////////////////////////// - // Poke a scalar object into the SIMD array - ////////////////////////////////////////////////////// - template - void pokeSite(const sobj &s,Lattice &l,const std::vector &site){ - - GridBase *grid=l._grid; - - typedef typename vobj::scalar_type scalar_type; - typedef typename vobj::vector_type vector_type; - - int Nsimd = grid->Nsimd(); - - assert( l.checkerboard== l._grid->CheckerBoard(site)); - assert( sizeof(sobj)*Nsimd == sizeof(vobj)); - - int rank,odx,idx; - // Optional to broadcast from node 0. - grid->GlobalCoorToRankIndex(rank,odx,idx,site); - grid->Broadcast(grid->BossRank(),s); - - std::vector buf(Nsimd); - - // extract-modify-merge cycle is easiest way and this is not perf critical - if ( rank == grid->ThisRank() ) { - extract(l._odata[odx],buf); - buf[idx] = s; - merge(l._odata[odx],buf); - } - - return; - }; - - - ////////////////////////////////////////////////////////// - // Peek a scalar object from the SIMD array - ////////////////////////////////////////////////////////// - template - void peekSite(sobj &s,const Lattice &l,const std::vector &site){ - - GridBase *grid=l._grid; - - typedef typename vobj::scalar_type scalar_type; - typedef typename vobj::vector_type vector_type; - - int Nsimd = grid->Nsimd(); - - assert( l.checkerboard == l._grid->CheckerBoard(site)); - - int rank,odx,idx; - grid->GlobalCoorToRankIndex(rank,odx,idx,site); - - std::vector buf(Nsimd); - extract(l._odata[odx],buf); - - s = buf[idx]; - - grid->Broadcast(rank,s); - - return; - }; - - - ////////////////////////////////////////////////////////// - // Peek a scalar object from the SIMD array - ////////////////////////////////////////////////////////// - template - void peekLocalSite(sobj &s,const Lattice &l,std::vector &site){ - - GridBase *grid = l._grid; - - typedef typename vobj::scalar_type scalar_type; - typedef typename vobj::vector_type vector_type; - - int Nsimd = grid->Nsimd(); - - assert( l.checkerboard== l._grid->CheckerBoard(site)); - assert( sizeof(sobj)*Nsimd == sizeof(vobj)); - - static const int words=sizeof(vobj)/sizeof(vector_type); - int odx,idx; - idx= grid->iIndex(site); - odx= grid->oIndex(site); - - scalar_type * vp = (scalar_type *)&l._odata[odx]; - scalar_type * pt = (scalar_type *)&s; - - for(int w=0;w - void pokeLocalSite(const sobj &s,Lattice &l,std::vector &site){ - - GridBase *grid=l._grid; - - typedef typename vobj::scalar_type scalar_type; - typedef typename vobj::vector_type vector_type; - - int Nsimd = grid->Nsimd(); - - assert( l.checkerboard== l._grid->CheckerBoard(site)); - assert( sizeof(sobj)*Nsimd == sizeof(vobj)); - - static const int words=sizeof(vobj)/sizeof(vector_type); - int odx,idx; - idx= grid->iIndex(site); - odx= grid->oIndex(site); - - scalar_type * vp = (scalar_type *)&l._odata[odx]; - scalar_type * pt = (scalar_type *)&s; - - for(int w=0;w +auto PeekIndex(const Lattice &lhs,int i) -> Lattice(lhs._odata[0],i))> +{ + Lattice(lhs._odata[0],i))> ret(lhs._grid); + ret.checkerboard=lhs.checkerboard; + parallel_for(int ss=0;ssoSites();ss++){ + ret._odata[ss] = peekIndex(lhs._odata[ss],i); + } + return ret; +}; +template +auto PeekIndex(const Lattice &lhs,int i,int j) -> Lattice(lhs._odata[0],i,j))> +{ + Lattice(lhs._odata[0],i,j))> ret(lhs._grid); + ret.checkerboard=lhs.checkerboard; + parallel_for(int ss=0;ssoSites();ss++){ + ret._odata[ss] = peekIndex(lhs._odata[ss],i,j); + } + return ret; +}; +//////////////////////////////////////////////////////////////////////////////////////////////////// +// Poke internal indices of a Lattice object +//////////////////////////////////////////////////////////////////////////////////////////////////// +template +void PokeIndex(Lattice &lhs,const Lattice(lhs._odata[0],0))> & rhs,int i) +{ + parallel_for(int ss=0;ssoSites();ss++){ + pokeIndex(lhs._odata[ss],rhs._odata[ss],i); + } } +template +void PokeIndex(Lattice &lhs,const Lattice(lhs._odata[0],0,0))> & rhs,int i,int j) +{ + parallel_for(int ss=0;ssoSites();ss++){ + pokeIndex(lhs._odata[ss],rhs._odata[ss],i,j); + } +} + +////////////////////////////////////////////////////// +// Poke a scalar object into the SIMD array +////////////////////////////////////////////////////// +template +void pokeSite(const sobj &s,Lattice &l,const std::vector &site){ + + GridBase *grid=l._grid; + + typedef typename vobj::scalar_type scalar_type; + typedef typename vobj::vector_type vector_type; + + int Nsimd = grid->Nsimd(); + + assert( l.checkerboard== l._grid->CheckerBoard(site)); + assert( sizeof(sobj)*Nsimd == sizeof(vobj)); + + int rank,odx,idx; + // Optional to broadcast from node 0. + grid->GlobalCoorToRankIndex(rank,odx,idx,site); + grid->Broadcast(grid->BossRank(),s); + + std::vector buf(Nsimd); + + // extract-modify-merge cycle is easiest way and this is not perf critical + if ( rank == grid->ThisRank() ) { + extract(l._odata[odx],buf); + buf[idx] = s; + merge(l._odata[odx],buf); + } + + return; +}; + + +////////////////////////////////////////////////////////// +// Peek a scalar object from the SIMD array +////////////////////////////////////////////////////////// +template +void peekSite(sobj &s,const Lattice &l,const std::vector &site){ + + GridBase *grid=l._grid; + + typedef typename vobj::scalar_type scalar_type; + typedef typename vobj::vector_type vector_type; + + int Nsimd = grid->Nsimd(); + + assert( l.checkerboard == l._grid->CheckerBoard(site)); + + int rank,odx,idx; + grid->GlobalCoorToRankIndex(rank,odx,idx,site); + + std::vector buf(Nsimd); + extract(l._odata[odx],buf); + + s = buf[idx]; + + grid->Broadcast(rank,s); + + return; +}; + + +////////////////////////////////////////////////////////// +// Peek a scalar object from the SIMD array +////////////////////////////////////////////////////////// +template +void peekLocalSite(sobj &s,const Lattice &l,std::vector &site){ + + GridBase *grid = l._grid; + + typedef typename vobj::scalar_type scalar_type; + typedef typename vobj::vector_type vector_type; + + int Nsimd = grid->Nsimd(); + + assert( l.checkerboard== l._grid->CheckerBoard(site)); + assert( sizeof(sobj)*Nsimd == sizeof(vobj)); + + static const int words=sizeof(vobj)/sizeof(vector_type); + int odx,idx; + idx= grid->iIndex(site); + odx= grid->oIndex(site); + + scalar_type * vp = (scalar_type *)&l._odata[odx]; + scalar_type * pt = (scalar_type *)&s; + + for(int w=0;w +void pokeLocalSite(const sobj &s,Lattice &l,std::vector &site){ + + GridBase *grid=l._grid; + + typedef typename vobj::scalar_type scalar_type; + typedef typename vobj::vector_type vector_type; + + int Nsimd = grid->Nsimd(); + + assert( l.checkerboard== l._grid->CheckerBoard(site)); + assert( sizeof(sobj)*Nsimd == sizeof(vobj)); + + static const int words=sizeof(vobj)/sizeof(vector_type); + int odx,idx; + idx= grid->iIndex(site); + odx= grid->oIndex(site); + + scalar_type * vp = (scalar_type *)&l._odata[odx]; + scalar_type * pt = (scalar_type *)&s; + + for(int w=0;w