ci: rename 1/2-gpu-runner labels to 1/2-gpu-h100 (#21008)

This commit is contained in:
Lianmin Zheng
2026-03-20 06:04:15 -07:00
committed by GitHub
parent 104b10f70a
commit 2d7a262ca3
11 changed files with 57 additions and 57 deletions
+2 -2
View File
@@ -6,7 +6,7 @@ with metadata for the performance dashboard.
Usage:
python3 scripts/ci/utils/save_diffusion_metrics.py \
--gpu-config 1-gpu-runner \
--gpu-config 1-gpu-h100 \
--run-id 12345678 \
--output test/diffusion-metrics-1gpu.json \
--results-json test/diffusion-results.json
@@ -129,7 +129,7 @@ def main():
parser.add_argument(
"--gpu-config",
required=True,
help="GPU configuration (e.g., 1-gpu-runner, 2-gpu-runner)",
help="GPU configuration (e.g., 1-gpu-h100, 2-gpu-h100)",
)
parser.add_argument(
"--run-id",
+2 -2
View File
@@ -416,8 +416,8 @@ CUDA_SUITE_TO_RUNNER = {
"stage-a-test-small-1-gpu": "1-gpu-5090",
"stage-a-cpu-only": "ubuntu-latest",
"stage-b-test-small-1-gpu": "1-gpu-5090",
"stage-b-test-large-1-gpu": "1-gpu-runner",
"stage-b-test-large-2-gpu": "2-gpu-runner",
"stage-b-test-large-1-gpu": "1-gpu-h100",
"stage-b-test-large-2-gpu": "2-gpu-h100",
"stage-b-test-4-gpu-b200": "4-gpu-b200",
"stage-c-test-4-gpu-h100": "4-gpu-h100",
"stage-c-test-8-gpu-h200": "8-gpu-h200",