From 7fa40337b70edd9c293dc6c4806450235643b2dc Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Fri, 24 Nov 2023 11:36:11 +0800 Subject: [PATCH] support torch 2.1.1 --- scripts/github_actions/generate_build_matrix.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/scripts/github_actions/generate_build_matrix.py b/scripts/github_actions/generate_build_matrix.py index 151a005..60fc06d 100755 --- a/scripts/github_actions/generate_build_matrix.py +++ b/scripts/github_actions/generate_build_matrix.py @@ -153,10 +153,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.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 } if test_only_latest_torch: - latest = "2.1.0" + latest = "2.1.1" matrix = {latest: matrix[latest]} if for_windows or for_macos: