forked from Karylab-cklius/vllm
22c6542fa7
Add power-of-2 page sizes >= 128 (128/256/512/1024) to the FlashInfer backend's supported kernel block sizes. These are served only by the trtllm-gen dynamic kernel, which requires Blackwell + GQA/MQA (num_qo_heads // num_kv_heads > 1), not MHA. Fail fast in FlashInferMetadataBuilder.__init__ when page_size >= 128 is requested without a usable trtllm-gen path (trtllm disabled, not Blackwell, or MHA) instead of silently hitting the native wrappers, and force the prefill path to trtllm so it does not fall back to FA2. Regenerate docs/design/attention_backends.md for the new block sizes. AI assistance (Claude) was used for this change. Co-authored-by: Claude Signed-off-by: Yongye Zhu <zyy1102000@gmail.com>