From 86bb0cc24b3345a9cc76331d7d9f1ba1b0b25b86 Mon Sep 17 00:00:00 2001 From: Peter Boyle Date: Wed, 10 Jun 2020 20:00:00 -0400 Subject: [PATCH] Keep on GPU --- Grid/qcd/utils/SUn.h | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/Grid/qcd/utils/SUn.h b/Grid/qcd/utils/SUn.h index 476c3d40..6f50af04 100644 --- a/Grid/qcd/utils/SUn.h +++ b/Grid/qcd/utils/SUn.h @@ -223,10 +223,10 @@ public: int i0, i1; su2SubGroupIndex(i0, i1, su2_index); - autoView( subgroup_v , subgroup,CpuWrite); - autoView( source_v , source,CpuRead); - autoView( Determinant_v , Determinant,CpuWrite); - thread_for(ss, grid->oSites(), { + autoView( subgroup_v , subgroup,AcceleratorWrite); + autoView( source_v , source,AcceleratorRead); + autoView( Determinant_v , Determinant,AcceleratorWrite); + accelerator_for(ss, grid->oSites(), 1, { subgroup_v[ss]()()(0, 0) = source_v[ss]()()(i0, i0); subgroup_v[ss]()()(0, 1) = source_v[ss]()()(i0, i1); @@ -257,9 +257,9 @@ public: su2SubGroupIndex(i0, i1, su2_index); dest = 1.0; // start out with identity - autoView( dest_v , dest, CpuWrite); - autoView( subgroup_v, subgroup, CpuRead); - thread_for(ss, grid->oSites(), + autoView( dest_v , dest, AcceleratorWrite); + autoView( subgroup_v, subgroup, AcceleratorRead); + acccelerator_for(ss, grid->oSites(),1, { dest_v[ss]()()(i0, i0) = subgroup_v[ss]()()(0, 0); dest_v[ss]()()(i0, i1) = subgroup_v[ss]()()(0, 1);