Skip to content

Commit

Permalink
Repack wheels with build-number 1 (#4354)
Browse files Browse the repository at this point in the history
Cherry-pick of 4383d28 and
5eb4ba5 from 2.0.X to 3.0.X branch
  • Loading branch information
malfet committed Jul 18, 2024
1 parent 91f24d8 commit b514024
Show file tree
Hide file tree
Showing 2 changed files with 14 additions and 2 deletions.
14 changes: 13 additions & 1 deletion .github/workflows/wheels.yml
Original file line number Diff line number Diff line change
Expand Up @@ -67,7 +67,19 @@ jobs:
export CIBW_BUILD="cp3*-manylinux_x86_64"
python3 -m cibuildwheel python --output-dir wheelhouse
- name: Append build number
shell: bash
run: |
set -ex
mkdir -p wheelhouse-1
find wheelhouse -type f
for i in $(find wheelhouse -type f); do
echo "Patching $i"
python3 -mwheel unpack "$i"
python3 -mwheel pack triton-3.0.0 --build-number 1 --dest-dir wheelhouse-1
rm -rf triton-3.0.0
done
- name: Upload wheels to PyPI
run: |
python3 -m twine upload wheelhouse/* -u __token__ -p ${{ secrets.PYPY_API_TOKEN }}
python3 -m twine upload wheelhouse-1/* -u __token__ -p ${{ secrets.PYPY_API_TOKEN }}
2 changes: 1 addition & 1 deletion python/setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -581,7 +581,7 @@ def get_install_requires():

setup(
name=os.environ.get("TRITON_WHEEL_NAME", "triton"),
version="3.0.0-post1" + os.environ.get("TRITON_WHEEL_VERSION_SUFFIX", ""),
version="3.0.0" + os.environ.get("TRITON_WHEEL_VERSION_SUFFIX", ""),
author="Philippe Tillet",
author_email="[email protected]",
description="A language and compiler for custom Deep Learning operations",
Expand Down

0 comments on commit b514024

Please sign in to comment.