3 people 90619351e3 [Attention] Mamba attention module refactor - LINEAR (#43556) ...

Signed-off-by: wangxiyuan <wangxiyuan1007@gmail.com>
Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>

2026-06-04 18:45:29 +08:00

..

test_attention_backends_selection.py

[Attention] Mamba attention module refactor - LINEAR (#43556)

2026-06-04 18:45:29 +08:00

test_attention_backends.py

[Attention] Make FlexAttention and FlashAttention use num-blocks first layouts (#42095)

2026-05-26 19:55:56 -07:00

test_attention_splitting.py

[Bugfix] Zero stale is_prefilling in padded CUDA graph rows for Mamba (#41873)

2026-05-21 15:42:42 -07:00

test_batch_reordering.py

[Attention] Support distinguishing between short extends and decodes (#37303)

2026-03-20 10:49:36 -07:00

test_chunked_local_attention.py

refactor hard coded device string in test files under tests/v1 and tests/lora (#37566)

2026-04-03 11:21:47 +08:00

test_gdn_metadata_builder.py

[Bugfix] Fix GDN attention crash with mixed decode/spec-decode batches (#34871)

2026-03-16 10:30:23 +01:00

test_indexer_deepseek_v4_slot_mapping.py

[Feat] DeepSeek V4 Rebased (#40860)

2026-04-26 18:31:08 -07:00

test_kv_head_stride_canonicalization.py

[Attention] Make FlexAttention and FlashAttention use num-blocks first layouts (#42095)

2026-05-26 19:55:56 -07:00

test_mamba_update_block_table.py

[Bugfix] mamba: run single-token extends as decodes (#42430)

2026-05-18 15:26:00 +00:00

test_mla_backends.py

[MLA Attention Backend] Add TOKENSPEED_MLA backend for DSR1/Kimi K25 prefill + decode on Blackwell (#41778)

2026-05-13 23:48:02 -07:00

test_mla_prefill_registry.py

[MLA][Attention] Add OOT MLA prefill backend registration mechanism (#43325)

2026-05-26 19:56:09 -07:00

test_mla_prefill_selector.py

[Attention] Remove deprecated MLA prefill arguments (#42555)

2026-05-14 10:34:06 -07:00

test_rocm_attention_backends_selection.py

[ROCm] Attention selector reordering (#36702)

2026-03-25 17:42:56 +08:00

test_sparse_mla_backends.py

refactor hard coded device string in test files under tests/v1 and tests/lora (#37566)

2026-04-03 11:21:47 +08:00

test_trtllm_attention_integration.py

Add nvfp4 kv cache support (#40177)

2026-05-01 04:55:16 +00:00

utils.py

[Bugfix] mamba: run single-token extends as decodes (#42430)

2026-05-18 15:26:00 +00:00