DeepSeek/DeepGEMM
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepGEMM synced 2025-06-26 23:15:49 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Refactor MMA template with CUTLASS (#87)

Browse Source 
* Refactor MMA with cutlass

* Update README.md

---------

Co-authored-by: Zhean Xu <xza@deepseek.com>
This commit is contained in:
Zhean Xu
2025-04-14 17:06:49 +08:00
committed by GitHub
parent 37aa127451
commit 4499c4ccbb
2 changed files with 48 additions and 880 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@@ -22,7 +22,7 @@ Despite its lightweight design, DeepGEMM's performance matches or exceeds expert
- [ ] CUDA PDL support
- [ ] More scaling granularity support via templates
- [ ] Larger TMA multicast size for some shapes
- [ ] MMA template refactor with CUTLASS
- [x] MMA template refactor with CUTLASS
- [ ] Optimizations for unaligned shapes
- [ ] Optimizations for power efficiency
- [ ] Remove shape limitations on N and K