Skip to content

Commit 21302da

Browse files
csukuangfjcsukuangfj
authored
Publish pre-built wheels for torch 2.4.1 (#1303)
1 parent 5735fa7 commit 21302da

3 files changed

Lines changed: 117 additions & 70 deletions

File tree

‎.github/workflows/run-tests-cpu.yml‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ jobs:
5252
fail-fast: false
5353
matrix:
5454
os: [ubuntu-latest, macos-latest]
55-
torch: ["2.4.0"]
55+
torch: ["2.4.1"]
5656
python-version: ["3.8", "3.9", "3.10", "3.11", "3.12"]
5757
build_type: ["Release", "Debug"]
5858

‎.github/workflows/run-tests.yml‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ jobs:
5252
matrix:
5353
os: [ubuntu-latest]
5454
cuda: ["12.4"]
55-
torch: ["2.4.0"]
55+
torch: ["2.4.1"]
5656
python-version: ["3.12"]
5757

5858
steps:

‎scripts/github_actions/generate_build_matrix.py‎

Lines changed: 115 additions & 68 deletions
Original file line numberDiff line numberDiff line change
@@ -84,27 +84,35 @@ def generate_build_matrix(
8484
},
8585
"1.7.0": {
8686
"python-version": ["3.6", "3.7", "3.8"],
87-
"cuda": ["10.1", "10.2", "11.0"]
88-
if not for_windows
89-
else ["10.1.243", "10.2.89", "11.0.3"],
87+
"cuda": (
88+
["10.1", "10.2", "11.0"]
89+
if not for_windows
90+
else ["10.1.243", "10.2.89", "11.0.3"]
91+
),
9092
},
9193
"1.7.1": {
9294
"python-version": ["3.6", "3.7", "3.8", "3.9"],
93-
"cuda": ["10.1", "10.2", "11.0"]
94-
if not for_windows
95-
else ["10.1.243", "10.2.89", "11.0.3"],
95+
"cuda": (
96+
["10.1", "10.2", "11.0"]
97+
if not for_windows
98+
else ["10.1.243", "10.2.89", "11.0.3"]
99+
),
96100
},
97101
"1.8.0": {
98102
"python-version": ["3.6", "3.7", "3.8", "3.9"],
99-
"cuda": ["10.1", "10.2", "11.1"]
100-
if not for_windows
101-
else ["10.1.243", "10.2.89", "11.1.1"],
103+
"cuda": (
104+
["10.1", "10.2", "11.1"]
105+
if not for_windows
106+
else ["10.1.243", "10.2.89", "11.1.1"]
107+
),
102108
},
103109
"1.8.1": {
104110
"python-version": ["3.6", "3.7", "3.8", "3.9"],
105-
"cuda": ["10.1", "10.2", "11.1"]
106-
if not for_windows
107-
else ["10.1.243", "10.2.89", "11.1.1"],
111+
"cuda": (
112+
["10.1", "10.2", "11.1"]
113+
if not for_windows
114+
else ["10.1.243", "10.2.89", "11.1.1"]
115+
),
108116
},
109117
"1.9.0": {
110118
"python-version": ["3.6", "3.7", "3.8", "3.9"],
@@ -116,120 +124,158 @@ def generate_build_matrix(
116124
},
117125
"1.10.0": {
118126
"python-version": ["3.6", "3.7", "3.8", "3.9"],
119-
"cuda": ["10.2", "11.1", "11.3"]
120-
if not for_windows
121-
else ["10.2.89", "11.1.1", "11.3.1"],
127+
"cuda": (
128+
["10.2", "11.1", "11.3"]
129+
if not for_windows
130+
else ["10.2.89", "11.1.1", "11.3.1"]
131+
),
122132
},
123133
"1.10.1": {
124134
"python-version": ["3.6", "3.7", "3.8", "3.9"],
125-
"cuda": ["10.2", "11.1", "11.3"]
126-
if not for_windows
127-
else ["10.2.89", "11.1.1", "11.3.1"],
135+
"cuda": (
136+
["10.2", "11.1", "11.3"]
137+
if not for_windows
138+
else ["10.2.89", "11.1.1", "11.3.1"]
139+
),
128140
},
129141
"1.10.2": {
130142
"python-version": ["3.6", "3.7", "3.8", "3.9"],
131-
"cuda": ["10.2", "11.1", "11.3"]
132-
if not for_windows
133-
else ["10.2.89", "11.1.1", "11.3.1"],
143+
"cuda": (
144+
["10.2", "11.1", "11.3"]
145+
if not for_windows
146+
else ["10.2.89", "11.1.1", "11.3.1"]
147+
),
134148
},
135149
"1.11.0": {
136150
"python-version": ["3.7", "3.8", "3.9", "3.10"],
137-
"cuda": ["10.2", "11.3", "11.5"]
138-
if not for_windows
139-
else ["11.3.1", "11.5.2"],
151+
"cuda": (
152+
["10.2", "11.3", "11.5"] if not for_windows else ["11.3.1", "11.5.2"]
153+
),
140154
},
141155
"1.12.0": {
142156
"python-version": ["3.7", "3.8", "3.9", "3.10"],
143-
"cuda": ["10.2", "11.3", "11.6"]
144-
if not for_windows
145-
else ["11.3.1", "11.6.2"],
157+
"cuda": (
158+
["10.2", "11.3", "11.6"] if not for_windows else ["11.3.1", "11.6.2"]
159+
),
146160
},
147161
"1.12.1": {
148162
"python-version": ["3.7", "3.8", "3.9", "3.10"],
149-
"cuda": ["10.2", "11.3", "11.6"]
150-
if not for_windows
151-
else ["11.3.1", "11.6.2"],
163+
"cuda": (
164+
["10.2", "11.3", "11.6"] if not for_windows else ["11.3.1", "11.6.2"]
165+
),
152166
},
153167
"1.13.0": {
154168
"python-version": ["3.7", "3.8", "3.9", "3.10", "3.11"],
155169
"cuda": ["11.6", "11.7"], # default 11.7
156170
},
157171
"1.13.1": {
158172
"python-version": ["3.7", "3.8", "3.9", "3.10", "3.11"],
159-
"cuda": ["11.6", "11.7"] # default 11.7
160-
if not for_windows
161-
else ["11.6.2", "11.7.1"],
173+
"cuda": (
174+
["11.6", "11.7"] # default 11.7
175+
if not for_windows
176+
else ["11.6.2", "11.7.1"]
177+
),
162178
},
163179
"2.0.0": {
164180
"python-version": ["3.8", "3.9", "3.10", "3.11"],
165-
"cuda": ["11.7", "11.8"] # default 11.7
166-
if not for_windows
167-
else ["11.7.1", "11.8.0"],
181+
"cuda": (
182+
["11.7", "11.8"] # default 11.7
183+
if not for_windows
184+
else ["11.7.1", "11.8.0"]
185+
),
168186
},
169187
"2.0.1": {
170188
"python-version": ["3.8", "3.9", "3.10", "3.11"],
171-
"cuda": ["11.7", "11.8"] # default 11.7
172-
if not for_windows
173-
else ["11.7.1", "11.8.0"],
189+
"cuda": (
190+
["11.7", "11.8"] # default 11.7
191+
if not for_windows
192+
else ["11.7.1", "11.8.0"]
193+
),
174194
},
175195
"2.1.0": {
176196
"python-version": ["3.8", "3.9", "3.10", "3.11"],
177-
"cuda": ["11.8", "12.1"] # default 12.1
178-
if not for_windows
179-
else ["11.8.0", "12.1.0"],
197+
"cuda": (
198+
["11.8", "12.1"] # default 12.1
199+
if not for_windows
200+
else ["11.8.0", "12.1.0"]
201+
),
180202
},
181203
"2.1.1": {
182204
"python-version": ["3.8", "3.9", "3.10", "3.11"],
183-
"cuda": ["11.8", "12.1"] # default 12.1
184-
if not for_windows
185-
else ["11.8.0", "12.1.0"],
205+
"cuda": (
206+
["11.8", "12.1"] # default 12.1
207+
if not for_windows
208+
else ["11.8.0", "12.1.0"]
209+
),
186210
},
187211
"2.1.2": {
188212
"python-version": ["3.8", "3.9", "3.10", "3.11"],
189-
"cuda": ["11.8", "12.1"] # default 12.1
190-
if not for_windows
191-
else ["11.8.0", "12.1.0"],
213+
"cuda": (
214+
["11.8", "12.1"] # default 12.1
215+
if not for_windows
216+
else ["11.8.0", "12.1.0"]
217+
),
192218
},
193219
"2.2.0": {
194220
"python-version": ["3.8", "3.9", "3.10", "3.11", "3.12"],
195-
"cuda": ["11.8", "12.1"] # default 12.1
196-
if not for_windows
197-
else ["11.8.0", "12.1.0"],
221+
"cuda": (
222+
["11.8", "12.1"] # default 12.1
223+
if not for_windows
224+
else ["11.8.0", "12.1.0"]
225+
),
198226
},
199227
"2.2.1": {
200228
"python-version": ["3.8", "3.9", "3.10", "3.11", "3.12"],
201-
"cuda": ["11.8", "12.1"] # default 12.1
202-
if not for_windows
203-
else ["11.8.0", "12.1.0"],
229+
"cuda": (
230+
["11.8", "12.1"] # default 12.1
231+
if not for_windows
232+
else ["11.8.0", "12.1.0"]
233+
),
204234
},
205235
"2.2.2": {
206236
"python-version": ["3.8", "3.9", "3.10", "3.11", "3.12"],
207-
"cuda": ["11.8", "12.1"] # default 12.1
208-
if not for_windows
209-
else ["11.8.0", "12.1.0"],
237+
"cuda": (
238+
["11.8", "12.1"] # default 12.1
239+
if not for_windows
240+
else ["11.8.0", "12.1.0"]
241+
),
210242
},
211243
"2.3.0": {
212244
"python-version": ["3.8", "3.9", "3.10", "3.11", "3.12"],
213-
"cuda": ["11.8", "12.1"] # default 12.1
214-
if not for_windows
215-
else ["11.8.0", "12.1.0"],
245+
"cuda": (
246+
["11.8", "12.1"] # default 12.1
247+
if not for_windows
248+
else ["11.8.0", "12.1.0"]
249+
),
216250
},
217251
"2.3.1": {
218252
"python-version": ["3.8", "3.9", "3.10", "3.11", "3.12"],
219-
"cuda": ["11.8", "12.1"] # default 12.1
220-
if not for_windows
221-
else ["11.8.0", "12.1.0"],
253+
"cuda": (
254+
["11.8", "12.1"] # default 12.1
255+
if not for_windows
256+
else ["11.8.0", "12.1.0"]
257+
),
222258
},
223259
"2.4.0": {
224260
"python-version": ["3.8", "3.9", "3.10", "3.11", "3.12"],
225-
"cuda": ["11.8", "12.1", "12.4"] # default 12.1
226-
if not for_windows
227-
else ["11.8.0", "12.1.0", "12.4.0"],
261+
"cuda": (
262+
["11.8", "12.1", "12.4"] # default 12.1
263+
if not for_windows
264+
else ["11.8.0", "12.1.0", "12.4.0"]
265+
),
266+
},
267+
"2.4.1": {
268+
"python-version": ["3.8", "3.9", "3.10", "3.11", "3.12"],
269+
"cuda": (
270+
["11.8", "12.1", "12.4"] # default 12.1
271+
if not for_windows
272+
else ["11.8.0", "12.1.0", "12.4.0"]
273+
),
228274
},
229275
# https://github.com/Jimver/cuda-toolkit/blob/master/src/links/windows-links.ts
230276
}
231277
if test_only_latest_torch:
232-
latest = "2.4.0"
278+
latest = "2.4.1"
233279
matrix = {latest: matrix[latest]}
234280

235281
if for_windows or for_macos:
@@ -300,7 +346,8 @@ def generate_build_matrix(
300346
{
301347
"torch": torch,
302348
"python-version": p,
303-
"image": "pytorch/manylinux-builder:cpu-2.4",
349+
# "image": "pytorch/manylinux-builder:cpu-2.4",
350+
"image": "pytorch/manylinux-builder:cpu-27677ead7c8293c299a885ae2c474bf445e653a5",
304351
}
305352
)
306353
elif version_ge(torch, "2.2.0"):

0 commit comments

Comments
 (0)