diff --git a/RAJAStream.cpp b/RAJAStream.cpp index 0ee2390..33687a1 100644 --- a/RAJAStream.cpp +++ b/RAJAStream.cpp @@ -21,6 +21,12 @@ RAJAStream::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(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);