Use tiling shmem for mat vec mul kernel

This commit is contained in:
2024-03-15 23:33:09 +01:00
parent 88f7fff217
commit dc86cddeb7
4 changed files with 54 additions and 24 deletions

View File

@@ -66,6 +66,10 @@ class Dense : public ILayer {
Layers::Activation activation;
// Precompute kernel launch parameters
int forwardGridSize;
int biasGridSize;
/**
* @brief Initialize the weights to zeros
*