Skip to main content
QUICK REVIEW

[论文解读] Accelerating the solution of families of shifted linear systems with CUDA

Richard Galvez, Greg van Anders|arXiv (Cornell University)|Feb 10, 2011
Particle physics theoretical and experimental studies参考文献 15被引用 11
一句话总结

本文提出了一种基于GPU加速的开源CUDA实现,用于求解具有多个偏移量的稀疏线性系统族的移位Krylov子空间求解器——具体为CG-M和BiCGStab-M方法。该方法相比单核CPU实现了超过12倍的保守加速,使得复杂量子场论的实用模拟成为可能,包括与大型强子对撞机(LHC)物理和超对称模型相关的理论。

ABSTRACT

We describe the GPU implementation of shifted or multimass iterative solvers for sparse linear systems of the sort encountered in lattice gauge theory. We provide a generic tool that can be used by those without GPU programming experience to accelerate the simulation of a wide array of theories. We stress genericity, which is important to allow the simulation of candidate theories for new physics at LHC, and for the study of various supersymmetric theories. We find significant speed ups, which we conservatively bound below at at least twelve times, that promise to put a variety of research questions within practical reach.

研究动机与目标

  • 开发一种通用的、基于GPU加速的求解器,用于求解格点规范理论和量子场论模拟中出现的移位线性系统族。
  • 在不从零重写核心求解器的前提下,实现对非QCD理论(如大型强子对撞机(LHC)候选新物理和超对称模型)的高效模拟。
  • 提供一个用户友好、开源的工具,无需CUDA编程知识,从而在高能物理和计算物理领域实现广泛采用。
  • 显著降低求解多个移位系统所带来的计算成本,而这是动力费米子模拟中的主要性能瓶颈。

提出的方法

  • 求解器采用GPU加速的多质量共轭梯度(CG-M)和多质量双共轭梯度稳定化(BiCGStab-M)方法,用于求解移位线性系统。
  • 利用数学性质:Krylov子空间在矩阵偏移下保持不变,从而允许在多个移位系统之间共享计算。
  • 通过在不同偏移量之间复用迭代常数,避免冗余的矩阵-向量乘法,最大限度减少计算开销。
  • 实现基于CUSP库构建,暴露高层C++接口,隐藏底层GPU编程细节。
  • 支持单精度和双精度算术运算,性能在不同数量的偏移量下进行测量。
  • 该方法具有通用性,可应用于任意形式为(A + σiI)xi = bi的问题,包括非理论应用,如求解具有多个右端项的二维Poisson方程。

实验结果

研究问题

  • RQ1GPU加速的Krylov子空间方法能否在格点规范理论模拟中,对求解移位线性系统族实现显著的性能提升?
  • RQ2通用的、开源的GPU求解器在多大程度上能降低非阿贝尔规范理论(含动力费米子)模拟的计算成本?
  • RQ3对于相同问题规模和精度,基于GPU的移位求解器与单核CPU实现相比性能如何?
  • RQ4该求解器是否可有效应用于量子色动力学(QCD)之外的场景,例如求解具有多个右端项的二维Poisson方程?
  • RQ5通过使用能保持共享Krylov子空间的智能初始猜测,是否可行将求解器扩展至处理具有不同右端项的系统?

主要发现

  • 在双精度下求解15个移位系统时,GPU加速的CG-M求解器相比单核CPU实现了超过12倍的保守加速。
  • 对于1000×1000的二维Poisson问题,包含15个移位量,基于GPU的移位求解器耗时不足单核CPU运行非移位CG求解器的1/12。
  • 由于算法开销减少,GPU实现的非移位CG求解器比移位求解器快约几倍,这符合预期。
  • 该求解器设计为通用且可重用,可在无需底层GPU编程的情况下,快速模拟多种量子场论。
  • 该方法的应用范围不仅限于高能物理,如其在求解具有多个移位量的二维Poisson方程中已成功验证。
  • 代码的开源特性与CUSP项目集成,确保了广泛可及性与未来研究的可扩展性。

更好的研究,从现在开始

从阅读论文到最终审阅,大幅缩短您的研究时间。

无需绑定信用卡

本解读由 AI 生成,并经人工编辑审核。