From 44a97780608b16b457d844109f366f3fc9b7884f Mon Sep 17 00:00:00 2001 From: Huy Do Date: Sat, 12 Sep 2026 15:58:31 -0700 Subject: [PATCH] Build the Linux wheels on OSDC runners One input on the matrix generator. pytorch/test-infra#8789 adds runner-fleet, which maps each row's validation_runner to its ARC equivalent: release-isolated on the refs these workflows publish from, regular otherwise, so pull request builds move off EC2 too. x86_64 cpu mt-l-x86iavx512-8-64 mt-rel-l-x86iavx512-44-340 x86_64 cuda mt-l-x86aavx2-29-113-a10g mt-rel-l-x86aavx2-29-113-l4 aarch64 any mt-l-arm64g4-16-62 mt-rel-l-arm64g3-44-340 The cuda x86_64 rows keep a device, which they need: test_cuda_linux.py exports a model through the CUDA partitioner and runs it against eager, and on a CPU runner that check would skip rather than fail. Depends on pytorch/test-infra#8789, which also carries the two fixes build_wheels_linux.yml needs to run on an OSDC pod at all. Authored with Claude Code. --- .github/workflows/build-wheels-aarch64-linux.yml | 1 + .github/workflows/build-wheels-cuda-aarch64-linux.yml | 1 + .github/workflows/build-wheels-cuda-linux.yml | 1 + .github/workflows/build-wheels-linux.yml | 1 + 4 files changed, 4 insertions(+) diff --git a/.github/workflows/build-wheels-aarch64-linux.yml b/.github/workflows/build-wheels-aarch64-linux.yml index 52f4ba0ddfd..eb79a9786c0 100644 --- a/.github/workflows/build-wheels-aarch64-linux.yml +++ b/.github/workflows/build-wheels-aarch64-linux.yml @@ -41,6 +41,7 @@ jobs: os: linux-aarch64 test-infra-repository: pytorch/test-infra test-infra-ref: main + runner-fleet: osdc with-cuda: disabled with-rocm: disabled python-versions: '["3.10", "3.11", "3.12", "3.13", "3.14"]' diff --git a/.github/workflows/build-wheels-cuda-aarch64-linux.yml b/.github/workflows/build-wheels-cuda-aarch64-linux.yml index ae646cb3bb4..882184d4dcc 100644 --- a/.github/workflows/build-wheels-cuda-aarch64-linux.yml +++ b/.github/workflows/build-wheels-cuda-aarch64-linux.yml @@ -45,6 +45,7 @@ jobs: os: linux-aarch64 test-infra-repository: pytorch/test-infra test-infra-ref: main + runner-fleet: osdc with-cuda: enable with-cpu: disable with-rocm: disable diff --git a/.github/workflows/build-wheels-cuda-linux.yml b/.github/workflows/build-wheels-cuda-linux.yml index 0a4b758e661..8de4ae4036c 100644 --- a/.github/workflows/build-wheels-cuda-linux.yml +++ b/.github/workflows/build-wheels-cuda-linux.yml @@ -45,6 +45,7 @@ jobs: os: linux test-infra-repository: pytorch/test-infra test-infra-ref: main + runner-fleet: osdc with-cuda: enable with-cpu: disable with-rocm: disable diff --git a/.github/workflows/build-wheels-linux.yml b/.github/workflows/build-wheels-linux.yml index 87354c83bba..15e6af41fb4 100644 --- a/.github/workflows/build-wheels-linux.yml +++ b/.github/workflows/build-wheels-linux.yml @@ -41,6 +41,7 @@ jobs: os: linux test-infra-repository: pytorch/test-infra test-infra-ref: main + runner-fleet: osdc with-cuda: disabled with-rocm: disabled python-versions: '["3.10", "3.11", "3.12", "3.13", "3.14"]'