Commit Graph
4 Commits
Author SHA1 Message Date
Ray Wangandfzyzcjy 9da4a23561 Add more GPU architectures support (#112)
* Add more GPU architectures support

* Update layout.py

* Optimize performance, Add SM90 support, Add 1D2D SM100 support

* Add fmtlib submodule at commit 553ec11

---------

Co-authored-by: fzyzcjy <5236035+fzyzcjy@users.noreply.github.com>
2025-07-18 11:32:22 +08:00
Zhean XuandChenggang Zhao 04278f6dee Weight gradient kernels for dense and MoE models (#95)
* Init weight gradient kernels.

* Support unaligned n,k and gmem stride

* Update docs

* Several cleanups

* Remove restrictions on N

* Add stride(0) assertions

---------

Co-authored-by: Chenggang Zhao <chenggangz@deepseek.com>
2025-05-14 14:47:58 +08:00
AcraeaTerpsicore 96b31fd6bb fix typo 2025-02-26 18:37:22 +08:00
Chenggang Zhao a6d97a1c1b Initial commit 2025-02-25 22:52:41 +08:00