求解大规模稀疏线性代数方程组序列的自适应amg预条件策略.pdf

求求解解大大规规模模稀稀疏疏线线性性代代数数方方程程组组序序列列的的自自适适应应AMG预预条条件件策策略略 徐小文, 莫则尧 and 恒斌 Citation: 中国科学: 信息科学 ; doi: 10.1360/N112016-00074 View online: http://engine.scichina.co /doi/10.1360/N112016-00074 Published by the 《中国科学》杂志社 Articles you may be interested in 求解两类Maxwell方程组棱元离散系统的快速算法和 自适应方法 中国科学: 数学 43, 333 (2013); 关于单取代烯类聚合物中键接异构序列的研究 中国科学B辑: 化学 生命科学 地学 19, 121 (1989); 受限Morgan问题 * 中国科学A辑: 数学 26, 513 (1996); 聚合反应动力学的图形方法 中国科学 11, 39 (1981); 任意初始点下的序列线性方程组方法 中国科学A辑: 数学 27, 24 (1997); 中国科学: 信息科学 2016 年 第46 卷 第10 期: 1411–1420 SCIENTIA SINICA Informationis 高性能科学计算若干前沿问题研究专刊 求解大规模稀疏线性代数方程组序列的自适应 预条件策略 * 徐小文 莫则尧 安恒斌 北京应用物理与计算数学研究所计算物理重点实验室, 北京 100094 中物院高性能数值模拟软件中心, 北京 100086 * 通信作者. E-mail: xwxu@ 收稿日期: 2016–03–20; 接受日期: 2016–07–07; 网络出版日期: 2016–10–25 国家重点基础研究发展计划(973) (批准号: 2011CB309702) 和国家自然科学基金 (批准号: 资助 项目 摘要 时间相关偏微分方程隐式离散后, 通常需要求解一个稀疏线性代数方程组序列. 利用序列中 相邻方程组性质的差异性与相似性, 自适应地选取预条件子, 提升方程组序列的并行求解效率, 从而 缩短总体求解时间, 是一个值得研究的问题. 本文针对科学与工程计算中广泛使用的代数多重网格 (AMG) 预条件子, 设计了方程组序列相关的自适应预条件策略. 通过惯性约束聚变(ICF) 的辐射流体 力学数值模拟典型应用, 验证了该策略的有效性. 测试结果表明, 在某高性能计算机的3125 个CPU 核上, 自适应预条件策略可将并行效率从47% 提升到61%, 将模拟总时间从19.7 h 降为14.5 h. 关键词 稀疏线性解法器 迭代方法 预条件子 代数多重网格算法(AMG) 并行计算 引言 在基于时间相关的偏微分方程 (PDE) 数值模拟中, 如果时间方向采用隐格式离散, 随着时间步进 以及非线性迭代向前推进, 需要求解一个稀疏线性代数方程组序列: { = = 1 } (1) 其中 , , 分别为第 个方程组的稀疏矩阵、未知向量和右端向量, 为该序列中方程组数目. 在很多数值模拟中, 典型应用的一次完整模拟所需求解的方程组数目高达数万、数十万, 甚至数百万, 求解该方程组序列的CPU 时间占总时间的80% 以上, 是主要性能瓶颈. 受计算和存储复杂度的限制, 迭代方法是当前大规模计算中求解该类方程组的主流方法. 迭代方 法的收敛速度往往受限于问题规模和应用特征, 需要采用预条件技术(preconditioning) 加速迭代收敛. 对于由偏微分方程离散得到的稀疏线性代数方程组, 代数多重网格 (AMG) 算法 是目前实