Skip to content

Commit 174bb0f

Browse files
authored
Switch everything to CPU linux v2
1 parent 8af4416 commit 174bb0f

File tree

1 file changed

+19
-15
lines changed

1 file changed

+19
-15
lines changed

Diff for: .github/workflows/run-readme-pr-linuxaarch64.yml

+19-15
Original file line numberDiff line numberDiff line change
@@ -14,10 +14,9 @@ jobs:
1414
id-token: write
1515
contents: read
1616
with:
17-
runner: linux.arm64.m7g.4xlarge
18-
docker-image: "pytorch/manylinux2_28_aarch64-builder:cuda12.6"
19-
gpu-arch-type: cuda-aarch64
20-
gpu-arch-version: "12.6"
17+
runner: linux.arm64.2xlarge
18+
docker-image: "pytorch/manylinux2_28_aarch64-builder:cpu-aarch64-main"
19+
gpu-arch-type: cpu-aarch64
2120
timeout: 60
2221
script: |
2322
echo "::group::Print machine info"
@@ -45,7 +44,6 @@ jobs:
4544
runner: linux.arm64.2xlarge
4645
docker-image: "pytorch/manylinux2_28_aarch64-builder:cpu-aarch64-main"
4746
gpu-arch-type: cpu-aarch64
48-
gpu-arch-version: "12.6"
4947
timeout: 60
5048
script: |
5149
echo "::group::Print machine info"
@@ -61,11 +59,13 @@ jobs:
6159
6260
test-gguf-cpu:
6361
uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
62+
permissions:
63+
id-token: write
64+
contents: read
6465
with:
6566
runner: linux.arm64.2xlarge
66-
docker-image: "manylinuxaarch64-builder:cpu-aarch64-main"
67+
docker-image: "pytorch/manylinux2_28_aarch64-builder:cpu-aarch64-main"
6768
gpu-arch-type: cpu-aarch64
68-
gpu-arch-version: "12.6"
6969
timeout: 60
7070
script: |
7171
echo "::group::Print machine info"
@@ -86,11 +86,13 @@ jobs:
8686
8787
test-advanced-cpu:
8888
uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
89+
permissions:
90+
id-token: write
91+
contents: read
8992
with:
90-
runner: linux.arm64.m7g.4xlarge
91-
docker-image: "pytorch/manylinuxaarch64-builder:cuda12.6-main"
92-
gpu-arch-type: cuda-aarch64
93-
gpu-arch-version: "12.6"
93+
runner: linux.arm64.2xlarge
94+
docker-image: "pytorch/manylinux2_28_aarch64-builder:cpu-aarch64-main"
95+
gpu-arch-type: cpu-aarch64
9496
timeout: 60
9597
script: |
9698
echo "::group::Print machine info"
@@ -111,11 +113,13 @@ jobs:
111113
112114
test-evaluation-cpu:
113115
uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
116+
permissions:
117+
id-token: write
118+
contents: read
114119
with:
115-
runner: linux.arm64.m7g.4xlarge
116-
docker-image: "pytorch/manylinuxaarch64-builder:cuda12.6-main"
117-
gpu-arch-type: cuda-aarch64
118-
gpu-arch-version: "12.6"
120+
runner: linux.arm64.2xlarge
121+
docker-image: "pytorch/manylinux2_28_aarch64-builder:cpu-aarch64-main"
122+
gpu-arch-type: cpu-aarch64
119123
timeout: 60
120124
script: |
121125
echo "::group::Print machine info"

0 commit comments

Comments
 (0)