Repository navigation
[Bugfix][Prefix caching] Fix inconsistent hybrid Mamba prefix cache boundaries - #60533
Conversation
…boundaries Extract the core prefix-cache changes from vllm-project#59826. Use a resend-safe checkpoint boundary, resolve checkpoint granularity during worker initialization, and inspect EAGLE attention proof before applying the returned hit limit. Preserve the local cache registration and retention fixes with their core regression tests. Mooncake connector, connector tests, and Mooncake CI setup are excluded. Co-authored-by: Codex <noreply@openai.com> Signed-off-by: Wei Zhao <weizha@aws-cmh-slurm-1-login-01.cm.cluster>
Ship the hash block size and cache-hit alignment to workers through KVCacheConfig instead of injecting them from the model runners, and share the fine-grained lookup veto with the coordinator. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com> Signed-off-by: zjy0516 <riverclouds.zhu@qq.com>
|
/ci run --allow-stale |
|
✅ Triggered Buildkite CI #93531 for commit
|
CI selector (shadow): 143 test steps (190 jobs) instead of 95 (120 jobs)Shadow mode: this changes nothing about what CI runs. It shows what the evidence-based selector would pick for this PR, next to today's rules. How it works. Feedback welcome: reply here if it would skip a step this change needs, or runs something unrelated.
Selector would run (143)
Would skip (today's rules run them) (16)
Would add (today's rules do not run them) (64)
AMD mirrors: would skip (10)
AMD mirrors: would add (41)
17 changed files · base |
|
/ci retry |
|
✅ Queued 2 failed job(s) for retry in Buildkite CI #93531. |
|
/ci run |
|
❌ This PR is 4 commits behind upstream |
|
/ci run --allow-stale |
|
✅ Triggered Buildkite CI #93635 for commit
|
|
/ci retry |
|
✅ No failed, timed-out, or expired jobs need retrying: https://buildkite.com/vllm/ci/builds/93635 |
|
/ci retry |
|
✅ Queued 2 failed job(s) for retry in Buildkite CI #93635. |
…artial tails Mooncake part of vllm-project#59826 by Wei Zhao, rebased onto main after the core fixes landed in vllm-project#60533. Save the full-attention KV through the Mamba checkpoint plus the EAGLE proof margin, retain the source blocks until the asynchronous save completes, and let the external lookup inspect the proof beyond the hit limit. partial_tail_block_ranges computes the tail blocks for both the scheduler (pinning) and the worker (puts). Also provisions a Mooncake master for the Kimi K3 prefix-cache CI and adds a Mooncake case to its resend test. Co-authored-by: Wei Zhao <51183510+wzhao18@users.noreply.github.com> Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01TwEq3d2Nyvagw8ujXUmfci Signed-off-by: Dao Le <Dao007forever@gmail.com> Signed-off-by: Dao007forever <daole@inferact.ai>
…#60533 Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com> Signed-off-by: zjy0516 <riverclouds.zhu@qq.com>
…oundaries (vllm-project#60533) Signed-off-by: Wei Zhao <weizha@aws-cmh-slurm-1-login-01.cm.cluster> Signed-off-by: zjy0516 <riverclouds.zhu@qq.com> Co-authored-by: Wei Zhao <weizha@aws-cmh-slurm-1-login-01.cm.cluster> Co-authored-by: Codex <noreply@openai.com> Co-authored-by: zjy0516 <riverclouds.zhu@qq.com> Co-authored-by: Claude Opus 5.5 <noreply@anthropic.com> Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>
This PR extracts the core prefix-caching fixes from #59826, excluding the Mooncake-specific changes.
P - 1cap and EAGLE’s block drop.