From 16b37f3119918c1e5a39f303e0d0892c65c07a90 Mon Sep 17 00:00:00 2001 From: Qiu Date: Wed, 5 Nov 2025 17:58:13 +0800 Subject: [PATCH] [bugfix] fix wrong `dcp_local_seq_lens` calc (#27518) Signed-off-by: Qiu --- vllm/v1/attention/backends/mla/common.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm/v1/attention/backends/mla/common.py b/vllm/v1/attention/backends/mla/common.py index 413d20ce04021..f57dfc1571b6e 100755 --- a/vllm/v1/attention/backends/mla/common.py +++ b/vllm/v1/attention/backends/mla/common.py @@ -800,7 +800,7 @@ class MLACommonMetadataBuilder(AttentionMetadataBuilder[M]): dcp_local_seq_lens[:num_decodes] = seq_lens[ :num_decodes ] // self.dcp_world_size + ( - self.dcp_rank <= (seq_lens[:num_decodes] - 1) % self.dcp_world_size + self.dcp_rank < seq_lens[:num_decodes] % self.dcp_world_size ) assert num_decodes + num_prefills == num_reqs