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

Move site and Ls looping into the kernels

Browse Source
This commit is contained in:
paboyle 2018-06-27 21:28:48 +01:00
parent 38e2a32ac9
commit 31f65beac8
1 changed files with 3 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
lib/qcd/action/fermion/WilsonFermion5D.cc
View File

@ -513,23 +513,12 @@ void WilsonFermion5D<Impl>::DhopInternalSerialComms(StencilImpl & st, LebesgueOr
DhopComputeTime-=usecond();
// Dhop takes the 4d grid from U, and makes a 5d index for fermion
auto U_v = U.View();
int Opt = WilsonKernelsStatic::Opt;
auto U_v = U.View();
auto in_v = in.View();
auto out_v = out.View();
if (dag == DaggerYes) {
accelerator_loop( ss, U_v, {
int sU = ss;
int sF = LLs * sU;
Kernels::DhopSiteDag(Opt,st,U_v,st.CommBuf(),sF,sU,LLs,1,in_v,out_v);
});
Kernels::DhopDag(Opt,st,U,st.CommBuf(),LLs,U_v.end(),in,out);
} else {
accelerator_loop( ss, U_v , {
int sU = ss;
int sF = LLs * sU;
Kernels::DhopSite(Opt,st,U_v,st.CommBuf(),sF,sU,LLs,1,in_v,out_v);
});
Kernels::Dhop(Opt,st,U,st.CommBuf(),LLs,U_v.end(),in,out);
}
DhopComputeTime+=usecond();
}