support torch 2.1.1 (#85)

This commit is contained in:
Fangjun Kuang 2023-11-26 17:04:52 +08:00 committed by GitHub
parent c1adbda9f0
commit 352aa43ce3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -153,10 +153,16 @@ def generate_build_matrix(enable_cuda, for_windows, for_macos, test_only_latest_
if not for_windows if not for_windows
else ["11.8.0", "12.1.0"], else ["11.8.0", "12.1.0"],
}, },
"2.1.1": {
"python-version": ["3.8", "3.9", "3.10", "3.11"],
"cuda": ["11.8", "12.1"] # default 12.1
if not for_windows
else ["11.8.0", "12.1.0"],
},
# https://github.com/Jimver/cuda-toolkit/blob/master/src/links/windows-links.ts # https://github.com/Jimver/cuda-toolkit/blob/master/src/links/windows-links.ts
} }
if test_only_latest_torch: if test_only_latest_torch:
latest = "2.1.0" latest = "2.1.1"
matrix = {latest: matrix[latest]} matrix = {latest: matrix[latest]}
if for_windows or for_macos: if for_windows or for_macos: