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

Replace cudaDeviceSyncronise with Kokkos::fence() function

Browse Source
This commit is contained in:
Tom Deakin 2016-06-30 16:32:52 +01:00
parent 8be4b26bb1
commit 04f321db85
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
KOKKOSStream.cpp
View File

@ -74,7 +74,7 @@ void KOKKOSStream<T>::copy()
{ {
c[index] = a[index]; c[index] = a[index];
}); });
cudaDeviceSynchronize(); Kokkos::fence();
} }
template <class T> template <class T>
@ -89,7 +89,7 @@ void KOKKOSStream<T>::mul()
{ {
b[index] = scalar*c[index]; b[index] = scalar*c[index];
}); });
cudaDeviceSynchronize(); Kokkos::fence();
} }
template <class T> template <class T>
@ -103,8 +103,7 @@ void KOKKOSStream<T>::add()
{ {
c[index] = a[index] + b[index]; c[index] = a[index] + b[index];
}); });
Kokkos::fence();
cudaDeviceSynchronize();
} }
template <class T> template <class T>
@ -119,8 +118,7 @@ void KOKKOSStream<T>::triad()
{ {
a[index] = b[index] + scalar*c[index]; a[index] = b[index] + scalar*c[index];
}); });
Kokkos::fence();
cudaDeviceSynchronize();
} }
void listDevices(void) void listDevices(void)