Avatar for the GalSim-developers user
GalSim-developers
JAX-GalSim
BlogDocsChangelog

feat: use None as default for trunc to avoid JAX branches on trunc value

#191Closed
Comparing
moffat-trunc-to-none
(
fd10c6c
) with
main
(
51df286
)
CodSpeed Performance Gauge
×5.3
Improvement
2
Untouched
34

Benchmarks

36 total
test_benchmark_moffat_init[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
×5.3
135.9 µs25.5 µs
test_benchmark_moffat_init[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
×2.1
491.8 µs233.7 µs
test_benchmarks_lanczos_interp[xval-no_conserve_dc-run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+12%
113.2 µs101.2 µs
test_benchmarks_lanczos_interp[kval-conserve_dc-run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+7%
46.3 µs43.1 µs
test_benchmark_interpimage_flux_frac[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+6%
2.5 ms2.3 ms
test_benchmarks_lanczos_interp[xval-conserve_dc-run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+5%
122.8 µs117.3 µs
test_benchmark_spergel_calcfluxrad[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+4%
192.4 µs184.3 µs
test_benchmark_spergel_conv[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+3%
279.5 ms272.2 ms
test_benchmark_invert_ab_noraise[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+2%
131.4 µs128.8 µs
test_benchmark_interpimage_flux_frac[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+1%
436.1 µs432.5 µs
test_benchmark_rng_discard[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+1%
109.9 µs109.2 µs
test_benchmarks_interpolated_image[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
5.1 ms5.1 ms
test_benchmark_spergel_kvalue[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
122.1 ms121.8 ms
test_benchmarks_lanczos_interp[xval-no_conserve_dc-run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
959.4 µs957.6 µs
test_benchmarks_metacal[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
179.6 ms179.3 ms
test_benchmark_spergel_calcfluxrad[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
1.1 ms1.1 ms
test_benchmarks_lanczos_interp[kval-conserve_dc-run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
432.6 µs432.3 µs
test_benchmark_spergel_init[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
96.5 µs96.5 µs
test_benchmark_moffat_conv_grad[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
13.6 s13.6 s
test_benchmark_spergel_xvalue[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
33.6 s33.6 s
test_benchmark_spergel_conv[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
2.6 s2.6 s
test_benchmark_moffat_conv[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
10.4 s10.4 s
test_benchmarks_lanczos_interp[kval-no_conserve_dc-run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
432.9 µs433.5 µs
test_benchmarks_interpolated_image[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
47.6 ms47.8 ms
test_benchmark_spergel_init[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
-1%
6 µs6.1 µs

Commits

Click on a commit to change the comparison range
Base
main
51df286
×5.2
feat: use None as default for trunc to avoid JAX branches on trunc value
a2049e5
4 days ago
by beckermr
+5.08%
fix: correctly vmap None in tests of vmapping
5eeedab
4 days ago
by beckermr
+0.26%
fix: fix local tests
6656353
4 days ago
by beckermr
-7.86%
fix: update submodule for testsing
87b0b8e
4 days ago
by beckermr
+18.37%
fix: init galsim moffat objects correctly
fd10c6c
4 days ago
by beckermr
© 2026 CodSpeed Technology
Home Terms Privacy Docs