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

Keep on GPU

Browse Source
This commit is contained in:
Peter Boyle 2020-06-10 20:00:00 -04:00
parent 84c19587e7
commit 86bb0cc24b
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
Grid/qcd/utils/SUn.h
View File

@ -223,10 +223,10 @@ public:
int i0, i1; int i0, i1;
su2SubGroupIndex(i0, i1, su2_index); su2SubGroupIndex(i0, i1, su2_index);
autoView( subgroup_v , subgroup,CpuWrite); autoView( subgroup_v , subgroup,AcceleratorWrite);
autoView( source_v , source,CpuRead); autoView( source_v , source,AcceleratorRead);
autoView( Determinant_v , Determinant,CpuWrite); autoView( Determinant_v , Determinant,AcceleratorWrite);
thread_for(ss, grid->oSites(), { accelerator_for(ss, grid->oSites(), 1, {
subgroup_v[ss]()()(0, 0) = source_v[ss]()()(i0, i0); subgroup_v[ss]()()(0, 0) = source_v[ss]()()(i0, i0);
subgroup_v[ss]()()(0, 1) = source_v[ss]()()(i0, i1); subgroup_v[ss]()()(0, 1) = source_v[ss]()()(i0, i1);
@ -257,9 +257,9 @@ public:
su2SubGroupIndex(i0, i1, su2_index); su2SubGroupIndex(i0, i1, su2_index);
dest = 1.0; // start out with identity dest = 1.0; // start out with identity
autoView( dest_v , dest, CpuWrite); autoView( dest_v , dest, AcceleratorWrite);
autoView( subgroup_v, subgroup, CpuRead); autoView( subgroup_v, subgroup, AcceleratorRead);
thread_for(ss, grid->oSites(), acccelerator_for(ss, grid->oSites(),1,
{ {
dest_v[ss]()()(i0, i0) = subgroup_v[ss]()()(0, 0); dest_v[ss]()()(i0, i0) = subgroup_v[ss]()()(0, 0);
dest_v[ss]()()(i0, i1) = subgroup_v[ss]()()(0, 1); dest_v[ss]()()(i0, i1) = subgroup_v[ss]()()(0, 1);