forked from Karylab-cklius/vllm
- Insert TOKENSPEED_MLA into SM10 MLA priority list (behind FLASHINFER_MLA);
supports_combination guards non-R1 / non-FP8 configs. Auto-regenerates
docs/design/attention_backends.md.
- Wire TOKENSPEED_MLA into tests/v1/attention/test_mla_backends.py and
parameterize the prefill backend (FLASH_ATTN, FLASHINFER, TRTLLM_RAGGED,
TOKENSPEED_MLA); per-test skip via validate_configuration. Drop standalone
test_tokenspeed_mla_{decode,prefill}.py - coverage now lives there.
- Add TOKENSPEED_MLA to mla_decode.yaml backends and 'tokenspeed' to
mla_prefill.yaml prefill_backends; mla_runner.py grows a registry-based
branch that sets attention_config.mla_prefill_backend directly instead of
the deprecated boolean flags.
Co-authored-by: Claude
Signed-off-by: Yongye Zhu <yongye@inferact.ai>
Signed-off-by: Yongye Zhu <zyy1102000@gmail.com>