search

GalSim-developers / JAX-GalSim

JAX port of GalSim, for parallelized, GPU accelerated, and differentiable galaxy image simulations.
Other
27 stars 3 forks source link

perf: use vmap for iimage #104

Closed beckermr closed 3 months ago

beckermr commented 3 months ago

This PR switches the code to vmap for interpolated images in order to increase performance.

codspeed-hq[bot] commented 3 months ago

CodSpeed Performance Report

Merging #104 will improve performances by 58.15%

Comparing vmap-iimage (33736d0) with main (57bea2b)

Summary

⚡ 2 improvements ✅ 9 untouched benchmarks

Benchmarks breakdown

Benchmark main vmap-iimage Change
⚡ test_benchmarks_interpolated_image[compile] 1.9 s 1.7 s +10.93%
⚡ test_benchmarks_interpolated_image[run] 77.3 ms 48.9 ms +58.15%
beckermr commented 3 months ago

@jecampagne @ismael-mendoza can you review this one? It is ready to go!