support torch 2.7.1 (#119)

This commit is contained in:
Fangjun Kuang 2025-06-05 11:30:38 +08:00 committed by GitHub
parent 6edd6a8370
commit 8cda08e0c5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
6 changed files with 18 additions and 10 deletions

View File

@ -3,8 +3,8 @@ name: build-wheels-cpu-macos
on:
push:
branches:
- wheel
# - torch-2.7.0
# - wheel
- torch-2.7.1
tags:
- '*'
workflow_dispatch:

View File

@ -3,8 +3,8 @@ name: build-wheels-cpu-arm64-ubuntu
on:
push:
branches:
- wheel
# - torch-2.7.0
# - wheel
- torch-2.7.1
tags:
- '*'
workflow_dispatch:

View File

@ -3,8 +3,8 @@ name: build-wheels-cpu-ubuntu
on:
push:
branches:
- wheel
# - torch-2.7.0
# - wheel
- torch-2.7.1
tags:
- '*'
workflow_dispatch:

View File

@ -4,7 +4,7 @@ on:
push:
branches:
- wheel
- torch-2.7.0
# - torch-2.7.1
tags:
- '*'
workflow_dispatch:

View File

@ -3,8 +3,8 @@ name: build-wheels-cpu-win64
on:
push:
branches:
- wheel
# - torch-2.7.0
# - wheel
- torch-2.7.1
tags:
- '*'
workflow_dispatch:

View File

@ -311,10 +311,18 @@ def generate_build_matrix(
else ["11.8.0", "12.6.2", "12.8.1"]
),
},
"2.7.1": {
"python-version": ["3.9", "3.10", "3.11", "3.12", "3.13"],
"cuda": (
["11.8", "12.6", "12.8"] # default 12.6
if not for_windows
else ["11.8.0", "12.6.2", "12.8.1"]
),
},
# https://github.com/Jimver/cuda-toolkit/blob/master/src/links/windows-links.ts
}
if test_only_latest_torch:
latest = "2.7.0"
latest = "2.7.1"
matrix = {latest: matrix[latest]}
if for_windows or for_macos: