diff --git a/.github/workflows/macos-cpu-wheels.yml b/.github/workflows/macos-cpu-wheels.yml index 1d2293f..5cd97a6 100644 --- a/.github/workflows/macos-cpu-wheels.yml +++ b/.github/workflows/macos-cpu-wheels.yml @@ -26,11 +26,11 @@ jobs: id: set-matrix run: | # outputting for debugging purposes - python ./scripts/github_actions/generate_build_matrix.py --for-macos - MATRIX=$(python ./scripts/github_actions/generate_build_matrix.py --for-macos) + # python ./scripts/github_actions/generate_build_matrix.py --for-macos + # MATRIX=$(python ./scripts/github_actions/generate_build_matrix.py --for-macos) - # python ./scripts/github_actions/generate_build_matrix.py --for-macos --test-only-latest-torch - # MATRIX=$(python ./scripts/github_actions/generate_build_matrix.py --for-macos --test-only-latest-torch) + python ./scripts/github_actions/generate_build_matrix.py --for-macos --test-only-latest-torch + MATRIX=$(python ./scripts/github_actions/generate_build_matrix.py --for-macos --test-only-latest-torch) echo "::set-output name=matrix::${MATRIX}" build_wheels_macos_cpu: diff --git a/.github/workflows/ubuntu-cpu-wheels.yml b/.github/workflows/ubuntu-cpu-wheels.yml index 6009974..b11c342 100644 --- a/.github/workflows/ubuntu-cpu-wheels.yml +++ b/.github/workflows/ubuntu-cpu-wheels.yml @@ -26,11 +26,11 @@ jobs: id: set-matrix run: | # outputting for debugging purposes - python ./scripts/github_actions/generate_build_matrix.py - MATRIX=$(python ./scripts/github_actions/generate_build_matrix.py) + # python ./scripts/github_actions/generate_build_matrix.py + # MATRIX=$(python ./scripts/github_actions/generate_build_matrix.py) - # python ./scripts/github_actions/generate_build_matrix.py --test-only-latest-torch - # MATRIX=$(python ./scripts/github_actions/generate_build_matrix.py --test-only-latest-torch) + python ./scripts/github_actions/generate_build_matrix.py --test-only-latest-torch + MATRIX=$(python ./scripts/github_actions/generate_build_matrix.py --test-only-latest-torch) echo "::set-output name=matrix::${MATRIX}" build-manylinux-wheels: diff --git a/.github/workflows/windows-x64-cpu-wheels.yml b/.github/workflows/windows-x64-cpu-wheels.yml index 88bd899..0103c32 100644 --- a/.github/workflows/windows-x64-cpu-wheels.yml +++ b/.github/workflows/windows-x64-cpu-wheels.yml @@ -26,11 +26,11 @@ jobs: id: set-matrix run: | # outputting for debugging purposes - python ./scripts/github_actions/generate_build_matrix.py --for-windows - MATRIX=$(python ./scripts/github_actions/generate_build_matrix.py --for-windows) + # python ./scripts/github_actions/generate_build_matrix.py --for-windows + # MATRIX=$(python ./scripts/github_actions/generate_build_matrix.py --for-windows) - # python ./scripts/github_actions/generate_build_matrix.py --for-windows --test-only-latest-torch - # MATRIX=$(python ./scripts/github_actions/generate_build_matrix.py --for-windows --test-only-latest-torch) + python ./scripts/github_actions/generate_build_matrix.py --for-windows --test-only-latest-torch + MATRIX=$(python ./scripts/github_actions/generate_build_matrix.py --for-windows --test-only-latest-torch) echo "::set-output name=matrix::${MATRIX}" build_wheels_win64_cpu: diff --git a/scripts/github_actions/generate_build_matrix.py b/scripts/github_actions/generate_build_matrix.py index bc6eb59..492e274 100755 --- a/scripts/github_actions/generate_build_matrix.py +++ b/scripts/github_actions/generate_build_matrix.py @@ -268,6 +268,15 @@ def generate_build_matrix(enable_cuda, for_windows, for_macos, test_only_latest_ 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: @@ -275,6 +284,9 @@ def generate_build_matrix(enable_cuda, for_windows, for_macos, test_only_latest_ 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")