Add cuda constructor declaration and error checking function

src/CUDAStream.cu

@@ -1,6 +1,24 @@
 
 #include "CUDAStream.h"
 
+void check_error(void)
+{
+  cudaError_t err = cudaGetLastError();
+  if (err != cudaSuccess)
+  {
+    std::cerr << "Error: " << cudaGetErrorString(err) << std::endl;
+    exit(err);
+  }
+}
+
+template <class T>
+CUDAStream<T>::CUDAStream(const unsigned int ARRAY_SIZE)
+{
+  // Create device buffers
+  cudaMalloc(&d_a, ARRAY_SIZE*sizeof(T));
+   
+}
+
 template <typename T>
 __global__ void copy_kernel(const T * a, T * c)
 {

src/CUDAStream.h

@@ -1,4 +1,5 @@
 
+#include <iostream>
 
 #include "Stream.h"
 
@@ -13,6 +14,9 @@ class CUDAStream : public Stream<T>
 
 
   public:
+
+    CUDAStream(const unsigned int);
+
     void copy();
     void add();
     void mul();

src/main.cpp

@@ -25,7 +25,10 @@ int main(int argc, char *argv[])
   std::vector<double> c(ARRAY_SIZE, 0.0);
 
   Stream<double> *stream;
-  stream = new CUDAStream<double>();
+
+  // Use the CUDA implementation
+  stream = new CUDAStream<double>(ARRAY_SIZE);
+
   stream->copy();
 
   delete[] stream;