NVIDIA 称 cuPhoton.xBeam 在 64 块 GPU 上 70 秒处理完 10 TB 黑洞望远镜数据,CPU 基线要 8.89 小时——约 460 倍加速
NVIDIA 称,其 cuPhoton 工具包中的 xBeam 相关器在 NVIDIA Vera Rubin NVL72 系统上使用 64 块 GPU 与 GPUDirect Storage 兼容模式,70 秒处理完一个 10 TB 数据集,而 x86 CPU 基线耗时 31,992 秒(8.89 小时),约快 460 倍。这个数字描述的是射电干涉测量中”相关”这一步的端到端处理时间加速比。xBeam 是 cuPhoton 工具包里的 FX 相关器,用于将各望远镜站点记录的电压流处理成”可见度”——每对站点、每个频率通道、每个短时间区间对应一个复数值,作为后续校准和图像重建的输入。
这个数字怎么来的
据 NVIDIA 技术博客,约 460 倍这一加速比的测量口径是”总处理时间”(total processing time),涵盖读取数据、准备数据和相关计算的端到端全过程。硬件条件为 Vera Rubin NVL72 系统上的 64 块 GPU,数据通过 GPUDirect Storage(GDS)兼容模式读入 GPU 显存。被处理的对象是一份 10 TB 的望远镜电压记录。对照基线是 x86 CPU,耗时 31,992 秒。GPU 端耗时 70 秒,由此得出约 460 倍的比值。
NVIDIA 称,xBeam 的 FX 流程依次完成五步:读取 VDIF 帧并构建索引文件;根据作业文件中的延迟模型对每个站点做延迟校正和条纹旋转;对每个站点的电压流做 FFT 拆分为频率通道;对每对站点将频谱互相关并在积分时间内累加;将结果写为 DiFX 的 SWIN 格式或 xBeam 自有格式。xBeam 不自行计算延迟,而是使用作业文件提供的延迟多项式;它也不追踪黑洞附近的光线路径。
NVIDIA 称,扩展方式是按时间切分观测数据:在单节点上用 --devices 参数让多块 GPU 各自处理一段时间片段;跨节点则用 --shard-count 和 --shard-index 分配分片,最后由 merge-shards 工具将各分片的 SWIN 输出合并为完整的可见度数据集。64 块 GPU 即对应 --shard-count 64,索引从 0 到 63。博客还提到,在同一份 10 TB 记录上,GPUDirect Storage 的读取带宽随 GPU 数量增长,从 4 块 GPU 时的 20 GB/s 提升到 64 块 GPU 时的 170 GB/s。
和什么比
据 NVIDIA 技术博客,对照基线是 x86 CPU,耗时 31,992 秒(约 8.89 小时)。该对照基线的 CPU 型号、核心数及软件版本均未在测试条件中列出。NVIDIA 称 DiFX 是用作验证 xBeam 结果的 CPU 相关器,下游校准和成像工具(如 HOPS、rPICARD、eht-imaging)从相关器产出的可见度开始工作。
约 460 倍说明了什么、没说明什么
据 NVIDIA 技术博客,从 1 块 GPU 扩展到 64 块 GPU 的过程中,加速比持续增长,博客原文表述为”在测试范围内加速比继续增加”。这意味着在已测试的 GPU 数量区间内尚未出现明显的扩展瓶颈。约 460 倍这个数字覆盖的是读取、准备和相关计算的完整流程,而非单独的计算内核加速。NVIDIA 称,xBeam 产出的可见度是下游校准和图像重建的输入,校准与成像不在 xBeam 的处理范围内。此外,该加速比的测量条件包含 GPUDirect Storage 兼容模式,读取带宽随 GPU 数量增长这一特征本身也是端到端时间缩短的组成部分,而非纯计算加速。
主题分类:科学计算/天文
参考资料
- Run GPU-Powered Black Hole Interferometry with NVIDIA cuPhoton — NVIDIA Developer Blog, 2026-11-13