Skip to content

[Fast Start] Cache the MTP draft model in a separate daemon group - #57312

Merged
Isotr0py merged 13 commits into
vllm-project:mainfrom
liusy58:weight-cache-mtp-draft
Sep 22, 2026
Merged

Isotr0py merged 13 commits into
vllm-project:mainfrom
liusy58:weight-cache-mtp-draft

Conversation

@liusy58

@liusy58 liusy58 commented Sep 17, 2026 •

Copy link
Copy Markdown
Contributor

Purpose

Under --load-format ipc_cache, the MTP draft model was not served by the weight cache daemon: the engine either reloaded it from disk or sent it to the target daemon with a mismatching fingerprint.

The launcher now starts a second daemon group for the MTP draft, with its own cache key, sockets (*_draft0.sock) and rendezvous port (--weight-cache-draft-master-port). On the engine side, get_draft_load_config routes an MTP draft to that group and lets other draft types (EAGLE3, DFlash, DSpark) fall back to disk. Only MTP drafts are cached; the draft must use the target's TP size.

Test Plan

Test Result


Essential Elements of an Effective PR Description Checklist
  • The purpose of the PR, such as "Fix some issue (link existing issues this PR will resolve)".
  • The test plan, such as providing test command.
  • The test results, such as pasting the results comparison before and after, or e2e results
  • (Optional) The necessary documentation update, such as updating supported_models.md and examples for a new model.

Signed-off-by: liusy58 <mg21330037@smail.nju.edu.cn>

@claude claude Bot left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Claude Code Review

This pull request is from a fork — automated review is disabled. A repository maintainer can comment @claude review to run a one-time review.

Signed-off-by: liusy58 <mg21330037@smail.nju.edu.cn>
@liusy58 liusy58 mentioned this pull request Sep 17, 2026
14 tasks
@Isotr0py Isotr0py self-assigned this Sep 19, 2026
Comment thread vllm/model_executor/model_loader/utils.py Outdated
Comment thread vllm/model_executor/model_loader/weight_cache/protocol.py
liusy58 and others added 6 commits September 20, 2026 08:20
Signed-off-by: liusy58 <mg21330037@smail.nju.edu.cn>
Signed-off-by: Isotr0py <Isotr0py@outlook.com>
Signed-off-by: Isotr0py <Isotr0py@outlook.com>
Signed-off-by: Isotr0py <Isotr0py@outlook.com>
Signed-off-by: Isotr0py <Isotr0py@outlook.com>
Signed-off-by: Isotr0py <Isotr0py@outlook.com>
Signed-off-by: Isotr0py <Isotr0py@outlook.com>
Signed-off-by: Isotr0py <Isotr0py@outlook.com>
@Isotr0py
Isotr0py enabled auto-merge (squash) September 21, 2026 14:29
@github-actions github-actions Bot added the ready ONLY add when PR is ready to merge/full CI is needed label Sep 21, 2026
@Isotr0py

Copy link
Copy Markdown
Member

/ci run --allow-stale

@github-actions

Copy link
Copy Markdown

✅ Triggered Buildkite CI #90226 for commit 6dff2753df7c.

Signed-off-by: Isotr0py <Isotr0py@outlook.com>
@Isotr0py

Copy link
Copy Markdown
Member

/ci run --allow-stale

@github-actions

Copy link
Copy Markdown

✅ Triggered Buildkite CI #90251 for commit cec5d084f0aa.

⚠️ This PR is 9 commits behind upstream main. Running CI at your own risk because --allow-stale was requested; outdated CI configuration may cause failures. Before merging, merge or rebase onto the latest main, then rerun /ci run on the latest PR commit.

@mergify mergify Bot added the qwen Related to Qwen models label Sep 21, 2026
Signed-off-by: Isotr0py <Isotr0py@outlook.com>
@Isotr0py

Copy link
Copy Markdown
Member

/ci retry

@github-actions

Copy link
Copy Markdown

✅ Triggered Buildkite CI #90327 for commit 9d0ad0c0a68e, running 1 failed step(s) from Buildkite CI #90251.

@Isotr0py
Isotr0py merged commit c723a83 into vllm-project:main Sep 22, 2026
15 checks passed
@github-project-automation github-project-automation Bot moved this from Backlog to Done in Sprint - DFlash Sep 22, 2026
khluu added a commit that referenced this pull request Sep 23, 2026
…ture (#58107)

Signed-off-by: kimi-no-na-wa <kimi-no-na-wa@users.noreply.github.com>
Co-authored-by: kimi-no-na-wa <kimi-no-na-wa@users.noreply.github.com>
Co-authored-by: Kevin H. Luu <khluu000@gmail.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

dflash mrv2 Model Runner V2 specific qwen Related to Qwen models ready ONLY add when PR is ready to merge/full CI is needed speculative-decoding

Projects

Status: Done

Development

Successfully merging this pull request may close these issues.

2 participants