From 7c28a6386b1905e154815afcfc0f26e1458aaa3d Mon Sep 17 00:00:00 2001
From: Matthew Martineau <m.martineau@hotmail.co.uk>
Date: Thu, 5 May 2016 17:22:29 +0100
Subject: [PATCH] Added the Kokkos and RAJA implementations

---
 main.cpp | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/main.cpp b/main.cpp
index 96f4e5c..1a92ebc 100644
--- a/main.cpp
+++ b/main.cpp
@@ -22,9 +22,12 @@
 #include "CUDAStream.h"
 #elif defined(OCL)
 #include "OCLStream.h"
+#elif defined(RAJA)
+#include "RAJAStream.hpp"
+#elif defined(KOKKOS)
+#include "KOKKOSStream.hpp"
 #endif
 
-
 unsigned int ARRAY_SIZE = 52428800;
 unsigned int num_times = 10;
 unsigned int deviceIndex = 0;
@@ -80,6 +83,14 @@ void run()
   // Use the OpenCL implementation
   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
 
   stream->write_arrays(a, b, c);