基于JASMIN框架多物理耦合程序的性能优化及分析.pdfVIP

第３２卷 第４期 计　 　 算　 　 物　 　 理 Ｖｏｌ．３２，Ｎｏ．４ ２０１５年７月 　 ＣＨＩＮＥＳＥ ＪＯＵＲＮＡＬ ＯＦ ＣＯＭＰＵＴＡＴＩＯＮＡＬ ＰＨＹＳＩＣＳ　 Ｊｕｌ． ，２０１５ 文章编号：１００１⁃２４６Ｘ（２０１５）０４⁃０４３１⁃０６ 基于ＪＡＳＭＩＮ框架多物理耦合程序的 性能优化及分析 任　 健，　 武林平，　 申卫东 （北京应用物理与计算数学研究所，北京　 １０００９４） 摘　 要：基于并行应用支撑软件框架ＪＡＳＭＩＮ的辐射流体与粒子输运耦合程序ＲＨＳｎ２Ｄ，采用最小邦元固定处理器 数目的并行策略，计算实际模型的并行规模扩展至８ １９２ 核，并行效率约为１６％．集成程序时间分析，验证软件框 架底层ＭＰＩ并行环境聚合通信对于并行优化算法（尤其是辐射流体计算时间）的影响． 关键词：多物理耦合；并行计算；性能优化；ＲＨＳｎ２Ｄ程序；并行软件支撑框架ＪＡＳＭＩＮ 中图分类号：ＴＰ３０１ 文献标志码：Ａ ０　 引言 惯性约束聚变（Ｉｎｅｒｔｉａｌ Ｃｏｎｆｉｎｅｍｅｎｔ Ｆｕｓｉｏｎ，ＩＣＦ）复杂物理问题的数值模拟中，涉及多个物理过程的耦合 求解，主要包括流体力学、辐射传输（扩散近似或者输运描述）、核反应动力学、粒子输运和激光传输等［１－２］． 由于描述不同物理过程的微分方程其数学属性差异较大，其数值计算方法大相径庭．例如，在ＩＣＦ数值模拟 ［２］ ［３］ 研究中的ＬＡＲＥＤ系列 及其集成程序 ，流体力学过程采用相容拉格朗日方法，模拟辐射多群扩散过程采 用源迭代方法，其中二维扩散算子采用九点差分格式，模拟辐射多群输运过程则采用离散纵坐标（Ｓｎ）方法， 而且不同物理过程描述的计算区域也各不相同． 面对实际应用问题中复杂物理过程的大规模并行计算，如何设计多物理耦合程序的并行策略，获得较好 的整体并行效率及扩展度，成为激光聚变应用程序研究中的重要课题［１－２］．基于并行应用支撑软件框架（Ｊ Ａｄａｐｔｉｖｅ Ｓｔｒｕｃｔｕｒｅｄ Ｍｅｓｈ ａｐｐｌｉｃａｔｉｏｎｓ Ｉｎｆｒａｓｔｒｕｃｔｕｒｅ，简称ＪＡＳＭＩＮ）的联邦计算［４－６］ ，ＲＨＳｎ２Ｄ程序实现了辐射 ［７］ 流体与粒子输运的多物理耦合并行计算，采用１０２４个处理器可用达到３６％的并行效率 ．但是，由于实际 模型网格数目较少，ＲＨＳｎ２Ｄ程序并行扩展性受限（最大规模仅为１０２４ 核 ＣＰＵ），而且随着 ＣＰＵ核数的增 ［７］ 加，辐射流体计算的并行效率下降明显，不利于整体并行效率的提高 ． 针对粒子输运计算量远大于辐射流体的特点，本文采用ＪＡＳＭＩＮ框架“最小邦元固定处理器数目”的并 行策略，改进ＲＨＳｎ２Ｄ程序的多物理耦合并行计算，从而控制辐射流体的计算时间，挖掘输运计算的并行扩 展度．典型算例表明，在相同计算条件下，ＲＨＳｎ２Ｄ程序并行规模由此前的１０２４ 核扩展至８ １９２ 核，并行效 率（相对２５６核）约为１６％．而且，通过分析集成程序的计算时间，验证了软件框架底层ＭＰＩ并行环境聚合通 信对于并行优化算法（尤其是辐射流体计算时间）的影响，可供ＪＡＳＭＩＮ框架下多物理耦合应用程序优化与 性能分析参考． １　 多物理耦合方程及其计算方法 辐射流体与粒子输运多物理耦合程序ＲＨＳｎ２Ｄ，其控制方程包括二维柱坐标下的流体力学方程组、多群 非定态粒子输运方程，在给定物质状态方程（ｅｑｕａｔｉｏｎ ｏｆ ｓｔａｔｅ，ＥＯＳ）与计算区域初边值条件下，多物理耦合控 制方程封闭可解． 收稿日期：２０１４ －０６－２５；修回日期：２０１４ －１０－１３ 基金项目：国家自然科学基金资助项目 作者简介：任健（１９７８－），男，硕士，副研究员，从事辐射流体计算方法与数值模拟研究，Ｅ⁃ｍａｉｌ：ｒｅｎ＿ｊｉａｎ＠ｉａｐｃｍ．ａｃ．ｃｎ ４３２