Skip to content

[DO NOT MERGE] Run-only: gb300 dsr1 measured power+temp validation#1686

Open
arygupt wants to merge 5 commits into
mainfrom
feat/dsr1-gb300-powercheck-run
Open

[DO NOT MERGE] Run-only: gb300 dsr1 measured power+temp validation#1686
arygupt wants to merge 5 commits into
mainfrom
feat/dsr1-gb300-powercheck-run

Conversation

@arygupt

@arygupt arygupt commented Jun 8, 2026

Copy link
Copy Markdown
Collaborator

⚠️ DO NOT MERGE — run-only validation PR

Purpose: produce the first measured power + temperature data for dsr1 disagg on GB300 now, without waiting for #1574 to merge.

This branch carries #1574's self-contained consumer code (utils/aggregate_power.py, utils/process_result.py + their tests) on top of main, plus the gb300-nv launcher perfmon wiring and a single 1-job changelog entry (dsr1-fp4-gb300-dynamo-sglang-powercheck). The changelog diff expands to exactly one gb300 job (1k/1k, conc 8, 1×prefill TP4 + 2×decode TP4) — verified with process_changelog.py --trim-conc.

Power AND temperature ride the same pipeline: the srt-slurm fork perfmon captures power.draw + temperature.gpu + utilization.gpu + memory.used; aggregate_power.py emits avg_power_w, per-stage prefill_avg_power_w/decode_avg_power_w, and per-worker avg_temp_c/peak_temp_c/avg_util_pct/avg_mem_used_mb.

Why not merge: it duplicates #1574's aggregate_power.py. The production landing of the gb300-nv launcher happens via the clean post-#1574 path (cherry-pick onto merged main). This PR is closed once the measured data lands.

Success criteria: job green + perf_samples_*.csv staged + agg JSON patched with the power/temp fields above.

🤖 Generated with Claude Code


Note

Medium Risk
Changes benchmark result JSON schema and multinode launcher behavior (fork checkout, recipe mutation, env-driven CSV selection); mistakes could publish wrong power numbers or break GB300 jobs, though aggregation remains best-effort and tests are broad.

Overview
Adds a GB300 measured-power validation path for DeepSeek-R1 FP4 disagg before the full NVIDIA sweep: a single-job benchmark config (dsr1-fp4-gb300-dynamo-sglang-powercheck), changelog entry, and runner wiring on gb300-cw and gb300-nv to clone the srt-slurm perfmon fork, inject monitoring: into recipes, stage perf_samples_*.csv via GPU_METRICS_CSV_GLOB, and fix CoreWeave recipe gaps (unlimited mem, longer health checks).

Telemetry pipeline expands aggregate_power.py and process_result.py to merge multi-node CSVs (GPU index namespacing), parse perfmon filenames for workers[], support disagg per-stage power/joules (prefill_avg_power_w, decode_avg_power_w, joules_per_input_token), add temp/util/memory fields, and resolve bench windows from srt-slurm date when Unix timestamps are absent. Multinode runs must not fall back to stale single-node gpu_metrics.csv when the glob is set.

CI reliability: multinode workflow pre/post cleanup now sudo rm -rf benchmark_logs so root-owned leftovers from cancelled jobs do not break checkout on shared runners.

Extensive tests cover multinode aggregation, disagg attribution, and process_result glob/disagg wiring.

Reviewed by Cursor Bugbot for commit 63cbe19. Bugbot is set up for automated code reviews on this repo. Configure here.

Standalone branch off main to RUN the gb300 dsr1 perfmon validation without waiting for #1574 to merge. Carries #1574's self-contained consumer code (utils/aggregate_power.py, utils/process_result.py + tests) so the Process-result step patches the agg JSON, plus the gb300-nv launcher perfmon wiring and a single 1-job changelog entry (dsr1-fp4-gb300-dynamo-sglang-powercheck).

