desihub / gpu_specter

Scratch work for porting spectroperfectionism extractions to GPUs
BSD 3-Clause "New" or "Revised" License
2 stars 3 forks source link

gpu - move code to gpu via cupy and numba #10

Closed lastephey closed 4 years ago

lastephey commented 4 years ago

Once we have done the legwork to make gpu_specter gpu-friendly, we can transition the cpu code itself to gpu code. We learned how to do this before and during the March 2020 hackathon so we can apply what we've already learned.

These sections of code include:

^^ These were all largely cupy or numba-ized during the hackathon. Some cpu parts remain inside cache_spots and evalcoeffs.

Additional functions that should be gpu-ized: