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