fix(engine): adapt colocated AWEX to SGLang scheduler APIs - #1723
Open
Le8r0nJames wants to merge 1 commit into
Open
Le8r0nJames wants to merge 1 commit into
Le8r0nJames wants to merge 1 commit into
Conversation
Le8r0nJames
requested review from
garrett4wade,
geshi001,
nuzant,
rchardx and
sitabulaixizawaluduo
as code owners
September 17, 2026 07:12
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Description
Follow-up to #1703 (Qwen3.5 hybrid AWEX training), on the colocated path introduced
by #1500.
TP worker, rather than silently treating missing nonzero ranks as rank zero.
Preserve typed acknowledgments for duplicate tagged requests.
checks confirm there is no active work; retain and restore the request queue.
to the resumed model's logical device, not LOCAL_RANK or a physical GPU index.
The newly allowlisted build is exactly 0.5.18.dev10+g85b539146. This is not a claim
of compatibility with every 0.5.18 variant. Existing supported builds remain
allowlisted; no transport plan, weight-conversion, or placement policy changes.
Related Issue
Follow-up to #1703.
No separate issue.
Type of Change
Checklist
pre-commit run --all-files)./docs/build_all.sh)main/review-prcommand/create-prBreaking Change Details (if applicable): None.
Additional Context
This is an independent change based on main, not the experimental runtime stack.
It reuses tests already developed for the runtime; no additional test scenarios
were introduced during the split.
Validation:
pytest -q tests/test_sglang_0518_compat.py tests/test_awex_sglang_plugin.py— 33 passed in the target Python 3.12 / Torch 2.9.1 environment.Changed-file pre-commit
hooks pass. Full-tree hooks were attempted but hit local Python 3.10 environment
limitations (missing tomllib and the installed Torch lacking CPUOffloadPolicy).
No full distributed end-to-end regression is claimed for this rebased branch.