support torch 2.3.1

This commit is contained in:
Fangjun Kuang 2024-06-06 10:15:25 +08:00
parent 40cc0a4a2c
commit 2efc89738d

View File

@ -201,10 +201,16 @@ def generate_build_matrix(enable_cuda, for_windows, for_macos, test_only_latest_
if not for_windows
else ["11.8.0", "12.1.0"],
},
"2.3.1": {
"python-version": ["3.8", "3.9", "3.10", "3.11", "3.12"],
"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
}
if test_only_latest_torch:
latest = "2.3.0"
latest = "2.3.1"
matrix = {latest: matrix[latest]}
if for_windows or for_macos: