Skip to content

Move fully_async example to main codebase - #1920

Merged
zhuzilin merged 3 commits into
mainfrom
zilin/fully-async
May 18, 2026
Merged

Move fully_async example to main codebase#1920
zhuzilin merged 3 commits into
mainfrom
zilin/fully-async

Conversation

@zhuzilin

Copy link
Copy Markdown
Contributor

No description provided.

@zhuzilin
zhuzilin force-pushed the zilin/fully-async branch from 5d33f01 to c27e802 Compare May 18, 2026 11:30
@zhuzilin
zhuzilin merged commit f0bce74 into main May 18, 2026
15 checks passed
@zhuzilin
zhuzilin deleted the zilin/fully-async branch May 18, 2026 11:46
CalvinXKY pushed a commit to vllm-project/vime that referenced this pull request Jun 2, 2026
Port of THUDM/slime#1920 — moves the fully-async rollout from an example into
the core package so it's usable via
--rollout-function-path vime.rollout.fully_async_rollout.generate_rollout_fully_async.

- vime/rollout/fully_async_rollout.py (new): promoted from
  examples/fully_async/fully_async_rollout.py and translated to vime's vLLM seam
  (vime.rollout.vllm_rollout, args.vllm_server_concurrency). Carries slime's
  promotion refinements: docstring, logging, __all__, atexit cleanup,
  num_engines concurrency scaling, ABORTED->data_buffer redirect, fan-out handling.
- vime/rollout/vllm_rollout.py: generate_and_rm_group return type
  -> list[Sample] | list[list[Sample]] + comment (analog of slime's sglang_rollout
  change; supports --custom-generate-function-path fan-out into multiple samples).
- examples/fully_async/: remove the now-duplicated fully_async_rollout.py
  (moved to core); repoint run-qwen3-4b-fully_async.sh + README at the core
  --rollout-function-path. (vime keeps the 4B example rather than swapping to
  0.5B as upstream did; the CI test below covers 0.5B.)
- tests/test_qwen2.5_0.5B_fully_async_short.py (new) + registered in the
  run-ci-short matrix (pr-test.yml + .j2 regenerated). Mirrors
  test_qwen2.5_0.5B_async_short, only flipping the rollout-function-path.

Refs: THUDM/slime#1920, #107

Signed-off-by: aoshen02 <aoshen@inferact.ai>
Co-authored-by: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
momo609 pushed a commit to vllm-project/vime that referenced this pull request Jun 8, 2026
Port of THUDM/slime#1920 — moves the fully-async rollout from an example into
the core package so it's usable via
--rollout-function-path vime.rollout.fully_async_rollout.generate_rollout_fully_async.

- vime/rollout/fully_async_rollout.py (new): promoted from
  examples/fully_async/fully_async_rollout.py and translated to vime's vLLM seam
  (vime.rollout.vllm_rollout, args.vllm_server_concurrency). Carries slime's
  promotion refinements: docstring, logging, __all__, atexit cleanup,
  num_engines concurrency scaling, ABORTED->data_buffer redirect, fan-out handling.
- vime/rollout/vllm_rollout.py: generate_and_rm_group return type
  -> list[Sample] | list[list[Sample]] + comment (analog of slime's sglang_rollout
  change; supports --custom-generate-function-path fan-out into multiple samples).
- examples/fully_async/: remove the now-duplicated fully_async_rollout.py
  (moved to core); repoint run-qwen3-4b-fully_async.sh + README at the core
  --rollout-function-path. (vime keeps the 4B example rather than swapping to
  0.5B as upstream did; the CI test below covers 0.5B.)
- tests/test_qwen2.5_0.5B_fully_async_short.py (new) + registered in the
  run-ci-short matrix (pr-test.yml + .j2 regenerated). Mirrors
  test_qwen2.5_0.5B_async_short, only flipping the rollout-function-path.

Refs: THUDM/slime#1920, #107

Signed-off-by: aoshen02 <aoshen@inferact.ai>
Co-authored-by: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant