From f3832524d38f9b7b7b15b989a27eae3cb8c205b2 Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Mon, 24 Jul 2023 09:27:39 +0800 Subject: [PATCH] fix CI tests --- scripts/github_actions/generate_build_matrix.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/scripts/github_actions/generate_build_matrix.py b/scripts/github_actions/generate_build_matrix.py index e135c57..93a3b86 100755 --- a/scripts/github_actions/generate_build_matrix.py +++ b/scripts/github_actions/generate_build_matrix.py @@ -131,15 +131,21 @@ def generate_build_matrix(enable_cuda, for_windows, for_macos, test_only_latest_ }, "1.13.1": { "python-version": ["3.7", "3.8", "3.9", "3.10", "3.11"], - "cuda": ["11.6", "11.7"], # default 11.7 + "cuda": ["11.6", "11.7"] # default 11.7 + if not for_windows + else ["11.6.2", "11.7.1"], }, "2.0.0": { "python-version": ["3.8", "3.9", "3.10", "3.11"], - "cuda": ["11.7", "11.8"], # default 11.7 + "cuda": ["11.7", "11.8"] # default 11.7 + if not for_windows + else ["11.7.1", "11.8.0"], }, "2.0.1": { "python-version": ["3.8", "3.9", "3.10", "3.11"], - "cuda": ["11.7", "11.8"], # default 11.7 + "cuda": ["11.7", "11.8"] # default 11.7 + if not for_windows + else ["11.7.1", "11.8.0"], }, } if test_only_latest_torch: