vllm-project/vllm
[Feature]: Extract KV-Cache update from all attention backends
开放
#32,335 创建于 2026年1月14日
feature requestgood first issuehelp wanted
仓库指标
- 星标
- (80,034 个星标)
- PR 合并指标
- (PR 指标待抓取)
描述
🚀 The feature, motivation and pitch
Similar to how https://github.com/vllm-project/vllm/pull/25954 extracts it from FlashAttn. Ideally, we want to cover all backends with kv-cache update from v1/attention/backends.
Backends:
- FlashAttention
- FlashInfer
- AiterFlashAttention (in progress)
- RocmAiterUnifiedAttention
- RocmAttention
- TritonAttention
- FlashAttentionDiffKV
- FlexAttention
- TreeAttention
MLA Backends:
- FlashAttnMLA
- FlashInferMLA
- FlashMLASparse
- FlashMLA
- AiterMLA
- ROCMAiterMLASparse
- CutlassMLA
- TritonMLA
After all backends are supported, we can remove slot_mapping from attention metadata.
Alternatives
No response
Additional context
No response
Before submitting a new issue...
- Make sure you already searched for relevant issues, and asked the chatbot living at the bottom right corner of the documentation page, which can answer lots of frequently asked questions.