BlogDocs

refactor: do interpolation class init on the host(#106)

Merged
Comparing
host-init-interp
(
997f855
) with
main
(
b6cbbcd
)
×3.2
IMPROVEMENTS
2
REGRESSIONS
0
UNTOUCHED
9
NEW
0
DROPPED
0
IGNORED
0

Benchmarks

Improved

test_benchmarks_lanczos_interp[kval-no_conserve_dc-compile]
tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[kval-no_conserve_dc-compile]
×3.2
1,558 ms
492 ms
test_benchmarks_lanczos_interp[xval-no_conserve_dc-compile]
tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[xval-no_conserve_dc-compile]
+63%
261.5 ms
160.2 ms

Passed

test_benchmarks_lanczos_interp[kval-conserve_dc-compile]
tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[kval-conserve_dc-compile]
+7%
5.4 s
5 s
test_benchmarks_lanczos_interp[xval-conserve_dc-compile]
tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[xval-conserve_dc-compile]
+4%
315.1 ms
302.8 ms
test_benchmarks_lanczos_interp[kval-no_conserve_dc-run]
tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[kval-no_conserve_dc-run]
+3%
4 ms
3.9 ms
test_benchmarks_interpolated_image[compile]
tests/jax/test_benchmarks.py::test_benchmarks_interpolated_image[compile]
+1%
1.7 s
1.6 s
test_benchmarks_metacal[run]
tests/jax/test_benchmarks.py::test_benchmarks_metacal[run]
+1%
1.4 s
1.3 s
test_benchmarks_lanczos_interp[kval-conserve_dc-run]
tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[kval-conserve_dc-run]
0%
42.3 ms
42.1 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-run]
tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[xval-no_conserve_dc-run]
0%
1.6 ms
1.6 ms
test_benchmarks_interpolated_image[run]
tests/jax/test_benchmarks.py::test_benchmarks_interpolated_image[run]
0%
48.9 ms
48.8 ms

Commits

Click on a commit to change the comparison range
base
main
b6cbbcd
+217%
fix: convert jax.lax.cond to python if
997f855
4 months ago
by beckermr
ResourcesHomePricingDocsBlogGitHubChangelog
Copyright © 2024 CodSpeed Technology SAS. All rights reserved.