Skip to content

Commit 1c86a9e

Browse files
committed
Migrate A100 CUDA CI jobs to OSDC runners
Move the A100-dependent jobs in cuda.yml (export-model-cuda-artifact, test-model-cuda-e2e) and cuda-perf.yml (export-models, benchmark-cuda) from pytorch/test-infra linux_job_v2 (AWS) to linux_job_v3 (OSDC/ARC). Runner labels are remapped per pytorch/.github/arc.yaml: linux.aws.a100 -> mt-l-x86iavx512-11-125-a100 and the A10G fallback linux.g5.4xlarge.nvidia.gpu -> mt-l-x86aavx2-29-113-a10g. Jobs that never run on A100 stay on linux_job_v2 / linux.g5.4xlarge.nvidia.gpu. Authored with Claude Code.
1 parent 129c687 commit 1c86a9e

2 files changed

Lines changed: 8 additions & 8 deletions

File tree

.github/workflows/cuda-perf.yml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -124,7 +124,7 @@ jobs:
124124
export-models:
125125
name: export-models
126126
needs: set-parameters
127-
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
127+
uses: pytorch/test-infra/.github/workflows/linux_job_v3.yml@main
128128
permissions:
129129
id-token: write
130130
contents: read
@@ -135,7 +135,7 @@ jobs:
135135
with:
136136
timeout: 90
137137
secrets-env: EXECUTORCH_HF_TOKEN
138-
runner: ${{ contains(matrix.model, 'Qwen3.5-35B-A3B') && 'linux.aws.a100' || 'linux.g5.4xlarge.nvidia.gpu' }}
138+
runner: ${{ contains(matrix.model, 'Qwen3.5-35B-A3B') && 'mt-l-x86iavx512-11-125-a100' || 'mt-l-x86aavx2-29-113-a10g' }}
139139
gpu-arch-type: cuda
140140
gpu-arch-version: "13.0"
141141
use-custom-docker-registry: false
@@ -192,7 +192,7 @@ jobs:
192192
contains(needs.changed-files.outputs.changed-files, '.ci/scripts/test_model_e2e.sh') ||
193193
needs.run-decision.outputs.is-full-run == 'true'
194194
)
195-
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
195+
uses: pytorch/test-infra/.github/workflows/linux_job_v3.yml@main
196196
permissions:
197197
id-token: write
198198
contents: read
@@ -201,7 +201,7 @@ jobs:
201201
fail-fast: false
202202
with:
203203
timeout: 90
204-
runner: ${{ contains(matrix.model, 'Qwen3.5-35B-A3B') && 'linux.aws.a100' || 'linux.g5.4xlarge.nvidia.gpu' }}
204+
runner: ${{ contains(matrix.model, 'Qwen3.5-35B-A3B') && 'mt-l-x86iavx512-11-125-a100' || 'mt-l-x86aavx2-29-113-a10g' }}
205205
gpu-arch-type: cuda
206206
gpu-arch-version: "13.0"
207207
use-custom-docker-registry: false

.github/workflows/cuda.yml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -229,7 +229,7 @@ jobs:
229229
contains(needs.changed-files.outputs.changed-files, '.ci/scripts/test_model_e2e.sh') ||
230230
needs.run-decision.outputs.is-full-run == 'true'
231231
)
232-
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
232+
uses: pytorch/test-infra/.github/workflows/linux_job_v3.yml@main
233233
permissions:
234234
id-token: write
235235
contents: read
@@ -342,7 +342,7 @@ jobs:
342342
with:
343343
timeout: 150
344344
secrets-env: EXECUTORCH_HF_TOKEN
345-
runner: ${{ (matrix.model.name == 'Qwen3.5-35B-A3B-HQQ-INT4' || matrix.model.name == 'gemma-4-31B-it-HQQ-INT4') && 'linux.aws.a100' || 'linux.g5.4xlarge.nvidia.gpu' }}
345+
runner: ${{ (matrix.model.name == 'Qwen3.5-35B-A3B-HQQ-INT4' || matrix.model.name == 'gemma-4-31B-it-HQQ-INT4') && 'mt-l-x86iavx512-11-125-a100' || 'mt-l-x86aavx2-29-113-a10g' }}
346346
gpu-arch-type: cuda
347347
gpu-arch-version: "13.0"
348348
use-custom-docker-registry: false
@@ -390,7 +390,7 @@ jobs:
390390
contains(needs.changed-files.outputs.changed-files, '.ci/scripts/test_model_e2e.sh') ||
391391
needs.run-decision.outputs.is-full-run == 'true'
392392
)
393-
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
393+
uses: pytorch/test-infra/.github/workflows/linux_job_v3.yml@main
394394
permissions:
395395
id-token: write
396396
contents: read
@@ -494,7 +494,7 @@ jobs:
494494
quant: "non-quantized"
495495
with:
496496
timeout: 90
497-
runner: ${{ (matrix.model.name == 'Qwen3.5-35B-A3B-HQQ-INT4' || matrix.model.name == 'gemma-4-31B-it-HQQ-INT4') && 'linux.aws.a100' || 'linux.g5.4xlarge.nvidia.gpu' }}
497+
runner: ${{ (matrix.model.name == 'Qwen3.5-35B-A3B-HQQ-INT4' || matrix.model.name == 'gemma-4-31B-it-HQQ-INT4') && 'mt-l-x86iavx512-11-125-a100' || 'mt-l-x86aavx2-29-113-a10g' }}
498498
gpu-arch-type: cuda
499499
gpu-arch-version: "13.0"
500500
use-custom-docker-registry: false

0 commit comments

Comments
 (0)