GitHub Actions run
Instrumentation
2 months ago 0d67579 fori_loop_unrollCompare
Base
Search a run
Head
Compareperf: try using fori_loop but unrolling everything
#141
fori_loop_unroll
2 months ago
Benchmarks
test_benchmarks_lanczos_interp[xval-conserve_dc-run]
tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[xval-conserve_dc-run]1 ms
test_benchmarks_lanczos_interp[kval-conserve_dc-compile]
tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[kval-conserve_dc-compile]110.3 ms
test_benchmarks_interpolated_image[compile]
tests/jax/test_benchmarks.py::test_benchmarks_interpolated_image[compile]1.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]848.6 µs
test_benchmarks_lanczos_interp[xval-conserve_dc-compile]
tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[xval-conserve_dc-compile]124.5 ms
test_benchmarks_lanczos_interp[kval-no_conserve_dc-compile]
tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[kval-no_conserve_dc-compile]110.4 ms
test_benchmark_spergel_xvalue[run]
tests/jax/test_benchmarks.py::test_benchmark_spergel_xvalue[run]24.5 s
test_benchmark_spergel_xvalue[compile]
tests/jax/test_benchmarks.py::test_benchmark_spergel_xvalue[compile]27.7 s
test_benchmarks_lanczos_interp[kval-no_conserve_dc-run]
tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[kval-no_conserve_dc-run]385.3 µs
test_benchmark_spergel_conv[run]
tests/jax/test_benchmarks.py::test_benchmark_spergel_conv[run]2.6 s
test_benchmark_spergel_kvalue[compile]
tests/jax/test_benchmarks.py::test_benchmark_spergel_kvalue[compile]727.5 ms
test_benchmark_spergel_kvalue[run]
tests/jax/test_benchmarks.py::test_benchmark_spergel_kvalue[run]137.2 ms
test_benchmark_invert_ab_noraise[run]
tests/jax/test_benchmarks.py::test_benchmark_invert_ab_noraise[run]719.7 µs
test_benchmarks_lanczos_interp[kval-conserve_dc-run]
tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[kval-conserve_dc-run]383.8 µs
test_benchmark_rng_discard[run]
tests/jax/test_benchmarks.py::test_benchmark_rng_discard[run]6.9 ms
test_benchmark_spergel_init[run]
tests/jax/test_benchmarks.py::test_benchmark_spergel_init[run]179.8 µs
test_benchmarks_interpolated_image[run]
tests/jax/test_benchmarks.py::test_benchmarks_interpolated_image[run]46.6 ms
test_benchmark_moffat_init[compile]
tests/jax/test_benchmarks.py::test_benchmark_moffat_init[compile]465 ms
test_benchmark_interpimage_flux_frac[run]
tests/jax/test_benchmarks.py::test_benchmark_interpimage_flux_frac[run]2.7 ms
test_benchmark_invert_ab_noraise[compile]
tests/jax/test_benchmarks.py::test_benchmark_invert_ab_noraise[compile]1.6 s
test_benchmark_gaussian_init[compile]
tests/jax/test_benchmarks.py::test_benchmark_gaussian_init[compile]15.8 ms
test_benchmark_gaussian_init[run]
tests/jax/test_benchmarks.py::test_benchmark_gaussian_init[run]85 µs
test_benchmark_rng_discard[compile]
tests/jax/test_benchmarks.py::test_benchmark_rng_discard[compile]235 ms
test_benchmarks_metacal[run]
tests/jax/test_benchmarks.py::test_benchmarks_metacal[run]185 ms
test_benchmarks_lanczos_interp[xval-no_conserve_dc-compile]
tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[xval-no_conserve_dc-compile]80.9 ms
test_benchmark_moffat_init[run]
tests/jax/test_benchmarks.py::test_benchmark_moffat_init[run]407.7 µs
test_benchmark_interpimage_flux_frac[compile]
tests/jax/test_benchmarks.py::test_benchmark_interpimage_flux_frac[compile]237.4 ms
test_benchmark_spergel_init[compile]
tests/jax/test_benchmarks.py::test_benchmark_spergel_init[compile]16 ms
test_benchmark_spergel_conv[compile]
tests/jax/test_benchmarks.py::test_benchmark_spergel_conv[compile]3.3 s