workflow #4
This file contains hidden or 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: Build wheels (dry run) | ||
| on: | ||
| push: | ||
| branches: [ main ] # adjust if you use a different default branch | ||
| workflow_dispatch: # manual trigger | ||
| jobs: | ||
| build: | ||
| strategy: | ||
| fail-fast: false | ||
| matrix: | ||
| os: [ubuntu-22.04, macos-14] # macos-14 is ARM64-only | ||
| python: ["3.9", "3.10", "3.11", "3.12", "3.13"] | ||
| runs-on: ${{ matrix.os }} | ||
| env: | ||
| CARGO_TERM_COLOR: always # nicer Cargo output | ||
| steps: | ||
| # 1. Check out sources (and submodules if libxsmm is vendored) | ||
| - uses: actions/checkout@v4 | ||
| with: { submodules: recursive } | ||
| # 2. OS-specific prerequisites | ||
| - name: Install Linux build deps | ||
| if: runner.os == 'Linux' | ||
| run: | | ||
| sudo apt-get update | ||
| sudo apt-get install -y build-essential pkg-config patchelf | ||
| # 3. Build static libxsmm on Linux | ||
| - name: Build libxsmm (static) | ||
| if: runner.os == 'Linux' | ||
| run: | | ||
| git clone --depth 1 https://github.com/libxsmm/libxsmm.git /tmp/libxsmm | ||
| make -C /tmp/libxsmm STATIC=1 | ||
| echo "LIBXSMM_DIR=/tmp/libxsmm" >> $GITHUB_ENV | ||
| # 4. Toolchains | ||
| - uses: dtolnay/rust-toolchain@stable | ||
| - uses: actions/setup-python@v5 | ||
| with: { python-version: ${{ matrix.python }} } | ||
| # 5. Install Python dependencies | ||
| - name: Install Python build deps | ||
| run: | | ||
| pip install maturin numpy | ||
| # Install auditwheel on Linux only | ||
| if [ "$RUNNER_OS" = "Linux" ]; then | ||
| pip install auditwheel | ||
| fi | ||
| # 6. Build the wheel | ||
| - name: Build wheel with maturin | ||
| env: | ||
| RUSTFLAGS: ${{ runner.os == 'Linux' && '-L native=/tmp/libxsmm/lib' || '' }} | ||
| run: | | ||
| if [ "$RUNNER_OS" = "Linux" ]; then | ||
| maturin build --release \ | ||
| --compatibility manylinux2014 \ | ||
| --locked \ | ||
| --features use-libxsmm \ | ||
| -i python | ||
| else | ||
| maturin build --release \ | ||
| --compatibility pypi \ | ||
| --locked \ | ||
| -i python | ||
| fi | ||
| # 7. auditwheel repair (Linux only) | ||
| - name: auditwheel repair | ||
| if: runner.os == 'Linux' | ||
| run: | | ||
| mkdir -p dist | ||
| for whl in target/wheels/*.whl; do | ||
| auditwheel repair "$whl" --wheel-dir dist/ | ||
| done | ||
| # 8. Quick import-smoke-test | ||
| - name: Smoke test wheel | ||
| run: | | ||
| # Install the wheel (prefer repaired wheel on Linux) | ||
| if [ "$RUNNER_OS" = "Linux" ] && [ -d dist ]; then | ||
| pip install --no-deps --force-reinstall dist/*.whl | ||
| else | ||
| pip install --no-deps --force-reinstall target/wheels/*.whl | ||
| fi | ||
| # Test import and basic functionality | ||
| python -c " | ||
| import maxsim_cpu | ||
| import numpy as np | ||
| print('MaxSim CPU imported successfully') | ||
| # Basic functionality test | ||
| q = np.random.randn(4, 128).astype(np.float32) | ||
| q = q / np.linalg.norm(q, axis=1, keepdims=True) | ||
| d = np.random.randn(2, 8, 128).astype(np.float32) | ||
| d = d / np.linalg.norm(d, axis=2, keepdims=True) | ||
| scores = maxsim_cpu.maxsim_scores(q, d) | ||
| print(f'Test passed! Scores shape: {scores.shape}') | ||
| assert scores.shape == (2,), f'Expected shape (2,), got {scores.shape}' | ||
| " | ||
| # 9. Upload artifacts | ||
| - uses: actions/upload-artifact@v4 | ||
| with: | ||
| name: wheels-${{ matrix.os }}-${{ matrix.python }} | ||
| path: | | ||
| target/wheels/*.whl | ||
| dist/*.whl | ||
| if-no-files-found: error | ||
