portelli/Grid
1
0
Fork 0
mirror of https://github.com/paboyle/Grid.git synced 2025-04-09 21:50:45 +01:00
Code Releases Activity

Accelerator loops

Browse Source
This commit is contained in:
paboyle 2018-01-24 13:41:12 +00:00
parent ba5ea5830b
commit e737591918
1 changed files with 8 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
lib/lattice/Lattice_unary.h
View File

@ -37,18 +37,18 @@ template<class obj> Lattice<obj> pow(const Lattice<obj> &rhs,RealD y){
Lattice<obj> ret(rhs._grid); Lattice<obj> ret(rhs._grid);
ret.checkerboard = rhs.checkerboard; ret.checkerboard = rhs.checkerboard;
conformable(ret,rhs); conformable(ret,rhs);
parallel_for(int ss=0;ss<rhs._grid->oSites();ss++){ accelerator_loop(ss,rhs,{
ret._odata[ss]=pow(rhs._odata[ss],y); ret._odata[ss]=pow(rhs._odata[ss],y);
} });
return ret; return ret;
} }
template<class obj> Lattice<obj> mod(const Lattice<obj> &rhs,Integer y){ template<class obj> Lattice<obj> mod(const Lattice<obj> &rhs,Integer y){
Lattice<obj> ret(rhs._grid); Lattice<obj> ret(rhs._grid);
ret.checkerboard = rhs.checkerboard; ret.checkerboard = rhs.checkerboard;
conformable(ret,rhs); conformable(ret,rhs);
parallel_for(int ss=0;ss<rhs._grid->oSites();ss++){ accelerator_loop(ss,rhs,{
ret._odata[ss]=mod(rhs._odata[ss],y); ret._odata[ss]=mod(rhs._odata[ss],y);
} });
return ret; return ret;
} }
@ -56,9 +56,9 @@ template<class obj> Lattice<obj> div(const Lattice<obj> &rhs,Integer y){
Lattice<obj> ret(rhs._grid); Lattice<obj> ret(rhs._grid);
ret.checkerboard = rhs.checkerboard; ret.checkerboard = rhs.checkerboard;
conformable(ret,rhs); conformable(ret,rhs);
parallel_for(int ss=0;ss<rhs._grid->oSites();ss++){ accelerator_loop(ss,rhs,{
ret._odata[ss]=div(rhs._odata[ss],y); ret._odata[ss]=div(rhs._odata[ss],y);
} });
return ret; return ret;
} }
@ -66,10 +66,9 @@ template<class obj> Lattice<obj> expMat(const Lattice<obj> &rhs, RealD alpha, In
Lattice<obj> ret(rhs._grid); Lattice<obj> ret(rhs._grid);
ret.checkerboard = rhs.checkerboard; ret.checkerboard = rhs.checkerboard;
conformable(ret,rhs); conformable(ret,rhs);
parallel_for(int ss=0;ss<rhs._grid->oSites();ss++){ accelerator_loop(ss,rhs,{
ret._odata[ss]=Exponentiate(rhs._odata[ss],alpha, Nexp); ret._odata[ss]=Exponentiate(rhs._odata[ss],alpha, Nexp);
} });
return ret; return ret;
} }