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

Roadmap for Multi-GPU cuda healpix fft #207

Open
5 tasks
ASKabalan opened this issue Jul 29, 2024 · 0 comments
Open
5 tasks

Roadmap for Multi-GPU cuda healpix fft #207

ASKabalan opened this issue Jul 29, 2024 · 0 comments
Labels
enhancement New feature or request

Comments

@ASKabalan
Copy link
Collaborator

Quick roadmap on the multi GPU healpix FFT

For the single GPU one

  • Merge first PR Cufft primitive #204 that uses CUFFT as backend for FFTs and does the spectral extensions and folding in cuda kernels
  • Merge all kernels in one to have parallel ffts in one kernel
  • Handle reality case with C2R transforms

For the multi GPU one

  • Discuss load balancing
  1. There is not way to distribute a healpix array in a clean way .. distributing must not have conditions on local slice index otherwise it will not scale
  2. Distributing probably requires having a seperate padding function, that transforms a healpix array into a padded one with the shape of the corresponding FTM
  3. Padded healpix array remains padded at all times to make sure that it is always distributable
  4. FLM are squares so they can be easily distributed but only in 1 dimension since the other is always (number of rings) is always odd
  • Implement padding operator
@ASKabalan ASKabalan added the enhancement New feature or request label Jul 29, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

1 participant