diff --git a/.github/workflows/build-wheels.yml b/.github/workflows/build-wheels.yml index b588bd7..c861d49 100644 --- 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