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

[Kokkos] Use global defined scalar value

Browse Source
This commit is contained in:
Tom Deakin 2016-10-24 13:25:54 +01:00
parent ac6158fa31
commit d1bebf12d9
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
KOKKOSStream.cpp
View File

@ -84,7 +84,7 @@ void KOKKOSStream<T>::mul()
View<double*, DEVICE> b(*d_b); View<double*, DEVICE> b(*d_b);
View<double*, DEVICE> c(*d_c); View<double*, DEVICE> c(*d_c);
const T scalar = 0.3; const T scalar = startScalar;
parallel_for(array_size, KOKKOS_LAMBDA (const int index) parallel_for(array_size, KOKKOS_LAMBDA (const int index)
{ {
b[index] = scalar*c[index]; b[index] = scalar*c[index];
@ -113,7 +113,7 @@ void KOKKOSStream<T>::triad()
View<double*, DEVICE> b(*d_b); View<double*, DEVICE> b(*d_b);
View<double*, DEVICE> c(*d_c); View<double*, DEVICE> c(*d_c);
const T scalar = 0.3; const T scalar = startScalar;
parallel_for(array_size, KOKKOS_LAMBDA (const int index) parallel_for(array_size, KOKKOS_LAMBDA (const int index)
{ {
a[index] = b[index] + scalar*c[index]; a[index] = b[index] + scalar*c[index];