Skip to content

GEMM Optimization

Detailed walkthrough of the 7-step GEMM optimization path.

Optimization Path

Summary

StepTechniqueTFLOPS (FP32)Speedup
1Naive0.51.0×
2Shared Memory2.04.0×
3Double Buffer3.57.0×
4Register Tiling6.012.0×
5WMMA50+100×
6MMA PTX60+120×
7Pipeline70+140×

References

Released under the MIT License.