xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-12 19:56:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Misc] Remove unnecessary fallback to prefill-decode attention (#19138)

Browse Source 
Signed-off-by: vllmellm <vllm.ellm@embeddedllm.com>
This commit is contained in:
vllmellm 2025-06-05 16:08:26 +08:00 committed by GitHub
parent da40380214
commit 18093084be
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
vllm/v1/attention/backends/triton_attn.py
View File

@ -171,10 +171,7 @@ class TritonAttentionImpl(AttentionImpl):
# Whenever making a change in this method, please benchmark the # Whenever making a change in this method, please benchmark the
# performance to make sure it does not introduce any overhead. # performance to make sure it does not introduce any overhead.
num_queries_per_kv = query.shape[1] // key.shape[1] use_prefill_decode_attn = self.force_prefill_decode_attn
num_q_is_pow2 = (num_queries_per_kv & (num_queries_per_kv - 1)) == 0
use_prefill_decode_attn = (self.force_prefill_decode_attn
or not num_q_is_pow2)
num_actual_tokens = attn_metadata.num_actual_tokens num_actual_tokens = attn_metadata.num_actual_tokens
if use_prefill_decode_attn: if use_prefill_decode_attn: