Optimize `compute_zdiff_gradp` - #1272

Draft

msimberg wants to merge 46 commits into

msimberg:zdiff-vertoffset-gradp

Commits on May 19, 2026

perf(metrics): replace nested python loops with np.searchsorted in compute_zdiff_gradp
msimberg
committed
d725195
perf(metrics): eliminate vertidx_gradp intermediate array in compute_zdiff_gradp
msimberg
committed
9aade50
perf(metrics): pre-compute ascending z_ifc and unroll per-side loops
msimberg
committed
bc8f00c
test(metrics): remove cpu_only mark from test_factory_zdiff_gradp
msimberg
committed
a496ec1
perf(metrics): batched 2D searchsorted via row-offset trick (Divakar)
msimberg
committed
411a961
refactor(metrics): use array_ns instead of manual backend detection
msimberg
committed
86e95db
perf(metrics): remove z_me mask, use do_exchange for zdiff_gradp halo
msimberg
committed
0bff01e
perf(metrics): remove internal z_me exchange, drop exchange parameter
msimberg
committed
1706d16
Only run common mpi tests
msimberg
committed
7afb739

Commits on May 20, 2026

Refactor compute_zdiff_gradp to only do do_exchange exchange
msimberg
committed
84bb917
Some more refactoring
msimberg
committed
ffa6397
Merge remote-tracking branch 'origin/main' into zdiff-vertoffset-gradp
msimberg
committed
1d48eeb

Commits on Jun 1, 2026

Apply pre-commit auto-fixes from merge
msimberg
committed
6ca86c9

Commits on Aug 24, 2026

Commits on Aug 25, 2026

Commits on Aug 26, 2026

Commits on Aug 27, 2026

Commits on Aug 28, 2026

[skip ci] simplify: remove tie correction, document premise
msimberg
committed
ee9861c

Commits on Aug 29, 2026

Commits on Aug 31, 2026

docs: replace figurative "carry" with "lower-bound update" and split long sentences (R71, R72) [skip ci]
msimberg
committed
2d0d538