mirror of https://github.com/tile-ai/tilelang.git synced 2026-10-24 07:04:00 +08:00

Files

T

Tong WUandcoderabbitai[bot] 3ab93cd76b [Enhancement] Keep max score attention across blocks in FlashAttention for better numerical stablity (#1269 )

* Implement max score retention across blocks in FlashAttention for improved stability

* fix manual pipeline parameters

* Update examples/flash_attention/example_gqa_fwd_varlen.py

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>

* fix typo

* more

* fix a previous typo

---------

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>

2025-11-17 21:53:19 +08:00

block_sparse_attn_triton.py

[Enhancement] Introduce flag to visualize shared memory merge plan (#496 )

2025-05-16 21:00:18 +08:00

example_tilelang_block_sparse_attn.py

[Refactor] Turn off ENABLE_FAST_MATH by default (#846 )

2025-09-18 16:07:13 +08:00

example_tilelang_sparse_gqa_decode_paged.py

[Enhancement] Keep max score attention across blocks in FlashAttention for better numerical stablity (#1269 )

2025-11-17 21:53:19 +08:00

example_tilelang_sparse_gqa_decode_varlen_indice.py

[Enhancement] Keep max score attention across blocks in FlashAttention for better numerical stablity (#1269 )