Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-06 01:34:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add clearing kernel

Browse Source
This commit is contained in:
Matúš Námešný
2024-04-11 19:49:09 +02:00
parent b49dddf34a
commit e86e04f6d6
2 changed files with 16 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
include/kernels/matmul.cuh
View File

@@ -62,6 +62,11 @@ __global__ void vec_scalar_sub(
const unsigned int w const unsigned int w
); );
__global__ void clear(
float* __restrict__ d_vector,
const unsigned int w
);
} // namespace CUDANet::Kernels } // namespace CUDANet::Kernels
#endif // CUDANET_MATMUL_H #endif // CUDANET_MATMUL_H

11
src/kernels/matmul.cu
View File

@@ -84,3 +84,14 @@ __global__ void Kernels::vec_scalar_sub(
} }
d_output[tid] = d_vector[tid] - d_scalar[0]; d_output[tid] = d_vector[tid] - d_scalar[0];
} }
__global__ void Kernels::clear(
float* __restrict__ d_vector,
const unsigned int w
) {
int tid = blockDim.x * blockIdx.x + threadIdx.x;
if (tid >= w) {
return;
}
d_vector[tid] = 0.0f;
}