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