Build wheels for torch==2.6

Luthaf · Feb 11, 2025 · c35a9a8 · c35a9a8
1 parent 54bac1a
commit c35a9a8
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/.github/workflows/build-wheels.yml b/.github/workflows/build-wheels.yml
@@ -83,7 +83,7 @@ jobs:
     name: ${{ matrix.name }} (torch v${{ matrix.torch-version }})
     strategy:
       matrix:
-        torch-version: ['2.3', '2.4', '2.5']
+        torch-version: ['2.3', '2.4', '2.5', '2.6']
         arch: ['arm64', 'x86_64']
         os: ['ubuntu-22.04', 'macos-14']
         exclude:
@@ -107,6 +107,7 @@ jobs:
           - {torch-version: '2.3',  python-version: '3.12', cibw-python: 'cp312-*'}
           - {torch-version: '2.4',  python-version: '3.12', cibw-python: 'cp312-*'}
           - {torch-version: '2.5',  python-version: '3.12', cibw-python: 'cp312-*'}
+          - {torch-version: '2.6',  python-version: '3.12', cibw-python: 'cp312-*'}
     steps:
       - uses: actions/checkout@v4
         with:
@@ -132,6 +133,7 @@ jobs:
           CIBW_ARCHS: ${{ matrix.cibw-arch }}
           CIBW_BUILD_VERBOSITY: 1
           CIBW_MANYLINUX_X86_64_IMAGE: quay.io/pypa/manylinux2014_x86_64
+          CIBW_MANYLINUX_AARCH64_IMAGE: quay.io/pypa/manylinux_2_28_aarch64
           CIBW_ENVIRONMENT: >
             VESIN_TORCH_BUILD_WITH_TORCH_VERSION=${{ matrix.torch-version }}.*
             PIP_EXTRA_INDEX_URL=https://download.pytorch.org/whl/cpu