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_tilelang_gemm_splitk_vectorize_atomicadd.py

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

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

example_tilelang_gemm_splitk.py

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

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

regression_example_gemm_splitk.py

[CI] Add preformance regression test script (#1489)

2025-12-22 16:56:15 +08:00

test_example_gemm_splitk.py

[Feature]:Add auto vectorize for atomic add (#686)

2025-08-03 17:23:18 +08:00