From ee4820b5e4763573211146a049ed392fb5390380 Mon Sep 17 00:00:00 2001
From: Tom Deakin <tom.deakin@bristol.ac.uk>
Date: Tue, 26 Apr 2016 14:50:22 +0100
Subject: [PATCH] Create CUDA device buffers

---
 src/CUDAStream.cu | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/CUDAStream.cu b/src/CUDAStream.cu
index 4305ceb..66304d1 100644
--- a/src/CUDAStream.cu
+++ b/src/CUDAStream.cu
@@ -16,7 +16,11 @@ CUDAStream<T>::CUDAStream(const unsigned int ARRAY_SIZE)
 {
   // Create device buffers
   cudaMalloc(&d_a, ARRAY_SIZE*sizeof(T));
-   
+  check_error();
+  cudaMalloc(&d_b, ARRAY_SIZE*sizeof(T));
+  check_error();
+  cudaMalloc(&d_c, ARRAY_SIZE*sizeof(T));
+  check_error();
 }
 
 template <typename T>