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

Create CUDA device buffers

Browse Source
This commit is contained in:
Tom Deakin 2016-04-26 14:50:22 +01:00
parent 03b01e190f
commit ee4820b5e4
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/CUDAStream.cu
View File

@ -16,7 +16,11 @@ CUDAStream<T>::CUDAStream(const unsigned int ARRAY_SIZE)
{ {
// Create device buffers // Create device buffers
cudaMalloc(&d_a, ARRAY_SIZE*sizeof(T)); 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> template <typename T>