Skip to content

Make the rollout-side offload/onload fan-out async to avoid blocking sync code - #1854

Open
fzyzcjy wants to merge 1 commit into
tom/refactor-miles/op3-9from
tom/refactor-miles/op3-10
Open

Make the rollout-side offload/onload fan-out async to avoid blocking sync code#1854
fzyzcjy wants to merge 1 commit into
tom/refactor-miles/op3-9from
tom/refactor-miles/op3-10

Conversation

@fzyzcjy

@fzyzcjy fzyzcjy commented Jul 28, 2026

Copy link
Copy Markdown
Collaborator

Part of #1837

@gemini-code-assist

Copy link
Copy Markdown
Contributor

Caution

The consumer version of Gemini Code Assist on GitHub has been sunset. All code review activity has officially ceased.

@fzyzcjy
fzyzcjy force-pushed the tom/refactor-miles/op3-10 branch from 1c21446 to 60bb61f Compare August 3, 2026 23:44
@fzyzcjy
fzyzcjy force-pushed the tom/refactor-miles/op3-9 branch 2 times, most recently from 448c30f to b9946a1 Compare August 4, 2026 00:39
@fzyzcjy
fzyzcjy force-pushed the tom/refactor-miles/op3-10 branch 2 times, most recently from 2eb2e1b to cd4c421 Compare August 4, 2026 06:56
@fzyzcjy
fzyzcjy force-pushed the tom/refactor-miles/op3-9 branch from b9946a1 to e10f4c5 Compare August 4, 2026 06:57
@fzyzcjy
fzyzcjy force-pushed the tom/refactor-miles/op3-10 branch from cd4c421 to de62d6c Compare August 10, 2026 06:06
@fzyzcjy
fzyzcjy force-pushed the tom/refactor-miles/op3-9 branch from e10f4c5 to b893f39 Compare August 10, 2026 06:07
@fzyzcjy
fzyzcjy force-pushed the tom/refactor-miles/op3-10 branch 4 times, most recently from 6a5194c to c407f31 Compare August 12, 2026 14:31
@fzyzcjy
fzyzcjy force-pushed the tom/refactor-miles/op3-9 branch 2 times, most recently from b95089c to a8e3e02 Compare August 12, 2026 23:56
@fzyzcjy
fzyzcjy force-pushed the tom/refactor-miles/op3-10 branch 2 times, most recently from 9340358 to 4821dfd Compare August 13, 2026 06:01
@fzyzcjy
fzyzcjy force-pushed the tom/refactor-miles/op3-9 branch from a8e3e02 to 16da4ad Compare August 13, 2026 06:01
…sync code

These call sites are about to become direct HTTP calls to the engines'
own servers. Turn them into `async def` + `await` now, while they still
go through ray, so the switch is a one-line change per call site rather
than a control-flow change.

ServerGroup.offload / onload / onload_weights_from_disk stop handing
their caller a list of pending handles and await their own fan-out;
RolloutServer gathers over groups instead of over engines, so the whole
server still offloads concurrently.
@fzyzcjy
fzyzcjy force-pushed the tom/refactor-miles/op3-9 branch from 16da4ad to c8fa640 Compare August 14, 2026 04:45
@fzyzcjy
fzyzcjy force-pushed the tom/refactor-miles/op3-10 branch from 4821dfd to 07cde82 Compare August 14, 2026 04:46

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

Approved for the #2176 safe-merge checkpoint: the async fan-out preserves the existing filtering, concurrency, and result semantics, with no blocking finding through the checked chain tip.

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