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_benchmark_moffat_conv_grad[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
-24%
1.3 s1.7 s
test_benchmark_spergel_xvalue[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
-23%
1.6 s2.1 s
test_benchmarks_lanczos_interp[xval-conserve_dc-run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
-14%
1 ms1.2 ms
test_benchmarks_lanczos_interp[xval-no_conserve_dc-run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+12%
113.2 µs101.2 µs
test_benchmark_invert_ab_noraise[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
-11%
1.1 ms1.2 ms
test_benchmark_rng_discard[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
-9%
564.9 µs623.5 µ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_moffat_conv[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
-7%
681.8 ms736.1 ms
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_benchmarks_metacal[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
-3%
18.3 ms18.8 ms
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_gaussian_init[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
-2%
6 µs6.1 µs
test_benchmark_spergel_kvalue[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
-1%
11.4 ms11.5 ms
test_benchmark_interpimage_flux_frac[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
+1%
436.1 µs432.5 µs
test_benchmark_gaussian_init[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
-1%
96.3 µs97 µs
test_benchmarks_lanczos_interp[kval-no_conserve_dc-run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
-1%
44.7 µs45 µs
test_benchmark_spergel_init[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
-1%
6 µs6.1 µ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%
47.6 ms47.8 ms
test_benchmarks_interpolated_image[run]
tests/jax/test_benchmarks.py
CodSpeed Performance Gauge
0%
5.1 ms5.1 ms

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 months ago
by beckermr
+5.08%
fix: correctly vmap None in tests of vmapping
5eeedab
4 months ago
by beckermr
+0.26%
fix: fix local tests
6656353
4 months ago
by beckermr
-7.86%
fix: update submodule for testsing
87b0b8e
4 months ago
by beckermr
+18.37%
fix: init galsim moffat objects correctly
fd10c6c
4 months ago
by beckermr
© 2026 CodSpeed Technology
Home Terms Privacy Docs