ci: rename 1/2-gpu-runner labels to 1/2-gpu-h100 (#21008)
This commit is contained in:
@@ -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",
|
||||
|
||||
@@ -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",
|
||||
|
||||
Reference in New Issue
Block a user