cory-master/BabelStream
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[RAJA] Declare pointers using RAJA_RESTRICT

Browse Source
This commit is contained in:
Tom Deakin 2017-04-06 10:15:11 +01:00
parent 1bd4adfe7b
commit 5f9b288570
1 changed files with 15 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
RAJAStream.cpp
View File

@ -50,9 +50,9 @@ RAJAStream<T>::~RAJAStream()
template <class T> template <class T>
void RAJAStream<T>::init_arrays(T initA, T initB, T initC) void RAJAStream<T>::init_arrays(T initA, T initB, T initC)
{ {
T* a = d_a; T* RAJA_RESTRICT a = d_a;
T* b = d_b; T* RAJA_RESTRICT b = d_b;
T* c = d_c; T* RAJA_RESTRICT c = d_c;
forall<policy>(index_set, [=] RAJA_DEVICE (int index) forall<policy>(index_set, [=] RAJA_DEVICE (int index)
{ {
a[index] = initA; a[index] = initA;
@ -73,8 +73,8 @@ void RAJAStream<T>::read_arrays(
template <class T> template <class T>
void RAJAStream<T>::copy() void RAJAStream<T>::copy()
{ {
T* a = d_a; T* RAJA_RESTRICT a = d_a;
T* c = d_c; T* RAJA_RESTRICT c = d_c;
forall<policy>(index_set, [=] RAJA_DEVICE (int index) forall<policy>(index_set, [=] RAJA_DEVICE (int index)
{ {
c[index] = a[index]; c[index] = a[index];
@ -84,8 +84,8 @@ void RAJAStream<T>::copy()
template <class T> template <class T>
void RAJAStream<T>::mul() void RAJAStream<T>::mul()
{ {
T* b = d_b; T* RAJA_RESTRICT b = d_b;
T* c = d_c; T* RAJA_RESTRICT c = d_c;
const T scalar = startScalar; const T scalar = startScalar;
forall<policy>(index_set, [=] RAJA_DEVICE (int index) forall<policy>(index_set, [=] RAJA_DEVICE (int index)
{ {
@ -96,9 +96,9 @@ void RAJAStream<T>::mul()
template <class T> template <class T>
void RAJAStream<T>::add() void RAJAStream<T>::add()
{ {
T* a = d_a; T* RAJA_RESTRICT a = d_a;
T* b = d_b; T* RAJA_RESTRICT b = d_b;
T* c = d_c; T* RAJA_RESTRICT c = d_c;
forall<policy>(index_set, [=] RAJA_DEVICE (int index) forall<policy>(index_set, [=] RAJA_DEVICE (int index)
{ {
c[index] = a[index] + b[index]; c[index] = a[index] + b[index];
@ -108,9 +108,9 @@ void RAJAStream<T>::add()
template <class T> template <class T>
void RAJAStream<T>::triad() void RAJAStream<T>::triad()
{ {
T* a = d_a; T* RAJA_RESTRICT a = d_a;
T* b = d_b; T* RAJA_RESTRICT b = d_b;
T* c = d_c; T* RAJA_RESTRICT c = d_c;
const T scalar = startScalar; const T scalar = startScalar;
forall<policy>(index_set, [=] RAJA_DEVICE (int index) forall<policy>(index_set, [=] RAJA_DEVICE (int index)
{ {
@ -121,8 +121,8 @@ void RAJAStream<T>::triad()
template <class T> template <class T>
T RAJAStream<T>::dot() T RAJAStream<T>::dot()
{ {
T* a = d_a; T* RAJA_RESTRICT a = d_a;
T* b = d_b; T* RAJA_RESTRICT b = d_b;
RAJA::ReduceSum<reduce_policy, T> sum(0.0); RAJA::ReduceSum<reduce_policy, T> sum(0.0);