DeepGEMM/deep_gemm
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