Use rayon
global thread pool
#417
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
# This file is autogenerated by maturin v1.7.0 | |
# To update, run | |
# | |
# maturin generate-ci github -o .github/workflows/CI.yml --pytest --zig | |
# | |
name: CI | |
on: | |
push: | |
branches: | |
- main | |
- '*' | |
tags: | |
- '*' | |
pull_request: | |
workflow_dispatch: | |
permissions: | |
contents: read | |
jobs: | |
linux: | |
runs-on: ${{ matrix.platform.runner }} | |
strategy: | |
fail-fast: false | |
matrix: | |
platform: | |
- runner: ubuntu-latest | |
target: x86_64 | |
- runner: ubuntu-latest | |
target: aarch64 | |
steps: | |
- uses: actions/checkout@v4 | |
- uses: actions/setup-python@v5 | |
with: | |
python-version: 3.x | |
- name: Build wheels | |
uses: PyO3/maturin-action@v1 | |
with: | |
target: ${{ matrix.platform.target }} | |
args: --release --out dist --zig | |
sccache: 'true' | |
manylinux: auto | |
env: | |
CFLAGS_aarch64_unknown_linux_gnu: "-D__ARM_ARCH=8" | |
- name: Upload wheels | |
uses: actions/upload-artifact@v4 | |
with: | |
name: wheels-linux-${{ matrix.platform.target }} | |
path: dist | |
- name: pytest | |
if: ${{ startsWith(matrix.platform.target, 'x86_64') }} | |
shell: bash | |
run: | | |
set -e | |
pip install uv | |
for version in 3.8 3.9 3.10 3.11 3.12; do | |
uv venv --preview --python $version | |
source .venv/bin/activate | |
uv pip install primp --no-index --find-links dist --force-reinstall | |
uv pip install pytest | |
pytest | |
done | |
- name: pytest | |
if: ${{ !startsWith(matrix.platform.target, 'x86') && matrix.platform.target != 'ppc64' }} | |
uses: uraimo/run-on-arch-action@v2 | |
with: | |
arch: ${{ matrix.platform.target }} | |
distro: ubuntu22.04 | |
githubToken: ${{ github.token }} | |
install: | | |
apt-get update | |
apt-get install -y --no-install-recommends python3 python3-pip | |
run: | | |
set -e | |
pip install uv | |
for version in 3.8 3.9 3.10 3.11 3.12; do | |
uv venv --preview --python $version | |
source .venv/bin/activate | |
uv pip install primp --no-index --find-links dist --force-reinstall | |
uv pip install pytest | |
pytest | |
done | |
musllinux: | |
runs-on: ${{ matrix.platform.runner }} | |
strategy: | |
fail-fast: false | |
matrix: | |
platform: | |
- runner: ubuntu-latest | |
target: x86_64 | |
- runner: ubuntu-latest | |
target: aarch64 | |
steps: | |
- uses: actions/checkout@v4 | |
- uses: actions/setup-python@v5 | |
with: | |
python-version: 3.x | |
- name: Build wheels | |
uses: PyO3/maturin-action@v1 | |
with: | |
target: ${{ matrix.platform.target }} | |
args: --release --out dist --zig | |
sccache: 'true' | |
manylinux: musllinux_1_2 | |
- name: Upload wheels | |
uses: actions/upload-artifact@v4 | |
with: | |
name: wheels-musllinux-${{ matrix.platform.target }} | |
path: dist | |
- name: QEMU | |
if: matrix.platform.target != 'x86_64' | |
uses: docker/setup-qemu-action@v3 | |
- name: pytest | |
uses: addnab/docker-run-action@v3 | |
with: | |
image: quay.io/pypa/musllinux_1_2_${{ matrix.platform.target }}:latest | |
options: -v ${{ github.workspace }}:/io -w /io | |
run: | | |
for version in 3.8 3.9 3.10 3.11 3.12; do | |
python$version -m venv .venv | |
source .venv/bin/activate | |
pip install primp --no-index --find-links dist --force-reinstall | |
pip install pytest | |
pytest | |
done | |
windows: | |
runs-on: ${{ matrix.platform.runner }} | |
strategy: | |
fail-fast: false | |
matrix: | |
platform: | |
- runner: windows-latest | |
target: x64 | |
steps: | |
- uses: actions/checkout@v4 | |
- uses: actions/setup-python@v5 | |
with: | |
python-version: 3.x | |
architecture: ${{ matrix.platform.target }} | |
- name: Install nasm | |
run: choco install nasm | |
- name: Build wheels | |
uses: PyO3/maturin-action@v1 | |
with: | |
target: ${{ matrix.platform.target }} | |
args: --release --out dist | |
sccache: 'true' | |
- name: Upload wheels | |
uses: actions/upload-artifact@v4 | |
with: | |
name: wheels-windows-${{ matrix.platform.target }} | |
path: dist | |
- name: pytest | |
if: ${{ !startsWith(matrix.platform.target, 'aarch64') }} | |
shell: bash | |
run: | | |
set -e | |
pip install uv | |
for version in 3.8 3.9 3.10 3.11 3.12; do | |
uv venv --preview --python $version | |
source .venv/Scripts/activate | |
uv pip install primp --no-index --find-links dist --force-reinstall | |
uv pip install pytest | |
pytest | |
done | |
macos: | |
runs-on: ${{ matrix.platform.runner }} | |
strategy: | |
fail-fast: false | |
matrix: | |
platform: | |
- runner: macos-12 | |
target: x86_64 | |
- runner: macos-14 | |
target: aarch64 | |
steps: | |
- uses: actions/checkout@v4 | |
- uses: actions/setup-python@v5 | |
with: | |
python-version: 3.x | |
- name: Build wheels | |
uses: PyO3/maturin-action@v1 | |
with: | |
target: ${{ matrix.platform.target }} | |
args: --release --out dist | |
sccache: 'true' | |
- name: Upload wheels | |
uses: actions/upload-artifact@v4 | |
with: | |
name: wheels-macos-${{ matrix.platform.target }} | |
path: dist | |
- name: pytest | |
run: | | |
set -e | |
pip install uv | |
for version in 3.8 3.9 3.10 3.11 3.12; do | |
uv venv --preview --python $version | |
source .venv/bin/activate | |
uv pip install primp --no-index --find-links dist --force-reinstall | |
uv pip install pytest | |
pytest | |
done | |
sdist: | |
runs-on: ubuntu-latest | |
steps: | |
- uses: actions/checkout@v4 | |
- name: Build sdist | |
uses: PyO3/maturin-action@v1 | |
with: | |
command: sdist | |
args: --out dist | |
- name: Upload sdist | |
uses: actions/upload-artifact@v4 | |
with: | |
name: wheels-sdist | |
path: dist | |
release: | |
name: Release | |
runs-on: ubuntu-latest | |
if: "startsWith(github.ref, 'refs/tags/')" | |
needs: [linux, musllinux, windows, macos, sdist] | |
steps: | |
- uses: actions/download-artifact@v4 | |
- name: Publish to PyPI | |
uses: PyO3/maturin-action@v1 | |
env: | |
MATURIN_PYPI_TOKEN: ${{ secrets.PYPI_API_TOKEN }} | |
with: | |
command: upload | |
args: --non-interactive --skip-existing wheels-*/* | |
benchmark: | |
permissions: | |
contents: write | |
runs-on: ubuntu-latest | |
needs: [linux] | |
steps: | |
- uses: actions/checkout@v4 | |
- name: Set up Python | |
uses: actions/setup-python@v5 | |
with: | |
python-version: 3.x | |
- name: Download wheels | |
uses: actions/download-artifact@v4 | |
with: | |
name: wheels-linux-x86_64 | |
- name: Install dependencies | |
run: | | |
pip install -r benchmark/requirements.txt | |
pip install primp --no-index --find-links ./ --force-reinstall | |
- name: Start Uvicorn server | |
run: | | |
uvicorn benchmark.server:app --host 0.0.0.0 --port 8000 & | |
sleep 10 | |
- name: Run benchmark | |
run: python benchmark/benchmark.py | |
- name: Generate image, commit to the temp branch, merge changes into main, delete temp branch | |
if: "startsWith(github.ref, 'refs/tags/')" | |
run: | | |
python benchmark/generate_image.py | |
git config --global user.name 'GitHub Actions' | |
git config --global user.email '[email protected]' | |
git add \*.jpg | |
git diff --quiet && git diff --staged --quiet || git commit -m "Update generated image" | |
git checkout -b update-generated-image | |
git push https://${{ secrets.PUSH_TOKEN }}@github.com/deedy5/primp.git update-generated-image || echo "No changes to push" | |
git fetch origin | |
git checkout main | |
git merge update-generated-image | |
git push https://${{ secrets.PUSH_TOKEN }}@github.com/deedy5/primp.git main | |
git push origin --delete update-generated-image |