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

Must offload the Coarsened matrix if Stencil buffers are device resident

Browse Source
This commit is contained in:
Peter Boyle 2019-12-10 19:32:12 -05:00
parent 9b6b0caa55
commit f2a4f13111
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
Grid/algorithms/CoarsenedMatrix.h
View File

@ -128,7 +128,7 @@ public:
for(int i=0;i<nbasis;i++){ for(int i=0;i<nbasis;i++){
blockProject(iProj,subspace[i],subspace); blockProject(iProj,subspace[i],subspace);
eProj=Zero(); eProj=Zero();
thread_for(ss, CoarseGrid->oSites(),{ accelerator_for(ss, CoarseGrid->oSites(),1,{
eProj[ss](i)=CComplex(1.0); eProj[ss](i)=CComplex(1.0);
}); });
eProj=eProj - iProj; eProj=eProj - iProj;
@ -307,10 +307,12 @@ public:
RealD Nin = norm2(in); RealD Nin = norm2(in);
SimpleCompressor<siteVector> compressor; SimpleCompressor<siteVector> compressor;
Stencil.HaloExchange(in,compressor); Stencil.HaloExchange(in,compressor);
auto in_v = in.View(); auto in_v = in.View();
auto out_v = out.View(); auto out_v = out.View();
thread_for(ss,Grid()->oSites(),{ accelerator_for(ss,Grid()->oSites(),1,{
siteVector res = Zero(); siteVector res = Zero();
siteVector nbr; siteVector nbr;
int ptype; int ptype;
@ -331,6 +333,7 @@ public:
} }
vstream(out_v[ss],res); vstream(out_v[ss],res);
}); });
RealD Nout= norm2(out); RealD Nout= norm2(out);
return Nout; return Nout;
}; };
@ -356,6 +359,7 @@ public:
conformable(in.Grid(),out.Grid()); conformable(in.Grid(),out.Grid());
SimpleCompressor<siteVector> compressor; SimpleCompressor<siteVector> compressor;
Stencil.HaloExchange(in,compressor); Stencil.HaloExchange(in,compressor);
auto point = [dir, disp](){ auto point = [dir, disp](){
@ -367,7 +371,7 @@ public:
auto out_v = out.View(); auto out_v = out.View();
auto in_v = in.View(); auto in_v = in.View();
thread_for(ss,Grid()->oSites(),{ accelerator_for(ss,Grid()->oSites(),1,{
siteVector res = Zero(); siteVector res = Zero();
siteVector nbr; siteVector nbr;
int ptype; int ptype;