3 people 5ecae3266c [ROCm][Perf][MLA] Add AITER FlashAttention MLA prefill backend (ROCM_AITER_FA) (#45033) ...

Signed-off-by: Xavier Aguilar <xavier.aguilarfruto@amd.com>
Signed-off-by: Xavier Aguilar <Xavier.AguilarFruto@amd.com>
Co-authored-by: TJian <tunjian.tan@embeddedllm.com>

2026-06-28 07:52:00 -07:00

..

test_attention_backends_selection.py

[Model] Remove MiniMaxText01, MiniMaxVL01, MiniMaxForCausalLM (#45993)

2026-06-22 15:20:46 +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_flashinfer_sparse_mla_sm120_api.py

Enable DeepSeek V4 and GLM-5.1 on SM120 (#43477)

2026-06-22 11:54:14 -07:00

test_gdn_metadata_builder.py

[Bugfix] Avoid racy accepted counts in async spec decode (#45100)

2026-06-22 08:53:16 +00: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

[Feature] Support DCP with FP8 KV cache in MLA decode path (#44044)

2026-06-24 19:28:17 +00:00

test_mla_prefill_quant_output.py

feat: MLA prefill enable FA4 fp8 output (#43050)

2026-06-16 07:10:59 -07:00

test_mla_prefill_registry.py

[ROCm][Perf][MLA] Add AITER FlashAttention MLA prefill backend (ROCM_AITER_FA) (#45033)

2026-06-28 07:52:00 -07:00

test_mla_prefill_selector.py

[ROCm][Perf][MLA] Add AITER FlashAttention MLA prefill backend (ROCM_AITER_FA) (#45033)

2026-06-28 07:52:00 -07:00

test_rocm_attention_backends_selection.py

[ROCm] Fix AITER_UNIFIED_ATTN Dispatching After AITER Bump (#46780)

2026-06-26 02:09:56 -07:00

test_sparse_mla_backends.py

Enable DeepSeek V4 and GLM-5.1 on SM120 (#43477)

2026-06-22 11:54:14 -07: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