From f5a59be48462aeacf093bb34dc329625cd4251ae Mon Sep 17 00:00:00 2001 From: Masoud Azizi Date: Fri, 22 Nov 2024 09:52:24 +0330 Subject: [PATCH] Update generate_build_matrix.py (#1309) --- scripts/github_actions/generate_build_matrix.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/scripts/github_actions/generate_build_matrix.py b/scripts/github_actions/generate_build_matrix.py index 09bd4e30f..abe1582ad 100755 --- a/scripts/github_actions/generate_build_matrix.py +++ b/scripts/github_actions/generate_build_matrix.py @@ -281,13 +281,25 @@ def generate_build_matrix( else ["11.8.0", "12.1.0", "12.4.0"] ), }, + "2.5.1": { + # Only Linux supports python 3.13 + "python-version": ["3.9", "3.10", "3.11", "3.12", "3.13"], + "cuda": ( + ["11.8", "12.1", "12.4"] # default 12.4 + if not for_windows + else ["11.8.0", "12.1.0", "12.4.0"] + ), + }, # https://github.com/Jimver/cuda-toolkit/blob/master/src/links/windows-links.ts } if test_only_latest_torch: - latest = "2.5.0" + latest = "2.5.1" matrix = {latest: matrix[latest]} if for_windows or for_macos: + if "2.5.1" in matrix: + matrix["2.5.1"]["python-version"].remove("3.13") + if "2.5.0" in matrix: matrix["2.5.0"]["python-version"].remove("3.13")