Commit Graph
5 Commits
Author SHA1 Message Date
Yongqi Zhuo 66e8e47a52 [Refactor] Recycle T.unroll(explicit=True) for early explicit unrolling (#2859)
[Refactor] Recycle syntax `T.unroll(explicit=True)` for early explicit unrolling
2026-08-04 13:52:54 +08:00
Chaofan Lin 50481cce09 [Refactor] Remove intrinsic compatibility facade (#2812) 2026-07-30 14:26:21 +08:00
Yongqi Zhuo 1cb4d4f3ac [CUDA] Support arbitrary TMEM layouts (#2785)
* [CUDA] Support arbitrary TMEM layouts

* Fix bugs
2026-07-29 13:07:28 +08:00
Yongqi Zhuo b1e1e250a2 [Backend] [CUDA] Support GMMA/UMMA lowering for sliced SMEM layout (actually arbitrary layout) (#2452)
* Add split-k for example_warp_specialize_flashmla

* Rotate QK issue to end of main loop to allow more latency hiding with TMA issue

* During rescale, read max score from REG instead of SMEM

* Move K_pe_shared_1 load

* Use stmatrix for SP0 write to reduce SMEM bank conflict

* Support arbitrary SMEM layout in MMA lowering

* Remove runtime guard for increase_descriptor_offset
2026-07-01 18:21:53 +08:00
Yongqi Zhuo 65dbc9837b [Backend] Support TMA lowering for arbitrary (swizzled) SMEM layout (#2380) 2026-06-18 18:42:01 +08:00