UCBerkeleySETI / hyperseti

A SETI / technosignature search code to find intelligent life beyond Earth
https://hyperseti.readthedocs.io
10 stars 4 forks source link

CODE OPTIMIZATION: Use less GPU memory #25

Closed telegraphic closed 1 year ago

telegraphic commented 3 years ago

The pipeline seems to be using much more GPU RAM than seems reasonable -- figure out why this is! Am I creating multiple copies of the same array?

texadactyl commented 3 years ago

Something to watch as it evolves: "Cuda Python" = cuda + numba, supported by nvidia. https://developer.nvidia.com/blog/numba-python-cuda-acceleration/

telegraphic commented 1 year ago

Stepped plan uses much less memory -- still optimization to do, but can now fit a full 2^20 channels on the GPU no issue.