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