Power AND temperature ride the same pipeline: the srt-slurm fork perfmon captures power.draw + temperature.gpu + utilization.gpu + memory.used; aggregate_power.py emits avg_power_w, per-stage prefill/decode power, and per-worker avg_temp_c/peak_temp_c/avg_util_pct/avg_mem_used_mb.

DO NOT MERGE ahead of #1574 — this duplicates #1574's aggregate_power.py. It exists to produce the measured dsr1-gb300 data now; close after the data lands. Production landing happens via the clean post-#1574 path.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
@github-actions

github-actions Bot commented Jun 8, 2026

Copy link
Copy Markdown
Contributor

Thanks for the contribution! For vLLM & SGLang, please ensure that your recipes is similar to the official vLLM recipes and/or the SGLang cookbook

If it is not, please create a PR first before we can merge your single node PR into the master branch. Let's ensure that the documentation is first class such that the entire ML community can benefit from your hard work! Thank you

PR authors are responsible for ensuring that after merging, all GitHub Action jobs fully pass. A lot of the time, failures are just flakes and simply re-running the failed jobs will fix it. If re-running failed jobs is attempted, PR authors are responsible for ensuring it passes. See GitHub's docs on re-running failed jobs: https://docs.github.com/en/actions/how-tos/manage-workflow-runs/re-run-workflows-and-jobs#re-running-failed-jobs-in-a-workflow

As a rule of thumb, generally, PR authors should request a review & get a PR approval from the respective companies' CODEOWNERS before requesting a review from core maintainers.

If additional help is needed, PR authors can reach out to core maintainers over Slack.

Comment thread utils/aggregate_power.py


# Back-compat shim — some external callers may have imported _parse_power.
_parse_power = _parse_numeric_cell
perf_csv_count=$(ls "$LOGS_DIR"/perf_samples_*.csv 2>/dev/null | wc -l | tr -d ' ')
if [ "$perf_csv_count" -gt 0 ]; then
mkdir -p "$GITHUB_WORKSPACE/perf_samples"
cp "$LOGS_DIR"/perf_samples_*.csv "$GITHUB_WORKSPACE/perf_samples/"

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Stale perf CSVs not cleared

Medium Severity

Measured-power staging copies into $GITHUB_WORKSPACE/perf_samples/ without clearing that directory first. process_result.py globs every perf_samples_*.csv there, so files left from an earlier job on the same runner can be aggregated with the current run and skew avg_power_w, worker counts, and joules fields.

Fix in Cursor Fix in Web

Reviewed by Cursor Bugbot for commit 9839a0c. Configure here.

@github-actions

github-actions Bot commented Jun 8, 2026

Copy link
Copy Markdown
Contributor

@github-actions

github-actions Bot commented Jun 8, 2026

Copy link
Copy Markdown
Contributor

…gb300-cw)

Retry on gb300-cw_2 failed instantly: launch_gb300-cw.sh hard-rejects dsr1 ('Unsupported model prefix/precision combination on gb300-cw: dsr1/fp4'). The bare 'gb300' runs-on label is shared across the NVIDIA (gb300-nv_*) and CoreWeave (gb300-cw_*) pools, and only launch_gb300-nv.sh handles dsr1+perfmon. Pinning to the gb300-nv label keeps it on the NVIDIA pool.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
echo "[perfmon] staged $perf_csv_count per-node perf_samples_*.csv to \$GITHUB_WORKSPACE/perf_samples/"
else
echo "[perfmon] WARNING: monitoring enabled but no perf_samples_*.csv found in $LOGS_DIR — measured power aggregation will be skipped" >&2
fi

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Perfmon failure omits glob guard

Medium Severity

When PERFMON_ENABLED is set but no perf_samples_*.csv files are found, the launcher logs a warning and does not set GPU_METRICS_CSV_GLOB. process_result.py then uses the single-node gpu_metrics.csv fallback, which can patch multinode agg JSON with stale single-node power instead of omitting telemetry.

