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
Files
d5470d3b4e7a6e3bbe426887e34dfc2f05aa49ee
DeepGEMM/deep_gemm/jit_kernels
History
Zhean Xu d5470d3b4e Init weight gradient kernels.
2025-05-06 17:16:27 +08:00
..
__init__.py
Init weight gradient kernels.
2025-05-06 17:16:27 +08:00
gemm.py
Init weight gradient kernels.
2025-05-06 17:16:27 +08:00
m_grouped_gemm.py
Use swizzling instead of padding (#86)
2025-04-14 15:20:58 +08:00
tuner.py
Initial commit
2025-02-25 22:52:41 +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
Init weight gradient kernels.
2025-05-06 17:16:27 +08:00