Skip to content

[DRAFT] feat(mlperf): add MLPerf RCP logging support for distributed GSM8K GRPO - #2317

Draft
susanbao wants to merge 2 commits into
atwigg/mlperffrom
sanbao/rcp
Draft

susanbao wants to merge 2 commits into
atwigg/mlperffrom
sanbao/rcp

Conversation

@susanbao

@susanbao susanbao commented Sep 20, 2026

Copy link
Copy Markdown
Collaborator

Summary

  • Add MLPerf Reference Convergence Point (RCP) logging support (mllog_utils.log_rcp_step_stats, GCS log upload/flushing, and init_print metadata fields).
  • Track policy_training_time, exposed_generation_time, weight_sync_time, global_valid_toks, and global_valid_seqs in StandardRLProgram.
  • Wire --rcp_logging and MLPerf configuration flags into run_gsm8k_dist_grpo.py and k8s_launcher.sh.
  • Install mlperf-logging in Dockerfile.

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