Create CUDA device buffers
This commit is contained in:
parent
03b01e190f
commit
ee4820b5e4
@ -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>
|
||||
|
||||
Loading…
Reference in New Issue
Block a user