Eoin-HoustounandClaude Fable 5 30e2394c83 [Bugfix] Record non-ImportError attention backend probe failures instead of crashing engine init (#51703) ...

Signed-off-by: Eoin <eoin@turintech.ai>
Co-authored-by: Claude Fable 5 <noreply@anthropic.com>

2026-08-20 02:36: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] Fix FlashInfer SM12x prefill with sinks (#52148)

2026-08-13 14:25:09 -07:00

test_attention_splitting.py

[Core] Add MRV2 virtual-batch PCP for MLA (#46570)

2026-07-19 02:53:15 +00:00

test_backend_per_kind.py

[Attention] Allow selecting a different attention backend per KV-cache group (#48012)

2026-07-17 15:19:02 -04: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_cpu_mla_backend.py

[CPU] Add MLA backend so DeepSeek-V2/V3 can run on CPU (#49453)

2026-08-06 16:51:41 +08:00

test_cuda_backend_probe_errors.py

[Bugfix] Record non-ImportError attention backend probe failures instead of crashing engine init (#51703)

2026-08-20 02:36:00 -07:00

test_dspark_noncausal_sparse_mla.py

[1/N] Add dense MHA path for sparse MLA short sequences (#47327)

2026-07-14 00:29:56 +00:00

test_flashinfer_dcp_spec_reorder.py

[ROCm][CI] fix flashinfer import check (#48647)

2026-07-14 20:54:19 +00:00

test_flashinfer_mla_dcp.py

[refactor] consolidate cp attn ops (#52839)

2026-08-19 18:04:11 -07:00

test_flashinfer_sparse_mla_sm120_api.py

[Bugfix][SM120][MLA] Disable dense prefill for FlashInfer sparse MLA (#51395)

2026-08-17 16:58:02 -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_group_head_counts.py

[Bugfix][CPU] Take an attention group's query head count from its layers (#51852)

2026-08-17 18:45:59 +08:00

test_group_sliding_window.py

[Bugfix] Take the sliding window from the layer, not the KV cache group (#51756)

2026-08-11 09:10:46 -07:00

test_indexer_dcp_localize.py

[refactor] consolidate cp attn ops (#52839)

2026-08-19 18:04:11 -07:00

test_indexer_deepseek_v4_slot_mapping.py

[Bugfix] Universally align block table width to 128 tokens (#50302)

2026-07-31 08:52:17 -07:00

test_indexer_native_next_n.py

[Attention][DSA] Take the native decode path for MTP=3 on SM90 (#52164)

2026-08-14 10:09:51 +00: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_linear_attention_metadata_builder.py

[Feature] Support MTP speculative decoding for Bailing hybrid models (#44880)

2026-07-06 10:38:50 +08:00

test_mamba_update_block_table.py

[Kernel] ReplaySSM: cache SSM inputs for faster Mamba2 standard decode (#48018)

2026-07-24 09:39:49 -07:00

test_mla_backends.py

[CPU] Add AMX-only high-performance MLA backend for DeepSeek V2/V3/R1 (#52616)

2026-08-19 10:28:52 +08:00

test_mla_context_chunks.py

[Attention][MLA] Per-request scheduling for MLA chunked context (#50613)

2026-08-06 16:45:44 +00:00

test_mla_noncausal.py

[New model] Kimi K3 (#50000)

2026-07-30 03:49:58 -07: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

[Attention][MLA] Fuse Kimi-K3 chunked-context K/V packing (#51772)

2026-08-12 22:42:57 -07:00

test_mla_prefill_selector.py

[CPU] Add AMX-only high-performance MLA backend for DeepSeek V2/V3/R1 (#52616)

2026-08-19 10:28:52 +08:00

test_mm_prefix.py

[Kernel][Model] Optimize FA4 mm_prefix range lookup (#50294)

2026-08-05 07:33:49 -07:00

test_replayssm_metadata_builder.py

[Kernel] ReplaySSM: cache SSM inputs for faster Mamba2 standard decode (#48018)

2026-07-24 09:39:49 -07:00

test_rocm_aiter_mla_fp8_decode_routing.py

[ROCm][MLA] [K3] Fix fp8 KV cache decode on the AITER MLA backend (#51011)

2026-08-10 11:42:10 -04:00

test_rocm_aiter_mla_mtp_split.py

[ROCm] Pad non-aligned AITER MLA heads (#51647)

2026-08-18 16:37:10 -05:00

test_rocm_attention_backends_selection.py

Enable gfx1250 ROCm architecture (#46516)

2026-07-31 01:41:12 +00:00

test_sparse_mla_backends.py

[Attention][MLA] FlashMLA sparse: DCP on the fp8_ds_mla mixed-batch path + MTP (#46514)

2026-08-19 00:02:09 -04:00

test_sparse_mla_mask.py

[Attention] Vectorize sparse MLA mask loads (#52217)

2026-08-18 20:32:41 -07:00

test_trtllm_attention_integration.py

[2/N][KV-Cache Layout Refactor] Pack K/V into the content dim across attention backends (#44455)

2026-07-11 11:11:16 -04:00

utils.py

[Attention] FlashAttention 4 SM100 FP8 kv cache support (#42569)

2026-07-20 10:53:27 -07:00