Skip to content

Build the Linux wheels on OSDC runners #22753

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
huydhn wants to merge 2 commits into
base: main
Choose a base branch
from
+20 −16
Open

Build the Linux wheels on OSDC runners #22753

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
a759ebb
Build the Linux wheels on OSDC runners
huydhn Sep 12, 2026
0d1e0b6
TEMPORARY: test against pytorch/test-infra#8789
huydhn Sep 12, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 5 additions & 4 deletions .github/workflows/build-wheels-aarch64-linux.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -35,17 +35,18 @@

jobs:
generate-matrix:
uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@build-wheels-linux-osdc-workspace
with:
package-type: wheel
os: linux-aarch64
test-infra-repository: pytorch/test-infra
test-infra-ref: main
test-infra-ref: build-wheels-linux-osdc-workspace
runner-fleet: osdc
with-cuda: disabled
with-rocm: disabled
python-versions: '["3.10", "3.11", "3.12", "3.13", "3.14"]'

build:

Check warning

Code scanning / CodeQL

Workflow does not contain permissions Medium

Actions job or workflow does not limit the permissions of the GITHUB_TOKEN. Consider setting an explicit permissions block, using the following as a minimal starting point: {}
needs: generate-matrix
permissions:
id-token: write
Expand All @@ -60,12 +61,12 @@
smoke-test-script: .ci/scripts/wheel/test_linux_aarch64.py
package-name: executorch
name: ${{ matrix.repository }}
uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@build-wheels-linux-osdc-workspace
with:
repository: ${{ matrix.repository }}
ref: ""
test-infra-repository: pytorch/test-infra
test-infra-ref: main
test-infra-ref: build-wheels-linux-osdc-workspace
build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
submodules: recursive
env-var-script: .ci/scripts/wheel/envvar_linux.sh
Expand Down
9 changes: 5 additions & 4 deletions .github/workflows/build-wheels-cuda-aarch64-linux.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -39,23 +39,24 @@

jobs:
generate-matrix:
uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@build-wheels-linux-osdc-workspace
with:
package-type: wheel
os: linux-aarch64
test-infra-repository: pytorch/test-infra
test-infra-ref: main
test-infra-ref: build-wheels-linux-osdc-workspace
runner-fleet: osdc
with-cuda: enable
with-cpu: disable
with-rocm: disable
python-versions: '["3.10", "3.11", "3.12", "3.13", "3.14"]'

# The generator emits every CUDA version it knows about. Publishing all of them would ship
# wheels for combinations nothing can verify, so this keeps only the rows a device exists for
# and a matching PyTorch is published for. These runners carry no accelerator, unlike the
# x86_64 ones, so the smoke test here checks packaging rather than execution. The script fails
# rather than emitting an empty matrix, because a workflow with no build job reads as a pass.
filter-matrix:

Check warning

Code scanning / CodeQL

Workflow does not contain permissions Medium

Actions job or workflow does not limit the permissions of the GITHUB_TOKEN. Consider setting an explicit permissions block, using the following as a minimal starting point: {}
needs: generate-matrix
runs-on: ubuntu-latest
outputs:
Expand Down Expand Up @@ -94,12 +95,12 @@
smoke-test-script: .ci/scripts/wheel/test_cuda_linux.py
package-name: executorch
name: ${{ matrix.repository }}
uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@build-wheels-linux-osdc-workspace
with:
repository: ${{ matrix.repository }}
ref: ""
test-infra-repository: pytorch/test-infra
test-infra-ref: main
test-infra-ref: build-wheels-linux-osdc-workspace
build-matrix: ${{ needs.filter-matrix.outputs.matrix }}
submodules: recursive
env-var-script: .ci/scripts/wheel/envvar_cuda_linux.sh
Expand Down
9 changes: 5 additions & 4 deletions .github/workflows/build-wheels-cuda-linux.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -39,22 +39,23 @@

jobs:
generate-matrix:
uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@build-wheels-linux-osdc-workspace
with:
package-type: wheel
os: linux
test-infra-repository: pytorch/test-infra
test-infra-ref: main
test-infra-ref: build-wheels-linux-osdc-workspace
runner-fleet: osdc
with-cuda: enable
with-cpu: disable
with-rocm: disable
python-versions: '["3.10", "3.11", "3.12", "3.13", "3.14"]'

# The generator emits every CUDA version it knows about. Publishing all of them would ship
# wheels for combinations nothing can verify, so this keeps only the rows with a GPU to run
# them on. The script fails rather than emitting an empty matrix, because a workflow with no
# build job reads as a pass.
filter-matrix:

Check warning

Code scanning / CodeQL

Workflow does not contain permissions Medium

Actions job or workflow does not limit the permissions of the GITHUB_TOKEN. Consider setting an explicit permissions block, using the following as a minimal starting point: {}
needs: generate-matrix
runs-on: ubuntu-latest
outputs:
Expand Down Expand Up @@ -93,12 +94,12 @@
smoke-test-script: .ci/scripts/wheel/test_cuda_linux.py
package-name: executorch
name: ${{ matrix.repository }}
uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@build-wheels-linux-osdc-workspace
with:
repository: ${{ matrix.repository }}
ref: ""
test-infra-repository: pytorch/test-infra
test-infra-ref: main
test-infra-ref: build-wheels-linux-osdc-workspace
build-matrix: ${{ needs.filter-matrix.outputs.matrix }}
submodules: recursive
env-var-script: .ci/scripts/wheel/envvar_cuda_linux.sh
Expand Down
9 changes: 5 additions & 4 deletions .github/workflows/build-wheels-linux.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -35,17 +35,18 @@

jobs:
generate-matrix:
uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@build-wheels-linux-osdc-workspace
with:
package-type: wheel
os: linux
test-infra-repository: pytorch/test-infra
test-infra-ref: main
test-infra-ref: build-wheels-linux-osdc-workspace
runner-fleet: osdc
with-cuda: disabled
with-rocm: disabled
python-versions: '["3.10", "3.11", "3.12", "3.13", "3.14"]'

build:

Check warning

Code scanning / CodeQL

Workflow does not contain permissions Medium

Actions job or workflow does not limit the permissions of the GITHUB_TOKEN. Consider setting an explicit permissions block, using the following as a minimal starting point: {}
needs: generate-matrix
permissions:
id-token: write
Expand All @@ -60,12 +61,12 @@
smoke-test-script: .ci/scripts/wheel/test_linux.py
package-name: executorch
name: ${{ matrix.repository }}
uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@build-wheels-linux-osdc-workspace
with:
repository: ${{ matrix.repository }}
ref: ""
test-infra-repository: pytorch/test-infra
test-infra-ref: main
test-infra-ref: build-wheels-linux-osdc-workspace
build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
submodules: recursive
env-var-script: .ci/scripts/wheel/envvar_linux.sh
Expand Down
Loading