trending/tilelang
1
0
Fork 0
mirror of https://github.com/tile-ai/tilelang.git synced 2026-11-01 08:23:24 +08:00
Code Issues Packages Projects Releases Wiki Activity
1,759 Commits 9 Branches 24 Tags
bc9515febdeeac6fa7b339d322adebd3add7e07c
Commit Graph
3 Commits
Author SHA1 Message Date
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