DeepSeek/DeepGEMM
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepGEMM synced 2025-06-26 23:15:49 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity
88a2b6c490
DeepGEMM/deep_gemm/jit_kernels
History
Ther-LF 88a2b6c490 Fixed the bug in get_swizzle_mode function related to elem_size setting.
2025-06-17 08:49:36 +00:00
..
__init__.py Weight gradient kernels for dense and MoE models (#95) 2025-05-14 14:47:58 +08:00
gemm.py Fixed the bug in get_swizzle_mode function related to elem_size setting. 2025-06-17 08:49:36 +00:00
m_grouped_gemm.py Fix JIT tests 2025-05-16 14:39:58 +08:00
runtime.py Unify kwargs usages 2025-05-15 16:53:52 +08:00
utils.py Fix get_col_major_tma_aligned_tensor to handle 2-dimensional inputs 2025-03-13 22:15:16 +08:00
wgrad_gemm.py Fix JIT tests 2025-05-16 14:39:58 +08:00