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

[RAJA] Use Index_type for iterator index type instead of hardcoding int

Browse Source
This commit is contained in:
Tom Deakin 2017-04-06 10:36:01 +01:00
parent 3331f62f42
commit c534600d04
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
RAJAStream.cpp
View File

@ -53,7 +53,7 @@ void RAJAStream<T>::init_arrays(T initA, T initB, T initC)
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)
forall<policy>(index_set, [=] RAJA_DEVICE (RAJA::Index_type index)
{
a[index] = initA;
b[index] = initB;
@ -75,7 +75,7 @@ void RAJAStream<T>::copy()
{
T* RAJA_RESTRICT a = d_a;
T* RAJA_RESTRICT c = d_c;
forall<policy>(index_set, [=] RAJA_DEVICE (int index)
forall<policy>(index_set, [=] RAJA_DEVICE (RAJA::Index_type index)
{
c[index] = a[index];
});
@ -87,7 +87,7 @@ void RAJAStream<T>::mul()
T* RAJA_RESTRICT b = d_b;
T* RAJA_RESTRICT c = d_c;
const T scalar = startScalar;
forall<policy>(index_set, [=] RAJA_DEVICE (int index)
forall<policy>(index_set, [=] RAJA_DEVICE (RAJA::Index_type index)
{
b[index] = scalar*c[index];
});
@ -99,7 +99,7 @@ void RAJAStream<T>::add()
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)
forall<policy>(index_set, [=] RAJA_DEVICE (RAJA::Index_type index)
{
c[index] = a[index] + b[index];
});
@ -112,7 +112,7 @@ void RAJAStream<T>::triad()
T* RAJA_RESTRICT b = d_b;
T* RAJA_RESTRICT c = d_c;
const T scalar = startScalar;
forall<policy>(index_set, [=] RAJA_DEVICE (int index)
forall<policy>(index_set, [=] RAJA_DEVICE (RAJA::Index_type index)
{
a[index] = b[index] + scalar*c[index];
});
@ -126,7 +126,7 @@ T RAJAStream<T>::dot()
RAJA::ReduceSum<reduce_policy, T> sum(0.0);
forall<policy>(index_set, [=] RAJA_DEVICE (int index)
forall<policy>(index_set, [=] RAJA_DEVICE (RAJA::Index_type index)
{
sum += a[index] * b[index];
});