Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

sky_imager speedups #12

Open
cbassa opened this issue Feb 3, 2024 · 0 comments
Open

sky_imager speedups #12

cbassa opened this issue Feb 3, 2024 · 0 comments

Comments

@cbassa
Copy link

cbassa commented Feb 3, 2024

Some different implementations of the sky_imager function, using different approaches to speed up the calculations, are given here.

The speed results are as follows:

numpy_real: 6.080s
numpy: 6.484s
numpy_float32_ravel: 4.013s
numpy_float32: 4.853s
cupy: 0.281s
simple: 6.530s
numba: 3.483s

Using numpy functionality, the best combination appears to be a combination of float32, using ravel to only compute on sky pixels, computing only the real part of the complex multiplication, and using numpy broadcasting.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant