[feature]dcp&pcp support mlapo (#5672)

### What this PR does / why we need it?
mlapo in deepseek is a huge performance improvement in decode, this pr
support pcp & dcp with mlapo

### Does this PR introduce _any_ user-facing change?
NO

### How was this patch tested?

- vLLM version: v0.13.0
- vLLM main:
2f4e6548ef

---------

Signed-off-by: zhenwenqi2024 <zhenwenqi_2022@qq.com>
This commit is contained in:
zhenwenqi2024
2026-01-08 23:49:23 +08:00
committed by GitHub
parent 6315a31399
commit 97f6be8108
7 changed files with 35 additions and 10 deletions

View File

@@ -800,7 +800,8 @@ class EagleProposer(VllmEagleProposer):
query_start_loc_cpu=query_start_loc_cpu,
seq_lens_cpu=common_attn_metadata.seq_lens_cpu,
num_reqs=common_attn_metadata.num_reqs,
num_actual_tokens=total_num_tokens,
num_actual_tokens=common_attn_metadata.num_actual_tokens
if self.pcp_size > 1 else total_num_tokens,
num_input_tokens=common_attn_metadata.num_input_tokens,
max_query_len=new_query_len_per_req.max().item(),
actual_seq_lengths_q=self.runner.actual_seq_lengths_q,