Skip to content

[Bugfix][NIXL] Fix Mamba prefill truncation ordering - #53523

Merged
ivanium merged 2 commits into
vllm-project:mainfrom
ZeldaHuang:fix/nixl-mamba-prefill-truncation
Aug 25, 2026
Merged

ivanium merged 2 commits into
vllm-project:mainfrom
ZeldaHuang:fix/nixl-mamba-prefill-truncation

Conversation

@ZeldaHuang

@ZeldaHuang ZeldaHuang commented Aug 24, 2026

Copy link
Copy Markdown
Contributor

Move producer-side N-1 truncation into on_new_request so it runs before the scheduler prefix-cache lookup. This keeps the request length stable while computing local cache hits and prevents zero-new-token assertions.

Fixes #53514

Purpose

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.

@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.

@mergify mergify Bot added bug Something isn't working kv-connector labels Aug 24, 2026
Move producer-side N-1 truncation into on_new_request so it runs before the scheduler prefix-cache lookup. This keeps the request length stable while computing local cache hits and prevents zero-new-token assertions.

Fixes vllm-project#53514

Co-authored-by: OpenAI Codex <noreply@openai.com>

Signed-off-by: Ziming Huang <zelda.huanghuang@gmail.com>
@ZeldaHuang
ZeldaHuang force-pushed the fix/nixl-mamba-prefill-truncation branch from 6357ee0 to cce91dd Compare August 24, 2026 04:51

@ywang96 ywang96 left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

@ywang96

ywang96 commented Aug 24, 2026

Copy link
Copy Markdown
Member

/ci run

@github-actions

Copy link
Copy Markdown

✅ Triggered Buildkite CI #85299 for commit 64e2dcdf274b.

@ivanium ivanium left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

LGTM. Also I realized that we need the same fix for MooncakeConnector (for PD, not MooncakeStoreConnector for kv offloading) too. We can merge this one first but could you also check and submit another follow-up one for MooncakeConnector 🙏

@ivanium
ivanium merged commit d4c4ceb into vllm-project:main Aug 25, 2026
41 checks passed
TomerBN-Nvidia added a commit to TomerBN-Nvidia/vllm that referenced this pull request Aug 27, 2026
Move producer-side N-1 truncation into on_new_request so it runs before the scheduler prefix-cache lookup. This keeps the request length stable while computing local cache hits and prevents zero-new-token EngineCore assertions.

Backport of vllm-project#53523 (d4c4ceb).

Co-authored-by: Ziming Huang <zelda.huanghuang@gmail.com>

Co-authored-by: Roger Wang <hey@rogerw.io>

Co-authored-by: OpenAI Codex <codex@openai.com>

Signed-off-by: tbarnatan <tbarnatan@nvidia.com>
khushali9 pushed a commit to khushali9/vllm that referenced this pull request Aug 29, 2026
)

Signed-off-by: Ziming Huang <zelda.huanghuang@gmail.com>
Co-authored-by: Roger Wang <hey@rogerw.io>
Signed-off-by: khushali9 <khushali.desai9@gmail.com>
am-cohere pushed a commit to am-cohere/vllm that referenced this pull request Sep 1, 2026
)

Signed-off-by: Ziming Huang <zelda.huanghuang@gmail.com>
Co-authored-by: Roger Wang <hey@rogerw.io>
mikeshawcode pushed a commit to mikeshawcode/vllm that referenced this pull request Sep 1, 2026
)

Signed-off-by: Ziming Huang <zelda.huanghuang@gmail.com>
Co-authored-by: Roger Wang <hey@rogerw.io>
Signed-off-by: mikeshawcode <michaelwshaw2@gmail.com>
mikeshawcode pushed a commit to mikeshawcode/vllm that referenced this pull request Sep 1, 2026
)

Signed-off-by: Ziming Huang <zelda.huanghuang@gmail.com>
Co-authored-by: Roger Wang <hey@rogerw.io>
Signed-off-by: mikeshawcode <michaelwshaw2@gmail.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

bug Something isn't working kv-connector

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[Bug]: NIXL Mamba P-side truncation can produce zero new tokens after a local prefix-cache hit

3 participants