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