yukuai e82c4139da Revert "Fixed the bug in get_swizzle_mode function related to elem_size setting. (#115)" ...

This reverts commit ac428e25e0.

This PR causes wgrad to hang during testing. Revert it until we resolve the issue

2025-06-23 17:13: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

Revert "Fixed the bug in get_swizzle_mode function related to elem_size setting. (#115)"

2025-06-23 17:13: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