[ENH] Add support for cuda and for torch.compile #777
Workflow file for this run
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
name: Wheels | |
on: | |
push: | |
branches: | |
- master | |
- torch | |
tags: | |
- v* | |
pull_request: | |
branches: | |
- master | |
- torch | |
jobs: | |
build_wheels: | |
name: Build wheels on ${{ matrix.os }} | |
runs-on: ${{ matrix.os }} | |
strategy: | |
matrix: | |
os: [ubuntu-22.04, windows-2019, macos-12] | |
steps: | |
- uses: actions/checkout@v2 | |
- name: Build Py3.11 wheels | |
uses: pypa/cibuildwheel@v2.16.5 | |
env: | |
CIBW_BEFORE_BUILD_LINUX: "yum install -y python-devel zlib-devel libjpeg-turbo-devel wget && python -m pip install -U pip pillow numpy && python -m pip install -r requirements.txt && python -m pip install -e ." | |
CIBW_BEFORE_BUILD: "python -m pip install -U pip numpy pillow && python -m pip install -r requirements.txt && python -m pip install -e ." | |
CIBW_BUILD: "cp311-*" | |
CIBW_SKIP: "*-manylinux_i686 cp311-win32 *musllinux*" | |
- name: Build Py3.10 wheels | |
uses: pypa/cibuildwheel@v2.16.5 | |
env: | |
CIBW_BEFORE_BUILD_LINUX: "yum install -y python-devel zlib-devel libjpeg-turbo-devel wget && python -m pip install -U pip pillow numpy && python -m pip install -r requirements.txt && python -m pip install -e ." | |
CIBW_BEFORE_BUILD: "python -m pip install -U pip numpy pillow && python -m pip install -r requirements.txt && python -m pip install -e ." | |
CIBW_BUILD: "cp310-*" | |
CIBW_SKIP: "*-manylinux_i686 cp310-win32 *musllinux*" | |
- name: Build Py3.8-3.9 wheels | |
uses: pypa/cibuildwheel@v2.16.5 | |
env: | |
CIBW_BEFORE_BUILD: "python -m pip install -U pip numpy && python -m pip install Pillow==8.3.2 && python -m pip install -r requirements.txt && python -m pip install -e ." | |
CIBW_BUILD: "cp39-* cp38-*" | |
CIBW_SKIP: "*-manylinux_i686 cp38-win32 cp39-win32 *musllinux*" | |
- uses: actions/upload-artifact@v2 | |
with: | |
path: ./wheelhouse/*.whl |