BlogDocs

perf: use vmap for iimage(#104)

Merged
Comparing
vmap-iimage
(
33736d0
) with
main
(
57bea2b
)
+58%
IMPROVEMENTS
2
REGRESSIONS
0
UNTOUCHED
9
NEW
0
DROPPED
0
IGNORED
0

Benchmarks

Improved

test_benchmarks_interpolated_image[run]
tests/jax/test_benchmarks.py::test_benchmarks_interpolated_image[run]
+58%
77.3 ms
48.9 ms
test_benchmarks_interpolated_image[compile]
tests/jax/test_benchmarks.py::test_benchmarks_interpolated_image[compile]
+11%
1.9 s
1.7 s

Passed

test_benchmarks_metacal[run]
tests/jax/test_benchmarks.py::test_benchmarks_metacal[run]
+4%
1.4 s
1.4 s
test_benchmarks_lanczos_interp[kval-no_conserve_dc-compile]
tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[kval-no_conserve_dc-compile]
0%
1.6 s
1.6 s
test_benchmarks_lanczos_interp[kval-conserve_dc-compile]
tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[kval-conserve_dc-compile]
0%
5.4 s
5.4 s
test_benchmarks_lanczos_interp[xval-conserve_dc-compile]
tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[xval-conserve_dc-compile]
0%
315.1 ms
315.3 ms
test_benchmarks_lanczos_interp[xval-no_conserve_dc-run]
tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[xval-no_conserve_dc-run]
0%
1.6 ms
1.6 ms
test_benchmarks_lanczos_interp[xval-conserve_dc-run]
tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[xval-conserve_dc-run]
0%
2.2 ms
2.2 ms
test_benchmarks_lanczos_interp[xval-no_conserve_dc-compile]
tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[xval-no_conserve_dc-compile]
0%
261.2 ms
261.8 ms
test_benchmarks_lanczos_interp[kval-conserve_dc-run]
tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[kval-conserve_dc-run]
0%
42.1 ms
42.2 ms
test_benchmarks_lanczos_interp[kval-no_conserve_dc-run]
tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[kval-no_conserve_dc-run]
-9%
4 ms
4.4 ms

Commits

Click on a commit to change the comparison range
base
main
57bea2b
+47%
perf: use vmap for iimage
4ee1302
4 months ago
by beckermr
-58%
fix: reconstruct tree nodes properly
c200072
4 months ago
by beckermr
0%
Merge branch 'main' into vmap-iimage
abb055b
4 months ago
by beckermr
+69%
Merge branch 'vmap-iimage' of https://github.com/GalSim-developers/JAX-GalSim into vmap-iimage
33736d0
4 months ago
by beckermr
ResourcesHomePricingDocsBlogGitHubChangelog
Getting StartedSample repositoryExplore repositoriesSupport
AboutCareersTwitterDiscordContact UsTerms of ServicePrivacy Policy
Copyright © 2024 CodSpeed Technology SAS. All rights reserved.