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

Add mul CUDA kernel

Browse Source
This commit is contained in:
Tom Deakin 2016-04-26 16:06:17 +01:00
parent dec0237353
commit 7a3a546a6e
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/CUDAStream.cu
View File

@ -82,10 +82,20 @@ void CUDAStream<T>::mul()
check_error(); check_error();
} }
template <typename T>
__global__ void add_kernel(const T * a, const T * b, T * c)
{
const int i = blockDim.x * blockIdx.x + threadIdx.x;
c[i] = a[i] + b[i];
}
template <class T> template <class T>
void CUDAStream<T>::add() void CUDAStream<T>::add()
{ {
return; add_kernel<<<1024, 1024>>>(d_a, d_b, d_c);
check_error();
cudaDeviceSynchronize();
check_error();
} }
template <class T> template <class T>