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"
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