Skip to content

Conversation

@Deleter-D
Copy link
Collaborator

Motivation

When MTP is enabled with Logprob, if chunk prefill occurs, it will cause the Worker to send some unexpected draft logprob packages to the Engine.

Modifications

Add checks when sending packages to avoid sending unexpected packets during the chunk prefill phase.

Usage or Command

Accuracy Tests

Checklist

  • Add at least a tag in the PR title.
    • Tag list: [[FDConfig],[APIServer],[Engine], [Scheduler], [PD Disaggregation], [Executor], [Graph Optimization], [Speculative Decoding], [RL], [Models], [Quantization], [Loader], [OP], [KVCache], [DataProcessor], [BugFix], [Docs], [CI], [Optimization], [Feature], [Benchmark], [Others], [XPU], [HPU], [GCU], [DCU], [Iluvatar], [Metax]]
    • You can add new tags based on the PR content, but the semantics must be clear.
  • Format your code, run pre-commit before commit.
  • Add unit tests. Please write the reason in this PR if no unit tests.
  • Provide accuracy results.
  • If the current PR is submitting to the release branch, make sure the PR has been submitted to the develop branch, then cherry-pick it to the release branch with the [Cherry-Pick] PR tag.

@paddle-bot
Copy link

paddle-bot bot commented Nov 26, 2025

Thanks for your contribution!

freeliuzc
freeliuzc previously approved these changes Nov 26, 2025
@Jiang-Jia-Jun Jiang-Jia-Jun merged commit bbcd92c into PaddlePaddle:feature/experimental_feature_20250908 Nov 27, 2025
22 of 25 checks passed
@Deleter-D Deleter-D deleted the exp_mtp_logprob_fix branch January 12, 2026 03:19
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants