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

Files

T

History

Liu YunuoandLeiWang1999 b9a5907a20 [Example] Migrate eligible examples to eager style (#2010 )

* [Example] Migrate eligible examples to eager style

* fix wrong shape annotation in gemm_split_k examples

* fix incorrect dtype in gemm_streamk example

* fix

* revert some changes

* Refactor attention functions to use new parameter structures and improve readability. Updated `example_amd_flash_attn_bwd.py` and `example_amd_flash_attn_fwd.py` to define input parameters more clearly. Enhanced shared memory usage in `flashattn_fwd` and `flashattn_sink` functions across various examples, ensuring compatibility with new configurations. Adjusted kernel definitions and added necessary checks for GPU capabilities.

* lint fix

* FIX

---------

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

2026-05-21 20:27:06 +08:00

example_linear_attn_bwd.py

[Refactor][Pipeline] Run pipeline rewriting before layout inference and stabilize tiled WS (#2002 )

2026-04-07 23:54:55 +08:00

example_linear_attn_fwd.py

[Refactor][Pipeline] Run pipeline rewriting before layout inference and stabilize tiled WS (#2002 )

2026-04-07 23:54:55 +08:00

example_mamba_chunk_scan.py

[Example] Migrate eligible examples to eager style (#2010 )

2026-05-21 20:27:06 +08:00

example_mamba_chunk_state.py

[Example] Migrate eligible examples to eager style (#2010 )