DeepGEMM/deep_gemm/jit
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 Initial commit 2025-02-25 22:52:41 +08:00
compiler.py Use swizzling instead of padding (#86) 2025-04-14 15:20:58 +08:00
interleave_ffma.py Performance: reducing the percentage of FFMA interleaving yields a slight performance gain, roughly 0.5% 2025-03-05 17:50:22 +08:00
runtime.py Initial commit 2025-02-25 22:52:41 +08:00
template.py Support multicasting on B 2025-03-25 14:56:42 +08:00