fy1214 3fc6728dee [add] fix smem_barrier size in wgrad way (#122)

2025-07-02 14:05:36 +08:00

..

include/deep_gemm

Fix illegal memory address when skipping -1 m indices (#113)

2025-06-16 10:44:31 +08:00

jit

Grouped GEMM skip useless computation for unaligned Ms (#103)

2025-05-27 13:43:38 +08:00

jit_kernels

[add] fix smem_barrier size in wgrad way (#122)

2025-07-02 14:05:36 +08:00

__init__.py

Weight gradient kernels for dense and MoE models (#95)

2025-05-14 14:47:58 +08:00

utils.py

Weight gradient kernels for dense and MoE models (#95)

2025-05-14 14:47:58 +08:00