diff --git a/.github/workflows/macos-cpu-wheels.yml b/.github/workflows/macos-cpu-wheels.yml index a9b5cd5..bc28d68 100644 --- a/.github/workflows/macos-cpu-wheels.yml +++ b/.github/workflows/macos-cpu-wheels.yml @@ -4,7 +4,7 @@ on: push: branches: # - wheel - - torch-2.7.1 + - torch-2.8.0 tags: - '*' workflow_dispatch: diff --git a/.github/workflows/ubuntu-arm64-cpu-wheels.yml b/.github/workflows/ubuntu-arm64-cpu-wheels.yml index 8e738c4..4e94943 100644 --- a/.github/workflows/ubuntu-arm64-cpu-wheels.yml +++ b/.github/workflows/ubuntu-arm64-cpu-wheels.yml @@ -4,7 +4,7 @@ on: push: branches: # - wheel - - torch-2.7.1 + - torch-2.8.0 tags: - '*' workflow_dispatch: diff --git a/.github/workflows/ubuntu-cpu-wheels.yml b/.github/workflows/ubuntu-cpu-wheels.yml index 28d8f7c..51bb676 100644 --- a/.github/workflows/ubuntu-cpu-wheels.yml +++ b/.github/workflows/ubuntu-cpu-wheels.yml @@ -4,7 +4,7 @@ on: push: branches: # - wheel - - torch-2.7.1 + - torch-2.8.0 tags: - '*' workflow_dispatch: diff --git a/.github/workflows/windows-x64-cpu-wheels.yml b/.github/workflows/windows-x64-cpu-wheels.yml index 349c9c5..5b9c12e 100644 --- a/.github/workflows/windows-x64-cpu-wheels.yml +++ b/.github/workflows/windows-x64-cpu-wheels.yml @@ -4,7 +4,7 @@ on: push: branches: # - wheel - - torch-2.7.1 + - torch-2.8.0 tags: - '*' workflow_dispatch: diff --git a/scripts/github_actions/generate_build_matrix.py b/scripts/github_actions/generate_build_matrix.py index 9fa92f2..00b5da7 100755 --- a/scripts/github_actions/generate_build_matrix.py +++ b/scripts/github_actions/generate_build_matrix.py @@ -319,10 +319,18 @@ def generate_build_matrix( else ["11.8.0", "12.6.2", "12.8.1"] ), }, + "2.8.0": { + "python-version": ["3.9", "3.10", "3.11", "3.12", "3.13"], + "cuda": ( + ["12.6", "12.8", "12.9"] # default 12.8 + if not for_windows + else ["12.6.2", "12.8.1", "12.9.1"] + ), + }, # https://github.com/Jimver/cuda-toolkit/blob/master/src/links/windows-links.ts } if test_only_latest_torch: - latest = "2.7.1" + latest = "2.8.0" matrix = {latest: matrix[latest]} if for_windows or for_macos: diff --git a/scripts/github_actions/install_torch.sh b/scripts/github_actions/install_torch.sh index faa9fed..451f6ab 100755 --- a/scripts/github_actions/install_torch.sh +++ b/scripts/github_actions/install_torch.sh @@ -281,6 +281,23 @@ case ${torch} in ;; esac ;; + 2.8.*) + case ${cuda} in + 12.6) + package="torch==${torch}+cu126" + # https://download.pytorch.org/whl/nightly/torch/ + url=https://download.pytorch.org/whl/torch/ + ;; + 12.8) + package="torch==${torch}+cu128" + url=https://download.pytorch.org/whl/torch/ + ;; + 12.9) + package="torch==${torch}+cu129" + url=https://download.pytorch.org/whl/torch/ + ;; + esac + ;; *) echo "Unsupported PyTorch version: ${torch}" exit 1