Skip to content

fix: support DDP online rollout - #38

Closed
0z5a wants to merge 1 commit into
ViperEkura:mainfrom
0z5a:codex/fix-ddp-online-rollout-0z5a
Closed

fix: support DDP online rollout#38
0z5a wants to merge 1 commit into
ViperEkura:mainfrom
0z5a:codex/fix-ddp-online-rollout-0z5a

Conversation

@0z5a

@0z5a 0z5a commented Sep 2, 2026

Copy link
Copy Markdown
Contributor

This PR replaces #29 because the original commits were attributed to the wrong GitHub account. The code and validation evidence are unchanged; the commit author and committer are now correctly linked to 0z5a.

Summary

  • expose executor-owned training and inference model views
  • unwrap DDP for rollout inference and keep CPU DDP device handling valid
  • fail fast for unsupported FSDP and compiled online rollout combinations

Validation

  • CPU DDP focused tests: 11 passed
  • GPU5 focused DDP rollout tests: 2 passed
  • bash scripts/pre_commit.sh --skip-deps
  • 598 passed, 50 skipped

- Expose executor-owned training and inference model views and unwrap DDP for the scheduler
- Fail fast for unsupported FSDP and torch.compile rollout configurations
- Cover single-process and two-rank online GRPO and DPO execution
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