Unroll loops for interpolated images

GalSim-developers / JAX-GalSim

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

Other

25 stars 3 forks source link

Unroll loops for interpolated images #98

Closed beckermr closed 2 months ago

beckermr commented 3 months ago

The interpolated image code has short fori_loops over the interpolation kernels. We should unroll these so they get executed in a single pass on the device.