Zhean Xu 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

5.6 KiB

Raw Blame History

View Raw