interpn 0.8.0__tar.gz → 0.8.2__tar.gz

{interpn-0.8.0 → interpn-0.8.2}/.cargo/config.toml

@@ -1,2 +1,2 @@
 [target.'cfg(any(target_arch = "x86_64", target_arch = "x64"))']
-rustflags = ["-C", "target-feature=+avx2,+fma"]
+rustflags = ["-C", "target-cpu=x86-64-v3"]

interpn-0.8.2/.github/workflows/release-python.yml

@@ -0,0 +1,363 @@
+# This file is autogenerated by maturin v1.8.3
+# To update, run
+#
+#    maturin generate-ci github
+#
+name: release-python
+
+on:
+  workflow_dispatch:
+
+permissions:
+  contents: read
+
+jobs:
+  test_python:
+    uses: ./.github/workflows/test-python.yml
+
+  linux_pgo:
+    runs-on: ${{ matrix.platform.runner }}
+    strategy:
+      matrix:
+        platform:
+          - runner: ubuntu-22.04
+            target: x86_64
+          # Exclude 32-bit linux due to OOM issues during compilation & no demand
+          # - runner: ubuntu-22.04
+          #   target: x86
+          - runner: ubuntu-22.04-arm
+            target: aarch64
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          architecture: ${{ matrix.platform.target }}
+
+      - name: Install LLVM
+        run: bash ./scripts/install_llvm.sh
+
+      - name: Clear profiles
+        run: rm -rf ./scripts/pgo-profiles; mkdir ./scripts/pgo-profiles; mkdir ./scripts/pgo-profiles/pgo.profraw
+
+      - name: Instrumented build
+        uses: PyO3/maturin-action@v1
+        with:
+          target: ${{ matrix.platform.target }}
+          args: --release --out dist --find-interpreter --verbose -- "-Cprofile-generate=${PWD}/scripts/pgo-profiles/pgo.profraw"
+          manylinux: auto
+
+      - name: Install instrumented
+        run: |
+          # Need recent pip for dep groups
+          python -m pip install -U pip
+
+          # Try each wheel until we find the right one
+          for whl in $(find dist/ -type f -name '*.whl'); do
+            echo "Trying $whl"
+            if ! python -m pip install "$whl"[pydantic] --group test; then
+              echo "Failed to install $whl"
+            else
+              echo "Installed $whl"
+              break
+            fi
+          done
+
+      - name: Instrumented profile workload
+        run: |
+          # Run profile workload
+          LLVM_PROFILE_FILE=scripts/pgo-profiles/pgo.profraw/pgo-%p-%m.profraw python ./scripts/profile_workload.py
+
+          # Remove instrumented wheel
+          rm -rf ./dist/
+
+          # Merge profiles
+          PROFILES=$(find scripts/pgo-profiles/pgo.profraw -name '*.profraw')
+          echo "Profiles to merge: $PROFILES"
+          /usr/lib/llvm-21/bin/llvm-profdata merge -o scripts/pgo-profiles/pgo.profdata $PROFILES
+
+      - name: Build wheels optimized
+        uses: PyO3/maturin-action@v1
+        with:
+          target: ${{ matrix.platform.target }}
+          args: --release --out dist --find-interpreter --verbose -- "-Cprofile-use=${{github.workspace}}/scripts/pgo-profiles/pgo.profdata" "-Cllvm-args=-pgo-warn-missing-function"
+          manylinux: auto
+
+      - name: Upload wheels
+        uses: actions/upload-artifact@v4
+        with:
+          name: wheels-linux-${{ matrix.platform.target }}-${{ matrix.python-version }}
+          path: dist
+
+  linux_cross:
+    runs-on: ${{ matrix.platform.runner }}
+    strategy:
+      matrix:
+        platform:
+          - runner: ubuntu-22.04
+            target: armv7
+          - runner: ubuntu-22.04
+            target: s390x
+          - runner: ubuntu-22.04
+            target: ppc64le
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+      - name: Build wheels
+        uses: PyO3/maturin-action@v1
+        with:
+          target: ${{ matrix.platform.target }}
+          args: --release --out dist --find-interpreter
+          manylinux: auto
+
+      - name: Upload wheels
+        uses: actions/upload-artifact@v4
+        with:
+          name: wheels-linux-${{ matrix.platform.target }}-${{ matrix.python-version }}
+          path: dist
+
+  musllinux:
+    runs-on: ${{ matrix.platform.runner }}
+    strategy:
+      matrix:
+        platform:
+          - runner: ubuntu-22.04
+            target: x86_64
+          - runner: ubuntu-22.04
+            target: x86
+          - runner: ubuntu-22.04
+            target: aarch64
+          - runner: ubuntu-22.04
+            target: armv7
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+      - name: Build wheels
+        uses: PyO3/maturin-action@v1
+        with:
+          target: ${{ matrix.platform.target }}
+          args: --release --out dist --find-interpreter --verbose
+
+          manylinux: musllinux_1_2
+      - name: Upload wheels
+        uses: actions/upload-artifact@v4
+        with:
+          name: wheels-musllinux-${{ matrix.platform.target }}-${{ matrix.python-version }}
+          path: dist
+
+  windows_pgo:
+    runs-on: ${{ matrix.platform.runner }}
+    strategy:
+      matrix:
+        platform:
+          - runner: windows-latest
+            target: x64
+    steps:
+      - uses: actions/checkout@v4
+
+      # Install Rust and Python separate from maturin and uv actions
+      # to try to avoid ending up with one or the other end up with
+      # the 32-bit version
+      - name: Install Rust toolchain
+        uses: dtolnay/rust-toolchain@stable
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+
+      - name: Install LLVM
+        run: choco install llvm --version="21.1.0" -y; llvm-profdata --version
+
+      - name: Clear profiles
+        shell: bash
+        run: rm -rf ./scripts/pgo-profiles; mkdir ./scripts/pgo-profiles; mkdir ./scripts/pgo-profiles/pgo.profraw
+
+      - name: Instrumented build
+        uses: PyO3/maturin-action@v1
+        with:
+          target: ${{ matrix.platform.target }}
+          args: --release --out dist --verbose -- "-Cprofile-generate=${PWD}/scripts/pgo-profiles/pgo.profraw"
+
+      - name: Install instrumented
+        shell: bash
+        run: |
+          # Need recent pip for dep groups
+          python -m pip install -U pip
+
+          # Try each wheel until we find the right one
+          for whl in $(find dist/ -type f -name '*.whl'); do
+            echo "Trying $whl"
+            if ! python -m pip install "$whl"[pydantic] --group test; then
+              echo "Failed to install $whl"
+            else
+              echo "Installed $whl"
+              break
+            fi
+          done
+
+      - name: Instrumented profile workload
+        shell: bash
+        run: |
+          # Run profile workload
+          LLVM_PROFILE_FILE=scripts/pgo-profiles/pgo.profraw/pgo-%p-%m.profraw python ./scripts/profile_workload.py
+
+          # Remove instrumented wheel
+          rm -rf ./dist/
+
+          # Merge profiles
+          PROFILES=$(find scripts/pgo-profiles/pgo.profraw -name '*.profraw')
+          echo "Profiles to merge: $PROFILES"
+          llvm-profdata merge -o scripts/pgo-profiles/pgo.profdata $PROFILES
+
+      - name: Build wheels optimized
+        uses: PyO3/maturin-action@v1
+        with:
+          target: ${{ matrix.platform.target }}
+          args: --release --out dist --verbose -- "-Cprofile-use=${{github.workspace}}/scripts/pgo-profiles/pgo.profdata" "-Cllvm-args=-pgo-warn-missing-function"
+
+      - name: Upload wheels
+        uses: actions/upload-artifact@v4
+        with:
+          name: wheels-windows-${{ matrix.platform.target }}-${{ matrix.python-version }}
+          path: dist
+
+  windows_cross:
+    runs-on: ${{ matrix.platform.runner }}
+    strategy:
+      matrix:
+        platform:
+          - runner: windows-latest
+            target: x86
+    steps:
+      - uses: actions/checkout@v4
+
+      # Install Rust and Python separate from maturin and uv actions
+      # to try to avoid ending up with one or the other end up with
+      # the 32-bit version
+      - name: Install Rust toolchain
+        uses: dtolnay/rust-toolchain@stable
+      - uses: actions/setup-python@v5
+        with:
+          architecture: ${{ matrix.platform.target }}
+
+
+      - name: Build wheels
+        uses: PyO3/maturin-action@v1
+        with:
+          target: ${{ matrix.platform.target }}
+          args: --release --out dist --verbose
+
+      - name: Upload wheels
+        uses: actions/upload-artifact@v4
+        with:
+          name: wheels-windows-${{ matrix.platform.target }}-${{ matrix.python-version }}
+          path: dist
+
+  macos_pgo:
+    runs-on: ${{ matrix.platform.runner }}
+    strategy:
+      matrix:
+        platform:
+          - runner: macos-15-intel
+            target: x86_64
+            # macos-15 runner is arm64
+          - runner: macos-15
+            target: aarch64
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+
+      - name: Install LLVM
+        run: bash ./scripts/install_llvm.sh
+
+      - name: Clear profiles
+        run: rm -rf ./scripts/pgo-profiles; mkdir ./scripts/pgo-profiles; mkdir ./scripts/pgo-profiles/pgo.profraw
+
+      - name: Instrumented build
+        uses: PyO3/maturin-action@v1
+        with:
+          target: ${{ matrix.platform.target }}
+          args: --release --out dist --find-interpreter --verbose -- "-Cprofile-generate=${PWD}/scripts/pgo-profiles/pgo.profraw"
+
+      - name: Install instrumented
+        run: |
+          # Need recent pip for dep groups
+          pip install -U pip
+
+          # Try each wheel until we find the right one
+          for whl in $(find dist/ -type f -name '*.whl'); do
+            echo "Trying $whl"
+            if ! python -m pip install "$whl"[pydantic] --group test; then
+              echo "Failed to install $whl"
+            else
+              echo "Installed $whl"
+              break
+            fi
+          done
+
+      - name: Instrumented profile workload
+        run: |
+          # Run profile workload
+          LLVM_PROFILE_FILE=scripts/pgo-profiles/pgo.profraw/pgo-%p-%m.profraw python ./scripts/profile_workload.py
+
+          # Remove instrumented wheel
+          rm -rf ./dist/
+
+          # Merge profiles
+          PROFILES=$(find scripts/pgo-profiles/pgo.profraw -name '*.profraw')
+          echo "Profiles to merge: $PROFILES"
+          $(brew --prefix llvm@21)/bin/llvm-profdata merge -o scripts/pgo-profiles/pgo.profdata $PROFILES
+
+      - name: Build wheels optimized
+        uses: PyO3/maturin-action@v1
+        with:
+          target: ${{ matrix.platform.target }}
+          args: --release --out dist --find-interpreter --verbose -- "-Cprofile-use=${{github.workspace}}/scripts/pgo-profiles/pgo.profdata" "-Cllvm-args=-pgo-warn-missing-function"
+
+      - name: Upload wheels
+        uses: actions/upload-artifact@v4
+        with:
+          name: wheels-macos-${{ matrix.platform.target }}-${{ matrix.python-version }}
+          path: dist
+
+  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/') || github.event_name == 'workflow_dispatch' }}
+    needs: [test_python, linux_pgo, linux_cross, musllinux, windows_pgo, windows_cross, macos_pgo, sdist]
+    permissions:
+      # Use to sign the release artifacts
+      id-token: write
+      # Used to upload release artifacts
+      contents: write
+      # Used to generate artifact attestation
+      attestations: write
+    steps:
+      - uses: actions/download-artifact@v4
+      - name: Generate artifact attestation
+        uses: actions/attest-build-provenance@v2
+        with:
+          subject-path: 'wheels-*/*'
+      - name: Publish to PyPI
+        if: github.ref == 'refs/heads/main'
+        uses: PyO3/maturin-action@v1
+        env:
+          MATURIN_PYPI_TOKEN: ${{ secrets.PYPI_API_TOKEN }}
+        with:
+          command: upload
+          args: --non-interactive --skip-existing wheels-*/*

{interpn-0.8.0 → interpn-0.8.2}/.github/workflows/test-python.yml

@@ -54,7 +54,7 @@ jobs:
           # - runner: ubuntu-latest
           - runner: ubuntu-24.04-arm  # aarch64
           - runner: macos-15-intel  # x86_64
-          - runner: macos-14  # aarch64
+          - runner: macos-15  # aarch64
     steps:
     - uses: actions/checkout@v3
     - name: Install uv
@@ -88,7 +88,7 @@ jobs:
     - name: Profile-Guided Optimization (PGO)
       shell: bash
       run: |
-        sh ./scripts/distr_pgo_install.sh
+        sh ./scripts/distr_pgo.sh
 
     - name: Test (with PGO)
       run: |

{interpn-0.8.0 → interpn-0.8.2}/.gitignore

@@ -74,9 +74,10 @@ docs/_build/
 # Extra
 *.svg
 
-# Unmerged PGO profiles
+# PGO profiles
 *.profraw
 *native.profdata
+*pgo-profiles/
 
 # cargo-semver-checks
-semver-checks/
+semver-checks/

{interpn-0.8.0 → interpn-0.8.2}/CHANGELOG.md

@@ -1,5 +1,26 @@
 # Changelog
 
+## 0.8.2 2025-11-12
+
+### Changed
+
+* Use `x86-64-v3` reference CPU as compilation target instead of manually specifying features
+  * Includes AVX2 and vector FMA
+
+## 0.8.1 2025-11-10
+
+### Added
+
+* Python
+  * Implement PGO for mac intel/aarch64, linux aarch64, and windows x64
+  * Add script for installing llvm on linux and mac to support PGO
+
+### Changed
+
+* Python
+  * Remove stored pgo profiles
+  * Update PGO scripts
+
 ## 0.8.0 2025-11-08
 
 Support ndarray inputs for python interpn function, require kwargs for most inputs,

{interpn-0.8.0 → interpn-0.8.2}/Cargo.lock

@@ -221,7 +221,7 @@ checksum = "b248f5224d1d606005e02c97f5aa4e88eeb230488bcc03bc9ca4d7991399f2b5"
 
 [[package]]
 name = "interpn"
-version = "0.8.0"
+version = "0.8.2"
 dependencies = [
  "criterion",
  "crunchy",

{interpn-0.8.0 → interpn-0.8.2}/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "interpn"
-version = "0.8.0"
+version = "0.8.2"
 edition = "2024"
 authors = ["James Logan <jlogan03@gmail.com>"]
 license = "MIT OR Apache-2.0"

{interpn-0.8.0 → interpn-0.8.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: interpn
-Version: 0.8.0
+Version: 0.8.2
 Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy

{interpn-0.8.0 → interpn-0.8.2}/benches/bench_cpu.py

@@ -469,7 +469,7 @@ def _plot_speedup_vs_dims(
                 yanchor="top",
             ),
             height=450,
-            margin=dict(t=60, l=60, r=200, b=90),
+            margin=dict(t=60, l=60, r=20, b=90),
             # legend=dict(
             #     orientation="v",
             #     yanchor="top",