14
14
id-token : write
15
15
contents : read
16
16
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
21
20
timeout : 60
22
21
script : |
23
22
echo "::group::Print machine info"
45
44
runner : linux.arm64.2xlarge
46
45
docker-image : " pytorch/manylinux2_28_aarch64-builder:cpu-aarch64-main"
47
46
gpu-arch-type : cpu-aarch64
48
- gpu-arch-version : " 12.6"
49
47
timeout : 60
50
48
script : |
51
49
echo "::group::Print machine info"
@@ -61,11 +59,13 @@ jobs:
61
59
62
60
test-gguf-cpu :
63
61
uses : pytorch/test-infra/.github/workflows/linux_job.yml@main
62
+ permissions :
63
+ id-token : write
64
+ contents : read
64
65
with :
65
66
runner : linux.arm64.2xlarge
66
- docker-image : " manylinuxaarch64 -builder:cpu-aarch64-main"
67
+ docker-image : " pytorch/manylinux2_28_aarch64 -builder:cpu-aarch64-main"
67
68
gpu-arch-type : cpu-aarch64
68
- gpu-arch-version : " 12.6"
69
69
timeout : 60
70
70
script : |
71
71
echo "::group::Print machine info"
@@ -86,11 +86,13 @@ jobs:
86
86
87
87
test-advanced-cpu :
88
88
uses : pytorch/test-infra/.github/workflows/linux_job.yml@main
89
+ permissions :
90
+ id-token : write
91
+ contents : read
89
92
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
94
96
timeout : 60
95
97
script : |
96
98
echo "::group::Print machine info"
@@ -111,11 +113,13 @@ jobs:
111
113
112
114
test-evaluation-cpu :
113
115
uses : pytorch/test-infra/.github/workflows/linux_job.yml@main
116
+ permissions :
117
+ id-token : write
118
+ contents : read
114
119
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
119
123
timeout : 60
120
124
script : |
121
125
echo "::group::Print machine info"
0 commit comments