interpn 0.8.2__tar.gz → 0.9.0__tar.gz

{interpn-0.8.2 → interpn-0.9.0}/.readthedocs.yml

@@ -14,6 +14,7 @@ build:
     pre_build:
       # Rust 1.90 required but not available yet
       - curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y --profile minimal --default-toolchain 1.90.0
+      - pip install -e .[pydantic] --group doc
 
 mkdocs:
   configuration: mkdocs.yml

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

@@ -1,5 +1,20 @@
 # Changelog
 
+## 0.9.0 2025-12-27
+
+### Added
+
+* Rust
+  * Add `deep-unroll` feature that sets crunchy unroll depth to 256 and enables 4D unrolled cubic interpolation
+    * This improves compile times in typical use-cases
+
+### Changed
+
+* Rust
+  * Gate 4D unrolled cubic interpolation behind `deep-unroll` feature
+  * Enable `deep-unroll` feature for Python builds
+  * Update pyo3 deps
+
 ## 0.8.2 2025-11-12
 
 ### Changed

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

@@ -221,12 +221,12 @@ checksum = "b248f5224d1d606005e02c97f5aa4e88eeb230488bcc03bc9ca4d7991399f2b5"
 
 [[package]]
 name = "interpn"
-version = "0.8.2"
+version = "0.9.0"
 dependencies = [
  "criterion",
  "crunchy",
  "itertools 0.14.0",
- "ndarray 0.17.1",
+ "ndarray",
  "num-traits",
  "numpy",
  "pyo3",
@@ -310,21 +310,6 @@ dependencies = [
  "autocfg",
 ]
 
-[[package]]
-name = "ndarray"
-version = "0.16.1"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "882ed72dce9365842bf196bdeedf5055305f11fc8c03dee7bb0194a6cad34841"
-dependencies = [
- "matrixmultiply",
- "num-complex",
- "num-integer",
- "num-traits",
- "portable-atomic",
- "portable-atomic-util",
- "rawpointer",
-]
-
 [[package]]
 name = "ndarray"
 version = "0.17.1"
@@ -370,12 +355,12 @@ dependencies = [
 
 [[package]]
 name = "numpy"
-version = "0.27.0"
+version = "0.27.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "0fa24ffc88cf9d43f7269d6b6a0d0a00010924a8cc90604a21ef9c433b66998d"
+checksum = "7aac2e6a6e4468ffa092ad43c39b81c79196c2bb773b8db4085f695efe3bba17"
 dependencies = [
  "libc",
- "ndarray 0.16.1",
+ "ndarray",
  "num-complex",
  "num-integer",
  "num-traits",
@@ -459,9 +444,9 @@ dependencies = [
 
 [[package]]
 name = "pyo3"
-version = "0.27.1"
+version = "0.27.2"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "37a6df7eab65fc7bee654a421404947e10a0f7085b6951bf2ea395f4659fb0cf"
+checksum = "ab53c047fcd1a1d2a8820fe84f05d6be69e9526be40cb03b73f86b6b03e6d87d"
 dependencies = [
  "indoc",
  "libc",
@@ -476,9 +461,9 @@ dependencies = [
 
 [[package]]
 name = "pyo3-build-config"
-version = "0.27.1"
+version = "0.27.2"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "f77d387774f6f6eec64a004eac0ed525aab7fa1966d94b42f743797b3e395afb"
+checksum = "b455933107de8642b4487ed26d912c2d899dec6114884214a0b3bb3be9261ea6"
 dependencies = [
  "python3-dll-a",
  "target-lexicon",
@@ -486,9 +471,9 @@ dependencies = [
 
 [[package]]
 name = "pyo3-ffi"
-version = "0.27.1"
+version = "0.27.2"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "2dd13844a4242793e02df3e2ec093f540d948299a6a77ea9ce7afd8623f542be"
+checksum = "1c85c9cbfaddf651b1221594209aed57e9e5cff63c4d11d1feead529b872a089"
 dependencies = [
  "libc",
  "pyo3-build-config",
@@ -496,9 +481,9 @@ dependencies = [
 
 [[package]]
 name = "pyo3-macros"
-version = "0.27.1"
+version = "0.27.2"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "eaf8f9f1108270b90d3676b8679586385430e5c0bb78bb5f043f95499c821a71"
+checksum = "0a5b10c9bf9888125d917fb4d2ca2d25c8df94c7ab5a52e13313a07e050a3b02"
 dependencies = [
  "proc-macro2",
  "pyo3-macros-backend",
@@ -508,9 +493,9 @@ dependencies = [
 
 [[package]]
 name = "pyo3-macros-backend"
-version = "0.27.1"
+version = "0.27.2"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "70a3b2274450ba5288bc9b8c1b69ff569d1d61189d4bff38f8d22e03d17f932b"
+checksum = "03b51720d314836e53327f5871d4c0cfb4fb37cc2c4a11cc71907a86342c40f9"
 dependencies = [
  "heck",
  "proc-macro2",

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

@@ -1,6 +1,6 @@
 [package]
 name = "interpn"
-version = "0.8.2"
+version = "0.9.0"
 edition = "2024"
 authors = ["James Logan <jlogan03@gmail.com>"]
 license = "MIT OR Apache-2.0"
@@ -17,11 +17,11 @@ crate-type = ["cdylib", "rlib"]
 [dependencies]
 # Rust lib deps
 num-traits = { version = "0.2.19", default-features = false, features = ["libm"] }
-crunchy = { version = "0.2.4", default-features = false, features = ["limit_256"] }
+crunchy = { version = "0.2.4", default-features = false }
 
 # Python bindings
-pyo3 = { version = "0.27.1", features = ["extension-module", "abi3-py310", "generate-import-lib"], optional = true }
-numpy = { version = "0.27.0", optional = true }
+pyo3 = { version = "0.27.2", features = ["extension-module", "abi3-py310", "generate-import-lib"], optional = true }
+numpy = { version = "0.27.1", optional = true }
 
 # Test-only utils
 itertools = { version = "0.14.0", optional = true }
@@ -32,9 +32,10 @@ criterion = "0.7.0"
 ndarray = "0.17.1"
 
 [features]
-default = ["std"]
+default = ["std", "crunchy/limit_64"]
 python = ["numpy", "pyo3", "std"]
 std = ["itertools"]
+deep-unroll = ["crunchy/limit_256"]
 fma = []
 
 [profile.release]

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: interpn
-Version: 0.8.2
+Version: 0.9.0
 Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy
@@ -21,6 +21,7 @@ Project-URL: changelog, https://github.com/jlogan03/interpn/blob/main/CHANGELOG.
 
 # InterpN
 
+[Writeup](https://jlogan.dev/blog/#2025-11-10-interpn-fast-interpolation) |
 [Repo](https://github.com/jlogan03/interpn) |
 [Python Docs](https://interpnpy.readthedocs.io/en/latest/) |
 [Rust Docs](https://docs.rs/interpn/latest/interpn/)
@@ -30,11 +31,6 @@ prioritizing correctness, performance, and compatiblity with memory-constrained
 
 Available as a rust crate and python library.
 
-These methods perform zero allocation when evaluated (except, optionally, for the output).
-Because of this, they have minimal per-call overhead, and are particularly
-effective when examining small numbers of observation points. See the
-[performance](https://interpnpy.readthedocs.io/en/latest/perf/) page for detailed benchmarks.
-
 ## Features
 
 | Feature →<br>↓ Interpolant Method | Regular<br>Grid | Rectilinear<br>Grid | Json<br>Serialization |
@@ -49,10 +45,12 @@ The methods provided here, while more limited in scope than scipy's,
 * use almost no RAM (and perform no heap allocations at all)
 * produce significantly improved floating-point error (by several orders of magnitude)
 * are json-serializable using Pydantic
-* can also be used easily in web and embedded applications via the Rust library
+* can also be used easily in web, embedded and gpu applications via the Rust library
 * are permissively licensed
 
-![ND throughput 1000 obs](./docs/throughput_vs_dims_1000_obs.svg)
+<br>
+
+<img src="./docs/speedup_vs_dims_1_obs_linear_inverted.svg" alt="">
 
 See [here](https://interpnpy.readthedocs.io/en/latest/perf/) for more info about quality-of-fit, throughput, and memory usage.
 
@@ -64,8 +62,7 @@ pip install interpn
 
 ## Profile-Guided Optimization
 
-To build the extension with profile-guided optimization using pre-built profiles, do `sh ./scripts/distr_pgo_install.sh`.
-You can also generate your own PGO profiles like `sh ./scripts/distr_pgo_profile.sh`.
+To build the extension with profile-guided optimization, do `sh ./scripts/distr_pgo.sh`
 after installing this extra compiler dependency:
 
 ```bash

{interpn-0.8.2 → interpn-0.9.0}/README.md

@@ -1,5 +1,6 @@
 # InterpN
 
+[Writeup](https://jlogan.dev/blog/#2025-11-10-interpn-fast-interpolation) |
 [Repo](https://github.com/jlogan03/interpn) |
 [Python Docs](https://interpnpy.readthedocs.io/en/latest/) |
 [Rust Docs](https://docs.rs/interpn/latest/interpn/)
@@ -9,11 +10,6 @@ prioritizing correctness, performance, and compatiblity with memory-constrained
 
 Available as a rust crate and python library.
 
-These methods perform zero allocation when evaluated (except, optionally, for the output).
-Because of this, they have minimal per-call overhead, and are particularly
-effective when examining small numbers of observation points. See the
-[performance](https://interpnpy.readthedocs.io/en/latest/perf/) page for detailed benchmarks.
-
 ## Features
 
 | Feature →<br>↓ Interpolant Method | Regular<br>Grid | Rectilinear<br>Grid | Json<br>Serialization |
@@ -28,10 +24,12 @@ The methods provided here, while more limited in scope than scipy's,
 * use almost no RAM (and perform no heap allocations at all)
 * produce significantly improved floating-point error (by several orders of magnitude)
 * are json-serializable using Pydantic
-* can also be used easily in web and embedded applications via the Rust library
+* can also be used easily in web, embedded and gpu applications via the Rust library
 * are permissively licensed
 
-![ND throughput 1000 obs](./docs/throughput_vs_dims_1000_obs.svg)
+<br>
+
+<img src="./docs/speedup_vs_dims_1_obs_linear_inverted.svg" alt="">
 
 See [here](https://interpnpy.readthedocs.io/en/latest/perf/) for more info about quality-of-fit, throughput, and memory usage.
 
@@ -43,8 +41,7 @@ pip install interpn
 
 ## Profile-Guided Optimization
 
-To build the extension with profile-guided optimization using pre-built profiles, do `sh ./scripts/distr_pgo_install.sh`.
-You can also generate your own PGO profiles like `sh ./scripts/distr_pgo_profile.sh`.
+To build the extension with profile-guided optimization, do `sh ./scripts/distr_pgo.sh`
 after installing this extra compiler dependency:
 
 ```bash

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

@@ -92,13 +92,7 @@ def _plot_normalized_vs_nobs(
     title: str,
     output_path: Path,
 ) -> None:
-    fig = make_subplots(
-        rows=2,
-        cols=1,
-        subplot_titles=["Linear", "Cubic"],
-        horizontal_spacing=0.08,
-    )
-    for row, kind in enumerate(["Linear", "Cubic"], start=1):
+    for kind in ["Linear", "Cubic"]:
         baseline_vals = throughputs.get(f"Scipy RegularGridInterpolator {kind}")
         if not baseline_vals:
             continue
@@ -108,7 +102,7 @@ def _plot_normalized_vs_nobs(
             for name, values in throughputs.items()
             if kinds.get(name) == kind and values
         ]
-
+        fig = make_subplots(rows=1, cols=1)
         for idx, (label, values) in enumerate(series):
             min_len = min(len(values), len(baseline_arr))
             if min_len == 0:
@@ -116,6 +110,12 @@ def _plot_normalized_vs_nobs(
             x_vals = np.array(ns[:min_len])
             ratios = values[:min_len] / baseline_arr[:min_len]
             is_baseline = label.startswith("Scipy RegularGridInterpolator")
+            legend_name = "Scipy" if is_baseline else "InterpN"
+            showlegend = True
+            for trace in fig.data:
+                if trace.name == legend_name:
+                    showlegend = False
+                    break
             fig.add_trace(
                 go.Scatter(
                     x=x_vals,
@@ -129,13 +129,9 @@ def _plot_normalized_vs_nobs(
                         size=8,
                     ),
                     opacity=1.0,
-                    name=label if not is_baseline else f"{label}<br>(baseline)",
-                    showlegend=True,
-                    legendgroup=kind,
-                    legendgrouptitle_text=kind,
-                ),
-                row=row,
-                col=1,
+                    name=legend_name,
+                    showlegend=showlegend,
+                )
             )
             if not is_baseline and label.startswith("InterpN"):
                 ones = np.ones_like(ratios)
@@ -146,81 +142,59 @@ def _plot_normalized_vs_nobs(
                     x=x_vals,
                     upper=upper,
                     lower=lower,
-                    row=row,
+                    row=1,
                     col=1,
                 )
-
-    fig.update_xaxes(
-        type="log",
-        row=1,
-        col=1,
-        showline=True,
-        linecolor="black",
-        linewidth=1,
-        mirror=True,
-        ticks="outside",
-        tickcolor="black",
-        showgrid=False,
-        zeroline=False,
-    )
-    fig.update_xaxes(
-        type="log",
-        title_text="Number of Observation Points",
-        row=2,
-        col=1,
-        showline=True,
-        linecolor="black",
-        linewidth=1,
-        mirror=True,
-        ticks="outside",
-        tickcolor="black",
-        showgrid=False,
-        zeroline=False,
-    )
-    fig.update_yaxes(
-        title_text="Speedup vs. Scipy",
-        row=1,
-        col=1,
-        showline=True,
-        linecolor="black",
-        linewidth=1,
-        mirror=True,
-        ticks="outside",
-        tickcolor="black",
-        showgrid=False,
-        zeroline=False,
-    )
-    fig.update_yaxes(
-        row=2,
-        col=1,
-        showline=True,
-        linecolor="black",
-        linewidth=1,
-        mirror=True,
-        ticks="outside",
-        tickcolor="black",
-        showgrid=False,
-        zeroline=False,
-    )
-    fig.update_layout(
-        title=dict(text=title, y=0.97, yanchor="top"),
-        height=450,
-        margin=dict(t=80, l=60, r=200, b=90),
-        legend=dict(
-            orientation="v",
-            yanchor="top",
-            y=1.0,
-            x=1.02,
-            xanchor="left",
-        ),
-        plot_bgcolor="rgba(0,0,0,0)",
-        paper_bgcolor="rgba(0,0,0,0)",
-    )
-    fig.write_image(str(output_path))
-    fig.write_html(
-        str(output_path.with_suffix(".html")), include_plotlyjs="cdn", full_html=False
-    )
-    fig.show()
+        fig.update_xaxes(
+            type="log",
+            title_text="Number of Observation Points",
+            row=1,
+            col=1,
+            showline=True,
+            linecolor="black",
+            linewidth=1,
+            mirror=True,
+            ticks="outside",
+            tickcolor="black",
+            showgrid=False,
+            zeroline=False,
+        )
+        fig.update_yaxes(
+            title_text="Normalized Throughput",
+            row=1,
+            col=1,
+            showline=True,
+            linecolor="black",
+            linewidth=1,
+            mirror=True,
+            ticks="outside",
+            tickcolor="black",
+            showgrid=False,
+            zeroline=False,
+        )
+        fig.update_layout(
+            title=dict(text=f"{title}, {kind}", y=0.97, yanchor="top"),
+            height=400,
+            margin=dict(t=60, l=60, r=60, b=110),
+            legend=dict(
+                orientation="h",
+                yanchor="top",
+                y=1.0,
+                x=1.0,
+                xanchor="right",
+            ),
+            plot_bgcolor="rgba(0,0,0,0)",
+            paper_bgcolor="rgba(0,0,0,0)",
+            font=dict(color="black"),
+        )
+        per_kind_output = output_path.with_stem(output_path.stem + f"_{kind.lower()}")
+        fig.write_image(str(per_kind_output))
+        fig.write_html(
+            str(per_kind_output.with_suffix(".html")),
+            include_plotlyjs="cdn",
+            full_html=False,
+        )
+        fig.show()
 
 
 def _plot_throughput_vs_dims(
@@ -368,6 +342,7 @@ def _plot_throughput_vs_dims(
         ),
         plot_bgcolor="rgba(0,0,0,0)",
         paper_bgcolor="rgba(0,0,0,0)",
+        font=dict(color="black"),
     )
     fig.write_image(str(output_path))
     fig.write_html(
@@ -479,6 +454,7 @@ def _plot_speedup_vs_dims(
             # ),
             plot_bgcolor="rgba(0,0,0,0)",
             paper_bgcolor="rgba(0,0,0,0)",
+            font=dict(color="black"),
         )
         output_path = output_dir / f"speedup_vs_dims_{nobs}_obs_{kind.lower()}.svg"
         fig.write_image(str(output_path))