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

Added CUDA device syncs to force proper timing

Browse Source
This commit is contained in:
Matthew Martineau 2016-05-06 21:05:20 +01:00
parent 894829cb05
commit 0f0454ec29
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
KOKKOSStream.cpp
View File

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