portelli/Grid
1
0
Fork 0
mirror of https://github.com/paboyle/Grid.git synced 2024-11-10 07:55:35 +00:00
Code Releases Activity

Coalesced loops.

Browse Source
This commit is contained in:
Peter Boyle 2019-06-15 08:03:57 +01:00
parent 736358b0cb
commit 20359ca15f
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
Grid/lattice/Lattice_unary.h
View File

@ -38,8 +38,8 @@ template<class obj> Lattice<obj> pow(const Lattice<obj> &rhs_i,RealD y){
auto rhs = rhs_i.View();
auto ret = ret_i.View();
ret.Checkerboard() = rhs.Checkerboard();
accelerator_loop(ss,rhs,{
ret[ss]=pow(rhs[ss],y);
accelerator_for(ss,rhs.size(),obj::Nsimd(),{
coalescedWrite(ret[ss],pow(rhs(ss),y));
});
return ret_i;
}
@ -48,8 +48,8 @@ template<class obj> Lattice<obj> mod(const Lattice<obj> &rhs_i,Integer y){
auto rhs = rhs_i.View();
auto ret = ret_i.View();
ret.Checkerboard() = rhs.Checkerboard();
accelerator_loop(ss,rhs,{
ret[ss]=mod(rhs[ss],y);
accelerator_for(ss,rhs.size(),obj::Nsimd(),{
coalescedWrite(ret[ss],mod(rhs(ss),y));
});
return ret_i;
}
@ -59,8 +59,8 @@ template<class obj> Lattice<obj> div(const Lattice<obj> &rhs_i,Integer y){
auto ret = ret_i.View();
auto rhs = rhs_i.View();
ret.Checkerboard() = rhs_i.Checkerboard();
accelerator_loop(ss,rhs,{
ret[ss]=div(rhs[ss],y);
accelerator_for(ss,rhs.size(),obj::Nsimd(),{
coalescedWrite(ret[ss],div(rhs(ss),y));
});
return ret_i;
}
@ -70,8 +70,8 @@ template<class obj> Lattice<obj> expMat(const Lattice<obj> &rhs_i, RealD alpha,
auto rhs = rhs_i.View();
auto ret = ret_i.View();
ret.Checkerboard() = rhs.Checkerboard();
accelerator_loop(ss,rhs,{
ret[ss]=Exponentiate(rhs[ss],alpha, Nexp);
accelerator_for(ss,rhs.size(),obj::Nsimd(),{
coalescedWrite(ret[ss],Exponentiate(rhs(ss),alpha, Nexp));
});
return ret_i;
}