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::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