name: python
on:
push:
tags: ["v*"]
pull_request:
paths:
- "bindings/python/**"
- "src/**"
- ".github/workflows/python.yml"
workflow_dispatch:
permissions:
contents: read
jobs:
wheels:
name: wheels (${{ matrix.platform.target }})
runs-on: ${{ matrix.platform.runner }}
strategy:
fail-fast: false
matrix:
platform:
- { runner: macos-14, target: aarch64-apple-darwin, args: "--features mac" }
- { runner: ubuntu-latest, target: x86_64, args: "--features cuda" }
steps:
- uses: actions/checkout@v4
- uses: PyO3/maturin-action@v1
with:
working-directory: bindings/python
rust-toolchain: stable
target: ${{ matrix.platform.target }}
manylinux: auto
args: --release --out dist ${{ matrix.platform.args }}
- uses: actions/upload-artifact@v4
with:
name: wheels-${{ matrix.platform.target }}
path: bindings/python/dist
sdist:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v4
- uses: PyO3/maturin-action@v1
with:
working-directory: bindings/python
rust-toolchain: stable
command: sdist
args: --out dist
- uses: actions/upload-artifact@v4
with:
name: sdist
path: bindings/python/dist
versions:
name: versions in sync
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v4
- run: |
rust=$(grep -m1 '^version = ' Cargo.toml | cut -d'"' -f2)
py=$(grep -m1 '^version = ' bindings/python/Cargo.toml | cut -d'"' -f2)
echo "rust=$rust pypi=$py"
test "$rust" = "$py"
smoke:
name: smoke (capture from Python, read back in Rust)
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v4
- uses: dtolnay/rust-toolchain@stable - uses: actions/setup-python@v5
with:
python-version: "3.12"
- uses: PyO3/maturin-action@v1
with:
working-directory: bindings/python
rust-toolchain: stable
command: build
manylinux: auto
args: --release --out dist --features cuda
- name: install the wheel
run: pip install bindings/python/dist/*.whl
- name: capture a run from Python
env:
VOR_RECORD: /tmp/smoke.vor
run: |
python - <<'PY'
import vor
vor.record_metric_unit("loss", "")
vor.enable() # no NVIDIA driver here: NVML degrades, no crash
@vor.profile
def step(i):
return i * 2
for i in range(64):
step(i)
vor.record_metric("loss", 1.0 / (i + 1))
vor.frame_mark()
vor.flush()
print("captured 64 frames")
PY
- name: read the capture back with the Rust reader
run: |
cargo run --quiet --example headless -- /tmp/smoke.vor | tee out.txt
grep -Eq "frames: [1-9][0-9]*" out.txt
publish:
name: publish to PyPI
if: startsWith(github.ref, 'refs/tags/v')
needs: [versions, smoke, wheels, sdist]
runs-on: ubuntu-latest
environment: pypi
permissions:
id-token: write steps:
- uses: actions/download-artifact@v4
with:
path: dist
merge-multiple: true
- uses: pypa/gh-action-pypi-publish@release/v1
with:
packages-dir: dist