Skip to content

[sglang-miles] Remove obsolete PD retract guard after rebootstrap support (#25372); squash into 17c3be8e5d (#23672/#23887) - #39119

Merged
guapisolo merged 1 commit into
sglang-milesfrom
jiajun/fix-sglang-miles
Sep 11, 2026
Merged

guapisolo merged 1 commit into
sglang-milesfrom
jiajun/fix-sglang-miles

Conversation

@guapisolo

@guapisolo guapisolo commented Sep 11, 2026

Copy link
Copy Markdown
Collaborator

Summary

Restore PD retract pause by removing an obsolete rejection on sglang-miles.

Symptom & Reproduction

  • Symptom: pause_generation(mode="retract") raises an assertion in both PD modes before pausing, blocking the supported retract/update/resume flow.
  • Reproduction: At 32839114c4ada2ae237581405a8ff39dc0db9e25, run python -m pytest -q test/registered/unit/managers/test_scheduler_pause_generation.py: 3 PD cases fail with pause_generation(mode='retract') is not supported in PD disaggregation mode yet; 17 pass. The existing tests expect PD retract to succeed.

Root Cause

  1. Scheduler.pause_generation received the old PD prohibition from [PD] Gate disagg event loops on _engine_paused; reject pause(retract) in PD #23672 when rebootstrap was unavailable.
  2. hold_rebootstrap and resume-time enqueueing were implemented in [PDD] Add true request retraction for PDD #25372, already merged into main.
  3. 17c3be8e5dc6579edce6715ba8e6ba05081170cd replayed the obsolete guard onto sglang-miles, preventing entry into that implementation.

Fix

Remove the 13-line guard and its outdated explanation. Keep mode validation, in-place pause handling, event-loop gates, and the existing retract/rebootstrap implementation unchanged.

Patch-stack maintenance: Squash this fix into 17c3be8e5dc6579edce6715ba8e6ba05081170cd, titled [8/35] [sglang-miles] Improve PD pause handling (#23672, #23887), when refreshing the sglang-miles patch stack. That is the branch commit that reintroduced the outdated restriction.

Verification

  • test_scheduler_pause_generation.py: all 20 unchanged tests pass after the fix.
  • Existing PD GPU test bodies: 5/5 pass, covering prefill idle retract, running decode retract, paused decode weight reload/cache flush/rebootstrap, in-place pause/resume, and prefill request-leak checks.
  • Runtime: Llama-3.1-8B-Instruct, Mooncake, both endpoints on one H200, CUDA graphs disabled. This does not establish multi-GPU coverage or prefill live-chunk weight consistency.
  • git diff --check: passed. Test processes exited; GPU memory and test ports were released.

Review Focus

  • Scheduler.pause_generation: PD retract must reach the existing implementation while preserving unified and in-place behavior.

CI States

Latest PR Test (Base): ❌ Run #34635912091
Latest PR Test (Extra): ❌ Run #34635911727
Latest PR Test (AMD ROCm 10): ❌ Run #34635911773

Allow PD pause_generation(mode="retract") to reach the existing rebootstrap
implementation. The old guard was replayed onto sglang-miles after main
added hold_rebootstrap and resume-time enqueueing in #25372.

When refreshing the sglang-miles patch stack, squash this fix into
17c3be8:
[8/35] [sglang-miles] Improve PD pause handling (#23672, #23887).

Verified: the unchanged pause suite improves from 3 failed / 17 passed to
20 passed. Five existing PD HTTP/GPU cases pass with Llama-3.1-8B-Instruct
and Mooncake, with both endpoints on one H200 and CUDA graphs disabled.
@guapisolo guapisolo changed the title fix(pd): remove obsolete retract rejection on sglang-miles [sglang-miles] Remove obsolete PD retract guard (squash into 17c3be8e5d) Sep 11, 2026
@guapisolo guapisolo changed the title [sglang-miles] Remove obsolete PD retract guard (squash into 17c3be8e5d) [sglang-miles] Remove obsolete PD retract guard after rebootstrap support (#25372); squash into 17c3be8e5d (#23672/#23887) Sep 11, 2026
@guapisolo
guapisolo merged commit 02b5e12 into sglang-miles Sep 11, 2026
79 of 89 checks passed
@guapisolo
guapisolo deleted the jiajun/fix-sglang-miles branch September 11, 2026 21:40
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant