From 2989b0b1186fa6022932804f5b39fbb2781ebf42 Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Fri, 24 Nov 2023 11:34:10 +0800 Subject: [PATCH] Build wheels for torch 2.1.1 (#1261) --- .github/workflows/run-tests-cpu.yml | 2 +- .github/workflows/run-tests.yml | 2 +- scripts/github_actions/generate_build_matrix.py | 8 +++++++- 3 files changed, 9 insertions(+), 3 deletions(-) diff --git a/.github/workflows/run-tests-cpu.yml b/.github/workflows/run-tests-cpu.yml index 3cf8fa30f..52af6fe78 100644 --- a/.github/workflows/run-tests-cpu.yml +++ b/.github/workflows/run-tests-cpu.yml @@ -53,7 +53,7 @@ jobs: fail-fast: false matrix: os: [ubuntu-latest, macos-latest] - torch: ["2.1.0"] + torch: ["2.1.1"] python-version: ["3.8", "3.9", "3.10", "3.11"] build_type: ["Release", "Debug"] diff --git a/.github/workflows/run-tests.yml b/.github/workflows/run-tests.yml index f1f4baf79..a768b9b39 100644 --- a/.github/workflows/run-tests.yml +++ b/.github/workflows/run-tests.yml @@ -54,7 +54,7 @@ jobs: matrix: os: [ubuntu-latest] cuda: ["12.1"] - torch: ["2.1.0"] + torch: ["2.1.1"] python-version: ["3.11"] steps: diff --git a/scripts/github_actions/generate_build_matrix.py b/scripts/github_actions/generate_build_matrix.py index e9dfabdab..a3c8f99bb 100755 --- a/scripts/github_actions/generate_build_matrix.py +++ b/scripts/github_actions/generate_build_matrix.py @@ -166,10 +166,16 @@ def generate_build_matrix( 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: