[main][bugfix] Change seq_lens in dummy attn_metadata to max_query_len (#4097)

### What this PR does / why we need it?
Currently, we set `seq_lens` in dummy attn_metadata to be
`max_model_len` to get max workspace for attention during capturing.
However, setting it consistently to be `max_model_len` causing dummy_run
to execute a long attention when running actual inference. For example,
if there is a single req with `seqs_lens` as [8] but `max_model_len` is
131072, the whole process will be slow down by dummy_run as it execute a
fake long-seq attention. Therefore, we instead set it to max_query_len,
which is also consistent with vLLM gpu implementation.

### Does this PR introduce _any_ user-facing change?
No.

### How was this patch tested?

- vLLM version: v0.11.0
- vLLM main:
83f478bb19

---------

Signed-off-by: Angazenn <supperccell@163.com>
This commit is contained in:
Angazenn
2025-11-12 17:31:39 +08:00
committed by GitHub
parent a123f355e9
commit fc7e5cd9dc

View File

@@ -2824,7 +2824,7 @@ class NPUModelRunner(LoRAModelRunnerMixin):
attn_metadata = {}
seq_lens = self.model_config.max_model_len
seq_lens = max_query_len
self.seq_lens_np[:num_reqs] = seq_lens
self.seq_lens_np[num_reqs:] = 0