Add bounds checking to implicit GEMM to allow arbitrary input shapes … · tracel-ai/burn@c21e096 · GitHub

Commit

Add bounds checking to implicit GEMM to allow arbitrary input shapes (#…

Browse files

…2354)

* Pad implicit GEMM to allow using it with any input shape

* Improve k padding

* Fix typo

Loading branch information

wingertge authored Oct 11, 2024

1 parent 6cdc34b commit c21e096