From e9c8ba5ef729962e86eb5c7a66ef9d352cbb3e3c Mon Sep 17 00:00:00 2001
From: paboyle <paboyle@ph.ed.ac.uk>
Date: Wed, 24 Jan 2018 13:39:54 +0000
Subject: [PATCH] Accelerator loosp

---
 lib/lattice/Lattice_trace.h | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/lib/lattice/Lattice_trace.h b/lib/lattice/Lattice_trace.h
index f97c8d97..f0951cd4 100644
--- a/lib/lattice/Lattice_trace.h
+++ b/lib/lattice/Lattice_trace.h
@@ -42,9 +42,9 @@ inline auto trace(const Lattice<vobj> &lhs)
   -> Lattice<decltype(trace(lhs._odata[0]))>
 {
   Lattice<decltype(trace(lhs._odata[0]))> ret(lhs._grid);
-  parallel_for(int ss=0;ss<lhs._grid->oSites();ss++){
+  accelerator_loop( ss, lhs, {
     ret._odata[ss] = trace(lhs._odata[ss]);
-  }
+  });
   return ret;
 };
     
@@ -55,9 +55,9 @@ template<int Index,class vobj>
 inline auto TraceIndex(const Lattice<vobj> &lhs) -> Lattice<decltype(traceIndex<Index>(lhs._odata[0]))>
 {
   Lattice<decltype(traceIndex<Index>(lhs._odata[0]))> ret(lhs._grid);
-  parallel_for(int ss=0;ss<lhs._grid->oSites();ss++){
+  accelerator_loop( ss, lhs, {
     ret._odata[ss] = traceIndex<Index>(lhs._odata[ss]);
-  }
+  });
   return ret;
 };