Matrix Multiplication Examples
Examples
- Matrix-vector multiplication (GEMV)
- Matrix multiplication with 4D tensors — note the difference from 3D tensors
- Split-k GEMM implementation in cuTile
For standard 2D GEMM and 3D batch matmul (BMM), see src/tilegym/ops/cutile/matmul.py and src/tilegym/ops/cutile/bmm.py in the TileGym repo.