mirror of
https://github.com/ROCm/ROCm.git
synced 2026-04-05 03:01:17 -04:00
319af1fb65
Ideally you would also build source distributions so that it is in principle possible to build `triton` on other platforms, but building `musllinux` wheels would at least help with openai/whisper#1328. I suspect you will also get people showing up at some point asking for `aarch64` wheels as well. It might be worth taking a look at the [`cibuildwheel` output matrix](https://cibuildwheel.readthedocs.io/en/stable/#what-does-it-do) to see what you are comfortable with shipping (particularly if you aren't shipping source distributions).
40 lines
1.2 KiB
YAML
40 lines
1.2 KiB
YAML
name: Wheels
|
|
on:
|
|
workflow_dispatch:
|
|
#schedule:
|
|
# - cron: "0 0 * * *"
|
|
|
|
jobs:
|
|
|
|
Build-Wheels:
|
|
|
|
runs-on: [self-hosted, V100]
|
|
|
|
steps:
|
|
|
|
- name: Checkout
|
|
uses: actions/checkout@v2
|
|
|
|
- name: Patch setup.py
|
|
run: |
|
|
#sed -i 's/name\=\"triton\"/name="triton-nightly"/g' python/setup.py
|
|
export LATEST_DATE=$(TZ=UTC0 git show --quiet --date='format-local:%Y%m%d' --format="%cd")
|
|
sed -i -r "s/version\=\"(.*)\"/version=\"\1-dev"$LATEST_DATE"\"/g" python/setup.py
|
|
echo "" >> python/setup.cfg
|
|
echo "[build_ext]" >> python/setup.cfg
|
|
echo "base-dir=/project" >> python/setup.cfg
|
|
|
|
- name: Build wheels
|
|
run: |
|
|
export CIBW_MANYLINUX_X86_64_IMAGE="quay.io/pypa/manylinux2014_x86_64:latest"
|
|
#export CIBW_MANYLINUX_PYPY_X86_64_IMAGE="quay.io/pypa/manylinux2014_x86_64:latest"
|
|
export CIBW_BEFORE_BUILD="pip install cmake;"
|
|
export CIBW_SKIP="{cp,pp}35-*"
|
|
export CIBW_BUILD="{cp,pp}3*-manylinux_x86_64 cp3*-musllinux_x86_64"
|
|
python3 -m cibuildwheel python --output-dir wheelhouse
|
|
|
|
|
|
- name: Upload wheels to PyPI
|
|
run: |
|
|
python3 -m twine upload wheelhouse/* -u __token__ -p ${{ secrets.PYPY_API_TOKEN }}
|