Skip to content

[sglang-miles] PD disaggragation for RL - #18646

Merged
yueming-yuan merged 1 commit into
sgl-project:sglang-milesfrom
yueming-yuan:sglang-miles-pd
Feb 12, 2026
Merged

[sglang-miles] PD disaggragation for RL#18646
yueming-yuan merged 1 commit into
sgl-project:sglang-milesfrom
yueming-yuan:sglang-miles-pd

Conversation

@yueming-yuan

Copy link
Copy Markdown
Collaborator

Summary

  • Add release_memory_occupation() / resume_memory_occupation() to DecodePreallocQueue and PrefillBootstrapQueue for clearing queues and deregistering/re-registering KV transfer buffers during weight updates
  • Add deregister_buffer_to_engine() to MooncakeKVManager as counterpart to existing register_buffer_to_engine()
  • Allow all requests to be retracted in decode disaggregation mode (min_reqs=0) to support prealloc prefill requests
  • Wire disaggregation queue release/resume into scheduler's release_memory_occupation / resume_memory_occupation lifecycle

- decode.py/prefill.py: Add release_memory_occupation() and
  resume_memory_occupation() to DecodePreallocQueue and
  PrefillBootstrapQueue for clearing queues and deregistering/
  re-registering KV transfer buffers during weight updates
- mooncake/conn.py: Add deregister_buffer_to_engine() to
  MooncakeKVManager as counterpart to register_buffer_to_engine()
- schedule_batch.py: Allow all requests to be retracted in decode
  disaggregation mode (min_reqs=0) for prealloc prefill requests
- scheduler_update_weights_mixin.py: Wire disaggregation queue
  release/resume into scheduler memory occupation lifecycle

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
@gemini-code-assist

Copy link
Copy Markdown
Contributor

Warning

You have reached your daily quota limit. Please wait up to 24 hours and I will start processing your requests again!

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

@yueming-yuan
yueming-yuan merged commit 1b2de49 into sgl-project:sglang-miles Feb 12, 2026
1 check passed
yueming-yuan added a commit that referenced this pull request Feb 13, 2026
Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>
zianglih pushed a commit to zianglih/sglang that referenced this pull request Feb 28, 2026
Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>
JessicaJiang-123 pushed a commit to JessicaJiang-123/sglang that referenced this pull request Jun 21, 2026
nanjiangwill pushed a commit to nanjiangwill/sglang that referenced this pull request Jul 7, 2026
XinyuJiangCMU pushed a commit to XinyuJiangCMU/sglang that referenced this pull request Jul 17, 2026
Arist12 pushed a commit to Arist12/sglang that referenced this pull request Aug 27, 2026
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.

2 participants