NVIDIA cuPhoton 是一套开源 CUDA-X 工具包,旨在解决高吞吐科学仪器面临的图像处理瓶颈。其核心思路是让图像数据从传感器读取到分类的整个流程始终保留在 GPU 上,避免 CPU 与 GPU 之间的反复传输。该工具包提供多个 GPU 加速模块,涵盖 FITS 数据直接加载、图像对齐、PSF 匹配与减除、偶极拟合以及候选目标审查等环节。在代表性多 GPU 工作负载中,cuPhoton 将图像加载速度提升最高约 14,900 倍,信号处理速度提升最高约 14,550 倍,相较 x86 CPU 基线效果显著。以 Vera C. Rubin 天文台的 LSSTCam 为例,其每 39 秒产生 3.2 吉像素曝光,每晚生成最高 20 TB 图像和 1000 万个候选目标,cuPhoton 可在 60 至 120 秒的警报窗口内完成处理,并支持跨多 GPU、多节点系统扩展。
📱 每天一份 AI 前沿日报
关注公众号,每天 09:00 推送 · 不错过任何重磅