faster-interp - Branch - GalSim-developers/JAX-GalSim

Blog Docs

perf: faster kval for Lanczos interpolated images via cached splines(#101)

Merged

Merging

faster-interp

(

be95a0a

) into

main

(

ec2153e

)

-21%

IMPROVEMENTS: 6

REGRESSIONS: 1

UNTOUCHED: 4

NEW: 0

DROPPED: 0

IGNORED: 0

Benchmarks

Failed

test_benchmarks_lanczos_interp[xval-no_conserve_dc-run]Regression

tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[xval-no_conserve_dc-run]

-21%

731.1 µs

929.9 µs

Improved

test_benchmarks_lanczos_interp[kval-conserve_dc-run]

tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[kval-conserve_dc-run]

×120

42,142 µs

340.6 µs

test_benchmarks_lanczos_interp[kval-conserve_dc-compile]

tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[kval-conserve_dc-compile]

×39

5,028.7 ms

129.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]

×9.3

4,143.6 µs

443.8 µs

test_benchmarks_metacal[run]

tests/jax/test_benchmarks.py::test_benchmarks_metacal[run]

×6.3

1,345.3 ms

212.4 ms

test_benchmarks_lanczos_interp[kval-no_conserve_dc-compile]

tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[kval-no_conserve_dc-compile]

×3.8

491.5 ms

129.4 ms

test_benchmarks_lanczos_interp[xval-conserve_dc-run]

tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[xval-conserve_dc-run]

+25%

975 µs

778.3 µs

Passed

test_benchmarks_interpolated_image[run]

tests/jax/test_benchmarks.py::test_benchmarks_interpolated_image[run]

48.9 ms

test_benchmarks_interpolated_image[compile]

tests/jax/test_benchmarks.py::test_benchmarks_interpolated_image[compile]

1.6 s

test_benchmarks_lanczos_interp[xval-no_conserve_dc-compile]

tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[xval-no_conserve_dc-compile]

88.1 ms

test_benchmarks_lanczos_interp[xval-conserve_dc-compile]

tests/jax/test_benchmarks.py::test_benchmarks_lanczos_interp[xval-conserve_dc-compile]

143.1 ms

143.2 ms

Commits

Click on a commit to change the comparison range

base

main

ec2153e

-88%

perf: unroll loops explicitly

8b5d43d

3 months ago

by beckermr

+1%

perf: unroll everything

4798a97

3 months ago

by beckermr

+99%

perf: try vampped interps

d960a29

3 months ago

by beckermr

fix: need to pass step for vmap test for jax metacal

a75b6be

3 months ago

by beckermr

+1%

style: blacken

ffc09c4

3 months ago

by beckermr

perf: force inlined interp table made once

5536c27

3 months ago

by beckermr

perf: use numpy data

368cef0

3 months ago

by beckermr

ref: use numpy and cache umax

d7edfd0

3 months ago

by beckermr

-34%

Merge branch 'main' into faster-interp

015a3c1

3 months ago

by beckermr

-3%

perf: cache at compile time for sinc

c3a4de8

3 months ago

by beckermr

+36%

perf: use fixed spacing to speed index lookups

fd2427c

3 months ago

by beckermr

test: add tests for different numbers of arrays

c1b3640

3 months ago

by beckermr

perf: try interpolated xals

0a2c547

3 months ago

by beckermr

Merge branch 'main' into faster-interp

d274f2a

3 months ago

by beckermr

+9,395%

test: run nearest tests now that bug is fixed

d7ead94

2 months ago

by beckermr

+62%

refactor: just use functools

3c48088

2 months ago

by beckermr

-28%

style: remove dead code

5da4385

2 months ago

by beckermr

-2,938%

Merge branch 'faster-interp' of https://github.com/GalSim-developers/JAX-GalSim into faster-interp

395fe9d

2 months ago

by beckermr

+1,203%

test: update tests for faster xval for lanczos

f18d6b6

2 months ago

by beckermr

-1,217%

test: put back more stringent tols for interps

87267a1

2 months ago

by beckermr

+2,928%

Merge branch 'main' into faster-interp

fb06338

2 months ago

by beckermr

+7%

fix: clean out old dead code + methods

7c62200

2 months ago

by beckermr

+15%

refactor: combine routines

19e1e89

2 months ago

by beckermr

-2,988%

Update jax_galsim/core/interpolate.py

c137e0e

2 months ago

by beckermr

-6,472%

style: blacken

be95a0a

2 months ago

by beckermr

ResourcesHome Pricing Docs Blog GitHub Changelog

Getting StartedSample repository Explore repositories Support

AboutCareers Twitter Discord Contact Us Terms of Service Privacy Policy