DeepSeek/DeepGEMM
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepGEMM synced 2025-05-06 00:04:22 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity
37aa127451
DeepGEMM/deep_gemm/jit_kernels
History
Chenggang Zhao 37aa127451
Use swizzling instead of padding (#86) 
* Add swizzling params

* Add TMA D descriptor

* Always use STSMx2

* Swizzling draft

* Compatible with padding

* Fix bugs

* Optimize swizzle performance

* Optimize expression

* Optimize TMA issues

* Fix README

* Stricter assertions
2025-04-14 15:20:58 +08:00
..
__init__.py fix typo 2025-02-26 18:37:22 +08:00
gemm.py Use swizzling instead of padding (#86) 2025-04-14 15:20:58 +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