mirror of https://github.com/tile-ai/tilelang.git synced 2026-10-02 06:34:36 +08:00

Files

T

History

Tong WUandLeiWang1999 228c7c0486 [Feature] Support iket profiler for CUDA backend (#2515 )

* Add experimental IKET CUDA backend hooks

* Keep IKET APIs under namespace

* Use real Perfetto screenshot for IKET example

* Show IKET events in Perfetto timeline screenshot

* Add IKET validation overhead example

* Remove IKET validation overhead example

* Fix IKET lint issues

* Move IKET integration to CUDA tools

* Add tools documentation and fix workflow regressions

---------

Co-authored-by: LeiWang1999 <leiwang1999@outlook.com>

2026-07-12 20:03:30 +08:00

images

[Tools] Introduce plot_layout to visualize the fragment layout (#68 )

2025-02-10 01:03:01 +08:00

fragment_mfma_load_a.py

[TIR][IR] Update to use tirx (#2216 )

2026-05-20 15:38:13 +08:00

fragment_mma_load_a.py

[TIR][IR] Update to use tirx (#2216 )

2026-05-20 15:38:13 +08:00

layout_swizzle.py

[Enhancement] Improve plot_layout visualization for Layouts (#1811 )

2026-02-08 22:37:47 +08:00

layout_transform.py

[Enhancement] Improve plot_layout visualization for Layouts (#1811 )

2026-02-08 22:37:47 +08:00

README.md

[Feature] Support iket profiler for CUDA backend (#2515 )

2026-07-12 20:03:30 +08:00

README.md

Layout Visualization Examples

These examples visualize TileLang layouts and fragments with tilelang.tools.plot_layout. See the Layout Visualization guide for installation, API details, compiler-driven layout visualization, and current limitations.

File	Coverage
`layout_transform.py`	Layout composition and transforms
`layout_swizzle.py`	Shared-memory swizzle mappings
`fragment_mma_load_a.py`	NVIDIA MMA fragment mapping
`fragment_mfma_load_a.py`	AMD MFMA fragment mapping

Install the visualization dependency and run an example from the repository root:

pip install "tilelang[vis]"
python examples/plot_layout/fragment_mma_load_a.py

Plots are written under ./tmp unless the example supplies another output directory. A representative fragment plot is available at images/base_layout.png.