PERF block reduce in wrapping

GalSim-developers / JAX-GalSim

JAX port of GalSim, for parallelized, GPU accelerated, and differentiable galaxy image simulations.

Other

25 stars 3 forks source link

Closed beckermr closed 3 months ago

beckermr commented 3 months ago

This PR uses a block-wise reduction algorithm for wrapping images. It should fix the performance issues of our previous double for loop.

fixes #96