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