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