GalSim-developers
JAX-GalSim
BlogDocsChangelog

perf: optimize flux frac for interpolated images by removing fori_loop

#134Merged
Comparing
opt-flux-frac
(
3abd2f0
) with
main
(
9b2c0b7
)
CodSpeed Performance Gauge
-34%
Improvements
2
Regressions
1
Untouched
22

Benchmarks

Failed

test_benchmark_interpimage_flux_frac[run]
tests/jax/test_benchmarks.py
Regression
CodSpeed Performance Gauge
-34%
1.8 ms2.7 ms

Improved

test_benchmark_spergel_init[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+98%
165.2 µs83.4 µs
test_benchmark_interpimage_flux_frac[compile]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+20%
282.5 ms235.4 ms

Passed

test_benchmarks_lanczos_interp[xval-no_conserve_dc-run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+12%
952.6 µs848.3 µs
test_benchmark_spergel_init[compile]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
16 ms16 ms
test_benchmarks_lanczos_interp[xval-conserve_dc-compile]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
124.5 ms124.3 ms
test_benchmark_gaussian_init[compile]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
15.8 ms15.8 ms
test_benchmarks_lanczos_interp[xval-no_conserve_dc-compile]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
80.7 ms80.7 ms
test_benchmarks_interpolated_image[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
46.3 ms46.3 ms
test_benchmark_rng_discard[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
6.9 ms6.9 ms
test_benchmarks_lanczos_interp[kval-no_conserve_dc-compile]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
110.3 ms110.3 ms
test_benchmark_spergel_xvalue[compile]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
27.7 s27.7 s
test_benchmark_spergel_xvalue[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
24.5 s24.5 s
test_benchmarks_lanczos_interp[kval-no_conserve_dc-run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
385.3 µs385.4 µs
test_benchmark_spergel_conv[compile]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
3.3 s3.3 s
test_benchmarks_lanczos_interp[xval-conserve_dc-run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
1.1 ms1.1 ms
test_benchmarks_lanczos_interp[kval-conserve_dc-compile]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
110.3 ms110.4 ms
test_benchmark_spergel_kvalue[compile]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
725.6 ms726.2 ms
test_benchmark_spergel_kvalue[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
137.2 ms137.3 ms
test_benchmark_spergel_conv[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
2.6 s2.6 s
test_benchmarks_metacal[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
185 ms185.3 ms
test_benchmarks_interpolated_image[compile]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
1.8 s1.8 s
test_benchmark_gaussian_init[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
84.2 µs84.4 µs
test_benchmarks_lanczos_interp[kval-conserve_dc-run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
383.9 µs385 µs
test_benchmark_rng_discard[compile]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
-1%
234.1 ms237.2 ms

Commits

Click on a commit to change the comparison range
Base
main
9b2c0b7
-0.44%
fix: use -inf for zeros
c615063
1 year ago
by beckermr
-33.46%
Merge branch 'main' into opt-flux-frac
3abd2f0
1 year ago
by beckermr
© 2026 CodSpeed Technology
Home Terms Privacy Docs