Yongye ZhuandRoger Wang 0d2732dd91 [MLA Attention Backend] Add TOKENSPEED_MLA backend for DSR1/Kimi K25 prefill + decode on Blackwell (#41778) ...

Signed-off-by: Yongye Zhu <zyy1102000@gmail.com>
Signed-off-by: Roger Wang <hey@rogerw.io>
Co-authored-by: Roger Wang <hey@rogerw.io>

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

..

test_attention_backends_selection.py

[Misc] Replace mamba_type string literals with MambaAttentionBackendEnum (#41188)

2026-05-11 03:59:36 +00:00

test_attention_backends.py

[Attention][Cleanup] Remove tree attention (#42121)

2026-05-08 18:36:19 -07:00

test_attention_splitting.py

[6/N][Attention] Move utils to more appropriate locations (#32215)

2026-01-13 05:38:52 -08: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

[Bugfix] Fix degenerate KV cache stride causing TMA cudaErrorIllegalInstruction (#40737)

2026-05-03 23:52:18 +00:00

test_mamba_update_block_table.py

[Model][Spec Decode] Nemotron-H MTP and Mamba Speculative Decoding Support (#33726)

2026-02-24 09:49:56 -08: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_selector.py

[Attention] Minor refactor: layer takes ownership of the MLA prefill backend (#41744)

2026-05-05 23:22:41 +00: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

[Core] Avoid seq_lens_cpu GPU->CPU sync (#40654)

2026-04-24 00:35:55 +00:00