Additional Locations (1)
Fix in Cursor Fix in Web

Reviewed by Cursor Bugbot for commit 5b3eb12. Configure here.

@github-actions

github-actions Bot commented Jun 8, 2026

Copy link
Copy Markdown
Contributor

@github-actions

github-actions Bot commented Jun 8, 2026

Copy link
Copy Markdown
Contributor

…eave)

The gb300-nv runner fleet wedged on its pre-run cleanup (stale NFS), so pivot the validation to the healthy CoreWeave GB300 pool. Adds a dsr1/fp4 branch to launch_gb300-cw.sh that clones the perfmon fork (reusing its recipes/gb300-fp4 dsr1 recipes), maps model.path 'dsr1' -> /mnt/vast/models/dsr1-fp4 (weights pre-staged on CW), injects monitoring: into the recipe, and stages perf_samples_*.csv + GPU_METRICS_CSV_GLOB for Process-result. Config runner gb300-nv -> gb300-cw.

CoreWeave-segment data (proxy for the NVIDIA campaign numbers) — validates the full perfmon -> aggregate_power -> power+temp pipeline end-to-end today while the gb300-nv fleet is fixed separately.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>

@cursor cursor Bot left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Cursor Bugbot has reviewed your changes and found 1 potential issue.

There are 3 total unresolved issues (including 2 from previous reviews).

Fix All in Cursor

❌ Bugbot Autofix is OFF. To automatically fix reported issues with cloud agents, enable autofix in the Cursor dashboard.

Reviewed by Cursor Bugbot for commit ce2c4ba. Configure here.

echo "[perfmon] WARNING: zero recipe YAMLs found under recipes/ — power data will be MISSING from this run." >&2
else
echo "[perfmon] injected monitoring: into $INJECTED_COUNT of $FOUND_COUNT recipes."
fi

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

dsr1 branch ignores framework

High Severity

The new dsr1 clone path keys only on MODEL_PREFIX, so any GB300-NV job with model-prefix: dsr1—including dynamo-trt configs—checks out the perfmon srt-slurm fork, enables PERFMON_ENABLED, and rewrites every recipe YAML. TRT (and other non–dynamo-sglang) dsr1 runs previously used the default sa-submission-q2-2026 checkout and can fail or run the wrong stack.

Fix in Cursor Fix in Web

Reviewed by Cursor Bugbot for commit ce2c4ba. Configure here.

@github-actions

github-actions Bot commented Jun 8, 2026

Copy link
Copy Markdown
Contributor

…hfs OOM)

First Path B run (job 6867) failed: pyxis killed unsquashfs (exit 137) extracting the container on the prefill node -> etcd never started -> decode workers crashed with 'Could not connect to etcd'. Root cause: the fork's gb300-fp4 recipe omits sbatch_directives.mem, so SLURM capped node memory and the cgroup OOM-killed the ~15-30GB squashfs extraction. Every working CW recipe (dsv4/glm5) sets mem: "0"; inject it (idempotent) alongside monitoring:.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
@github-actions

github-actions Bot commented Jun 8, 2026

Copy link
Copy Markdown
Contributor

… > 30min)

Run 6870 reached full server bringup but timed out: 'Model did not get healthy in 1800 seconds' while decode was still capturing CUDA graphs (25%, bs=184). dsr1 warmup = load 671B weights + FlashInfer autotune + capture 36 graph sizes (cuda-graph-max-bs 256) ~= 35min, over the default health_check (max_attempts 180 x interval 10 = 1800s). On timeout the orchestrator killed etcd -> lease errors -> teardown (not a crash). Inject health_check max_attempts=540 interval=10 (5400s/90min). It's a ceiling; the sweep proceeds as soon as servers are healthy.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
@github-actions

github-actions Bot commented Jun 9, 2026

Copy link
Copy Markdown
Contributor

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

Status: No status

Development

Successfully merging this pull request may close these issues.

1 participant