Wangzheee 26a603f518 fix some bug		2025-06-20 06:53:24 +00:00
..
include/deep_gemm	fix some bug	2025-06-20 06:53:24 +00:00
jit	Grouped GEMM skip useless computation for unaligned Ms (#103)	2025-05-27 13:43:38 +08:00
jit_kernels	fix some bug	2025-06-20 06:53:24 +00:00
__init__.py	support group_gemm_offset, group_gemm_offset_swapAB	2025-06-19 14:51:38 +00:00
utils.py	Weight gradient kernels for dense and MoE models (#95)	2025-05-14 14:47:58 +08:00