vllm-project/vllm

[Feature]: Extract KV-Cache update from all attention backends

Open

#32,335 opened on 2026幎1月14日

GitHub で芋る
 (51 comments) (0 reactions) (9 assignees)Python (16,816 forks)batch import
feature requestgood first issuehelp wanted

Repository metrics

Stars
 (80,034 stars)
PR merge metrics
 (平均マヌゞ 3d 17h) (30d で 993 merged PRs)

説明

🚀 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.

コントリビュヌタヌガむド