Add clearing kernel

This commit is contained in:
2024-04-11 19:49:09 +02:00
parent b49dddf34a
commit e86e04f6d6
2 changed files with 16 additions and 0 deletions

View File

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