portelli/Grid
1
0
Fork 0
mirror of https://github.com/paboyle/Grid.git synced 2025-04-04 19:25:56 +01:00
Code Releases Activity

Fast init on GPU

Browse Source
This commit is contained in:
Peter Boyle 2024-08-27 11:22:33 -04:00
parent 6cfb90e99f
commit 06db4ddea2
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
Grid/lattice/Lattice_base.h
View File

@ -236,17 +236,17 @@ public:
template<class sobj> inline Lattice<vobj> & operator = (const sobj & r){ template<class sobj> inline Lattice<vobj> & operator = (const sobj & r){
vobj vtmp; vobj vtmp;
vtmp = r; vtmp = r;
#if 1 #if defined(GRID_HIP) || defined(GRID_CUDA) || defined (GRID_SYCL)
auto me = View(CpuWrite);
thread_for(ss,me.size(),{
me[ss]= r;
});
#else
auto me = View(AcceleratorWrite); auto me = View(AcceleratorWrite);
accelerator_for(ss,me.size(),vobj::Nsimd(),{ accelerator_for(ss,me.size(),vobj::Nsimd(),{
auto stmp=coalescedRead(vtmp); auto stmp=coalescedRead(vtmp);
coalescedWrite(me[ss],stmp); coalescedWrite(me[ss],stmp);
}); });
#else
auto me = View(CpuWrite);
thread_for(ss,me.size(),{
me[ss]= r;
});
#endif #endif
me.ViewClose(); me.ViewClose();
return *this; return *this;