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

Added the Kokkos and RAJA implementations

Browse Source
This commit is contained in:
Matthew Martineau 2016-05-05 17:22:29 +01:00
parent 530b2adda2
commit 7c28a6386b
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
main.cpp
View File

@ -22,9 +22,12 @@
#include "CUDAStream.h" #include "CUDAStream.h"
#elif defined(OCL) #elif defined(OCL)
#include "OCLStream.h" #include "OCLStream.h"
#elif defined(RAJA)
#include "RAJAStream.hpp"
#elif defined(KOKKOS)
#include "KOKKOSStream.hpp"
#endif #endif
unsigned int ARRAY_SIZE = 52428800; unsigned int ARRAY_SIZE = 52428800;
unsigned int num_times = 10; unsigned int num_times = 10;
unsigned int deviceIndex = 0; unsigned int deviceIndex = 0;
@ -80,6 +83,14 @@ void run()
// Use the OpenCL implementation // Use the OpenCL implementation
stream = new OCLStream<T>(ARRAY_SIZE, deviceIndex); stream = new OCLStream<T>(ARRAY_SIZE, deviceIndex);
#elif defined(RAJA)
// Use the RAJA implementation
stream = new RAJAStream<T>(ARRAY_SIZE, deviceIndex);
#elif defined(KOKKOS)
// Use the Kokkos implementation
stream = new KOKKOSStream<T>(ARRAY_SIZE, deviceIndex);
#endif #endif
stream->write_arrays(a, b, c); stream->write_arrays(a, b, c);