Skip to content

Commit 2989b0b

Browse files
authored
Build wheels for torch 2.1.1 (#1261)
1 parent 45450bf commit 2989b0b

File tree

3 files changed

+9
-3
lines changed

3 files changed

+9
-3
lines changed

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

+1-1
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ jobs:
5353
fail-fast: false
5454
matrix:
5555
os: [ubuntu-latest, macos-latest]
56-
torch: ["2.1.0"]
56+
torch: ["2.1.1"]
5757
python-version: ["3.8", "3.9", "3.10", "3.11"]
5858
build_type: ["Release", "Debug"]
5959

.github/workflows/run-tests.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ jobs:
5454
matrix:
5555
os: [ubuntu-latest]
5656
cuda: ["12.1"]
57-
torch: ["2.1.0"]
57+
torch: ["2.1.1"]
5858
python-version: ["3.11"]
5959

6060
steps:

scripts/github_actions/generate_build_matrix.py

+7-1
Original file line numberDiff line numberDiff line change
@@ -166,10 +166,16 @@ def generate_build_matrix(
166166
if not for_windows
167167
else ["11.8.0", "12.1.0"],
168168
},
169+
"2.1.1": {
170+
"python-version": ["3.8", "3.9", "3.10", "3.11"],
171+
"cuda": ["11.8", "12.1"] # default 12.1
172+
if not for_windows
173+
else ["11.8.0", "12.1.0"],
174+
},
169175
# https://github.com/Jimver/cuda-toolkit/blob/master/src/links/windows-links.ts
170176
}
171177
if test_only_latest_torch:
172-
latest = "2.1.0"
178+
latest = "2.1.1"
173179
matrix = {latest: matrix[latest]}
174180

175181
if for_windows or for_macos:

0 commit comments

Comments
 (0)