基于GPU的混合精度平方根共轭梯度算法.pdfVIP

学兔兔 第33卷 第1期 仪 器 仪 表 学 报 V0l_33 No．1 2012年1月 Chinese Journal of Scientific Instrument Jan．2012 基于GPU的混合精度平方根共轭梯度算法：{： 李熙铭 ，欧阳丹彤 ，白洪涛 ， ， (1吉林大学计算机科学与技术学院 长春 130012；2吉林大学符号计算与知识工程教育部重点实验室 长春 130012； 3吉林大学公共计算机教学与研究中心 长春 130012；4吉林大学地球探测科学与技术学院 长春 130026) 摘 要：针对当前基于GPU的数值算法具有双精度数据性能低下的缺陷。提出了一种适于GPU统一计算架构Fermi—CUDA的 混合精度平方根共轭梯度算法用以求解稀疏线性方程组。该算法采用单精度内迭代与双精度外迭代结合的方法，以充分利用 GPU体系结构下单精度高性能和双精度高精度的优点。整个算法的计算部分完全在GPU端进行，减少了CPU和GPU之间的 数据通信。实现了基于GPU的平方根共轭梯度法、Jacobi迭代法和Gauss—Seidel迭代法，分析它们作为内迭代算子对算法收敛 性的影响。实验表明，该算法获得了与全双精度数据处理等同的计算精度，比GPU全双精度在浮点性能上提升近一倍，相对于 CPU全双精度串行算法，最大加速比达到70以上。 关键词：线性方程组；平方根共轭梯度算法；内外迭代子；混合精度；图形处理器 中图分类号：TP311．1 文献标识码：A 国家标准学科分类代码：520．3020 Mixed precision CGS algorithm based on GPU Li Ximing ，Ouyang Dantong ，Bai Hongtao ，。’ ( College of Computer Science and Technology，Jilin University，Changchun 130012，China； 2 Key Laboratory ofSymbolic Computation and Knowledge Engineering ofMinistry ofEducation，Jilin Unwenity， Changchun 130012，China；3 Centerfor Computer Fundamental Education，Jilin University，Changchun，130012，China； 4 College of Earth Survey Science and Technology，Jilin University，Changchun 130026，China) Abstract：GPU-based numerical algorithms have the shortcoming of low performance for double precision．We sug— gest a mixed precision conjugate gradient squared algorithm suitable for the GPU of Fermi--CUDA to solve sparse line·- ar equations．The scheme uses a combination of single—precision inner iteration and double—precision outer iteration to take the advantages of efficient single—-precision operation and accurate double·-precision operation under the GPU structure