From 3045208aaeb91c1dfd510266ce31028eab904433 Mon Sep 17 00:00:00 2001
From: James Price <j.price@bristol.ac.uk>
Date: Tue, 1 Nov 2016 16:18:43 +0000
Subject: [PATCH] [RAJA] Parallel first touch

---
 RAJAStream.cpp | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/RAJAStream.cpp b/RAJAStream.cpp
index 0ee2390..33687a1 100644
--- a/RAJAStream.cpp
+++ b/RAJAStream.cpp
@@ -21,6 +21,12 @@ RAJAStream<T>::RAJAStream(const unsigned int ARRAY_SIZE, const int device_index)
   d_a = new T[ARRAY_SIZE];
   d_b = new T[ARRAY_SIZE];
   d_c = new T[ARRAY_SIZE];
+  forall<policy>(index_set, [=] RAJA_DEVICE (int index)
+  {
+    d_a[index] = 0.0;
+    d_b[index] = 0.0;
+    d_c[index] = 0.0;
+  });
 #else
   cudaMallocManaged((void**)&d_a, sizeof(T)*ARRAY_SIZE, cudaMemAttachGlobal);
   cudaMallocManaged((void**)&d_b, sizeof(T)*ARRAY_SIZE, cudaMemAttachGlobal);