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

perf: try using fori_loop but unrolling everything

#141Merged
Comparing
fori_loop_unroll
(
2dac00e
) with
main
(
3e45fb5
)
CodSpeed Performance Gauge
0%
Improvements
0
Regressions
0
Untouched
29
New
2
Dropped
0
Ignored
0

Benchmarks

New

test_benchmark_spergel_calcfluxrad[compile]
tests/jax/test_benchmarks.py::test_benchmark_spergel_calcfluxrad[compile]
CodSpeed Performance Gauge
N/A
N/A
15.4 s
test_benchmark_spergel_calcfluxrad[run]
tests/jax/test_benchmarks.py::test_benchmark_spergel_calcfluxrad[run]
CodSpeed Performance Gauge
N/A
N/A
2.1 ms

Passed

test_benchmark_moffat_init[compile]
tests/jax/test_benchmarks.py::test_benchmark_moffat_init[compile]
CodSpeed Performance Gauge
+8%
500.6 ms
464.9 ms
test_benchmark_interpimage_flux_frac[run]
tests/jax/test_benchmarks.py::test_benchmark_interpimage_flux_frac[run]
CodSpeed Performance Gauge
+4%
2.8 ms
2.7 ms
test_benchmark_invert_ab_noraise[compile]
tests/jax/test_benchmarks.py::test_benchmark_invert_ab_noraise[compile]
CodSpeed Performance Gauge
+1%
1.6 s
1.6 s
test_benchmark_rng_discard[compile]
tests/jax/test_benchmarks.py::test_benchmark_rng_discard[compile]
CodSpeed Performance Gauge
0%
234.6 ms
233.5 ms
test_benchmarks_lanczos_interp[kval-no_conserve_dc-run]
tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[kval-no_conserve_dc-run]
CodSpeed Performance Gauge
0%
385.3 µs
384.7 µs
test_benchmarks_lanczos_interp[kval-no_conserve_dc-compile]
tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[kval-no_conserve_dc-compile]
CodSpeed Performance Gauge
0%
110.3 ms
110.2 ms
test_benchmark_spergel_kvalue[run]
tests/jax/test_benchmarks.py::test_benchmark_spergel_kvalue[run]
CodSpeed Performance Gauge
0%
137.3 ms
137.2 ms
test_benchmark_spergel_conv[compile]
tests/jax/test_benchmarks.py::test_benchmark_spergel_conv[compile]
CodSpeed Performance Gauge
0%
3.3 s
3.3 s
test_benchmark_gaussian_init[compile]
tests/jax/test_benchmarks.py::test_benchmark_gaussian_init[compile]
CodSpeed Performance Gauge
0%
15.8 ms
15.8 ms
test_benchmarks_metacal[run]
tests/jax/test_benchmarks.py::test_benchmarks_metacal[run]
CodSpeed Performance Gauge
0%
185 ms
184.9 ms
test_benchmarks_interpolated_image[compile]
tests/jax/test_benchmarks.py::test_benchmarks_interpolated_image[compile]
CodSpeed Performance Gauge
0%
1.8 s
1.8 s
test_benchmark_spergel_init[compile]
tests/jax/test_benchmarks.py::test_benchmark_spergel_init[compile]
CodSpeed Performance Gauge
0%
16 ms
16 ms
test_benchmark_spergel_xvalue[run]
tests/jax/test_benchmarks.py::test_benchmark_spergel_xvalue[run]
CodSpeed Performance Gauge
0%
24.5 s
24.5 s
test_benchmark_spergel_xvalue[compile]
tests/jax/test_benchmarks.py::test_benchmark_spergel_xvalue[compile]
CodSpeed Performance Gauge
0%
27.7 s
27.7 s
test_benchmarks_lanczos_interp[xval-conserve_dc-compile]
tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[xval-conserve_dc-compile]
CodSpeed Performance Gauge
0%
124.3 ms
124.3 ms
test_benchmark_gaussian_init[run]
tests/jax/test_benchmarks.py::test_benchmark_gaussian_init[run]
CodSpeed Performance Gauge
0%
85 µs
85 µs
test_benchmark_spergel_conv[run]
tests/jax/test_benchmarks.py::test_benchmark_spergel_conv[run]
CodSpeed Performance Gauge
0%
2.6 s
2.6 s
test_benchmark_spergel_kvalue[compile]
tests/jax/test_benchmarks.py::test_benchmark_spergel_kvalue[compile]
CodSpeed Performance Gauge
0%
727.1 ms
727.3 ms
test_benchmarks_lanczos_interp[kval-conserve_dc-compile]
tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[kval-conserve_dc-compile]
CodSpeed Performance Gauge
0%
110.4 ms
110.4 ms
test_benchmark_moffat_init[run]
tests/jax/test_benchmarks.py::test_benchmark_moffat_init[run]
CodSpeed Performance Gauge
0%
408.6 µs
408.8 µs
test_benchmarks_lanczos_interp[xval-no_conserve_dc-run]
tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[xval-no_conserve_dc-run]
CodSpeed Performance Gauge
0%
952.1 µs
953.3 µs
test_benchmark_invert_ab_noraise[run]
tests/jax/test_benchmarks.py::test_benchmark_invert_ab_noraise[run]
CodSpeed Performance Gauge
0%
720.3 µs
721.2 µs
test_benchmark_rng_discard[run]
tests/jax/test_benchmarks.py::test_benchmark_rng_discard[run]
CodSpeed Performance Gauge
0%
6.9 ms
6.9 ms
test_benchmarks_lanczos_interp[xval-no_conserve_dc-compile]
tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[xval-no_conserve_dc-compile]
CodSpeed Performance Gauge
0%
80.6 ms
80.7 ms
test_benchmarks_lanczos_interp[kval-conserve_dc-run]
tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[kval-conserve_dc-run]
CodSpeed Performance Gauge
0%
385.3 µs
386.3 µs
test_benchmark_interpimage_flux_frac[compile]
tests/jax/test_benchmarks.py::test_benchmark_interpimage_flux_frac[compile]
CodSpeed Performance Gauge
-1%
234.4 ms
235.7 ms
test_benchmarks_interpolated_image[run]
tests/jax/test_benchmarks.py::test_benchmarks_interpolated_image[run]
CodSpeed Performance Gauge
-1%
46.3 ms
46.6 ms
test_benchmark_spergel_init[run]
tests/jax/test_benchmarks.py::test_benchmark_spergel_init[run]
CodSpeed Performance Gauge
-1%
83.8 µs
85 µs
test_benchmarks_lanczos_interp[xval-conserve_dc-run]
tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[xval-conserve_dc-run]
CodSpeed Performance Gauge
-9%
1.1 ms
1.2 ms

Commits

Click on a commit to change the comparison range
Base
main
3e45fb5
0%
perf: try using fori_loop but unrolling everything
0571c59
2 months ago
by beckermr
0%
fix: cannot unroll discard since n is not static
5945ad1
2 months ago
by beckermr
0%
test: reduce matrix
cca6a20
2 months ago
by beckermr
0%
Update utils.py
e2fe74d
2 months ago
by beckermr
-53%
perf: try partial loop unrolling
0d67579
2 months ago
by beckermr
+54%
test: add spergel init benchmark
d15e76d
2 months ago
by beckermr
0%
style: pre the commit
2dac00e
2 months ago
by beckermr
© 2025 CodSpeed Technology
Home Terms PrivacyDocs