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

[Kokkos] Use long for iterator variable

Browse Source
This commit is contained in:
Tom Deakin 2017-04-05 21:57:55 +01:00
parent bf57cf578d
commit d9dfc3f552
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
KOKKOSStream.cpp
View File

@ -39,7 +39,7 @@ void KOKKOSStream<T>::init_arrays(T initA, T initB, T initC)
View<double*, DEVICE> a(*d_a); View<double*, DEVICE> a(*d_a);
View<double*, DEVICE> b(*d_b); View<double*, DEVICE> b(*d_b);
View<double*, DEVICE> c(*d_c); View<double*, DEVICE> c(*d_c);
parallel_for(array_size, KOKKOS_LAMBDA (const int index) parallel_for(array_size, KOKKOS_LAMBDA (const long index)
{ {
a[index] = initA; a[index] = initA;
b[index] = initB; b[index] = initB;
@ -70,7 +70,7 @@ void KOKKOSStream<T>::copy()
View<double*, DEVICE> b(*d_b); View<double*, DEVICE> b(*d_b);
View<double*, DEVICE> c(*d_c); View<double*, DEVICE> c(*d_c);
parallel_for(array_size, KOKKOS_LAMBDA (const int index) parallel_for(array_size, KOKKOS_LAMBDA (const long index)
{ {
c[index] = a[index]; c[index] = a[index];
}); });
@ -85,7 +85,7 @@ void KOKKOSStream<T>::mul()
View<double*, DEVICE> c(*d_c); View<double*, DEVICE> c(*d_c);
const T scalar = startScalar; const T scalar = startScalar;
parallel_for(array_size, KOKKOS_LAMBDA (const int index) parallel_for(array_size, KOKKOS_LAMBDA (const long index)
{ {
b[index] = scalar*c[index]; b[index] = scalar*c[index];
}); });
@ -99,7 +99,7 @@ void KOKKOSStream<T>::add()
View<double*, DEVICE> b(*d_b); View<double*, DEVICE> b(*d_b);
View<double*, DEVICE> c(*d_c); View<double*, DEVICE> c(*d_c);
parallel_for(array_size, KOKKOS_LAMBDA (const int index) parallel_for(array_size, KOKKOS_LAMBDA (const long index)
{ {
c[index] = a[index] + b[index]; c[index] = a[index] + b[index];
}); });
@ -114,7 +114,7 @@ void KOKKOSStream<T>::triad()
View<double*, DEVICE> c(*d_c); View<double*, DEVICE> c(*d_c);
const T scalar = startScalar; const T scalar = startScalar;
parallel_for(array_size, KOKKOS_LAMBDA (const int index) parallel_for(array_size, KOKKOS_LAMBDA (const long index)
{ {
a[index] = b[index] + scalar*c[index]; a[index] = b[index] + scalar*c[index];
}); });
@ -129,7 +129,7 @@ T KOKKOSStream<T>::dot()
T sum = 0.0; T sum = 0.0;
parallel_reduce(array_size, KOKKOS_LAMBDA (const int index, double &tmp) parallel_reduce(array_size, KOKKOS_LAMBDA (const long index, double &tmp)
{ {
tmp += a[index] * b[index]; tmp += a[index] * b[index];
}, sum); }, sum);