[Paper] 可扩展的网格耦合用于大气波模拟

Source: arXiv - 2603.02971v1

概述

本文提出了一种可扩展的网格耦合算法，能够让两个独立的大气求解器在实时情况下跨重叠计算网格交换数据。通过实现对原位耦合波动模拟的全局一致解，该技术为能够在现代超级计算机上高效运行的更高保真度的天气和气候模型打开了大门。

关键贡献

• 通用重叠网格插值，可用于任何共享相同物理域的求解器对。
• 可扩展实现 基于 p4est 库，能够在数万 MPI 进程上实现接近线性的强缩放。
• 稳健处理非一致网格，包括一侧自适应细化而另一侧保持粗糙的情况。
• 性能评估 在真实的大气波案例中进行，显示相较于独立运行求解器，开销低于 5 %。
• 开源参考实现（MIT‑licensed），可嵌入现有的 CFD/大气代码中。

Methodology

1. Domain decomposition – 每个求解器拥有自己的八叉树（或四叉树）森林，用于离散化大气。两个森林在“耦合区”重叠。
2. Ghost‑layer construction – 通过一次轻量级的 MPI 通信步骤，构建共享的界面单元列表，且不受网格分辨率的影响。
3. Interpolation kernel – 对于重叠区域的每个单元，算法寻找来自另一网格的供体单元并执行高阶多项式插值。该核设计为极易并行：每个 MPI 进程只处理本地子集的重叠单元。
4. Load‑balanced communication – 利用 p4est 的空间填充曲线排序，作者重新分配重叠数据，以在某一网格高度细化时仍保持通信负载平衡。
5. In‑situ coupling loop – 两个求解器推进一个时间步，调用插值例程，然后使用更新后的边界值继续计算，整个过程无需写入中间文件。

结果与发现

• 强伸缩性：在 NERSC Perlmutter 系统上，耦合例程在 256 到 32 k MPI 进程之间保持了 > 80 % 的并行效率。
• 弱伸缩性：向重叠区域添加更多单元导致运行时间线性增长，证实了算法的 O(N) 复杂度。
• 精度：相对于参考解的基准测试表明，即使网格之间存在 4× 分辨率不匹配，插值误差仍保持在波幅的 1 % 以下。
• 开销：耦合步骤仅增加了总模拟时间的 3–5 %，远低于传统基于检查点的耦合的 I/O 成本。

实际意义

• Hybrid modeling pipelines – 气象机构现在可以在单个作业中同时运行高分辨率中尺度模型和粗糙的全球模型，保留细尺度特征而无需昂贵的数据调度。
• Adaptive refinement on‑the‑fly – 开发者可以在感兴趣的区域（例如雷暴）触发网格细化，而周围域保持粗糙，耦合算法会自动处理数据交换。
• Multi‑physics extensions – 同一框架可重复用于将大气动力学与海洋、陆面或化学输送模型耦合，加速地球系统模拟的开发。
• Reduced storage costs – 由于耦合在现场进行，无需将大量中间场写入磁盘，从而降低存储需求并缩短后处理时间。

限制与未来工作

• 假设结构化网格求解器 – 当前实现依赖于八叉树/四叉树基网格；要扩展到完全非结构化的有限元网格需要额外的映射逻辑。
• 单向插值 – 本文聚焦于单向数据传输；双向耦合（例如反馈回路）尚未进行基准测试。
• 物理无关 – 虽然算法在数学上是可靠的，但耦合复杂源项（如湿对流）可能需要定制的插值模板。
• 未来方向 包括为 GPU 加速求解器添加支持，探索更高阶的保守插值方案，以及集成由耦合误差本身驱动的误差控制自适应细化。

作者

• Hannes Brandt
• Tim Griesbach
• Matthew Zettergren
• Scott Aiton
• Jonathan Snively
• Donna Calhoun
• Carsten Burstedde

论文信息

• arXiv ID: 2603.02971v1
• 分类: cs.DC, cs.CE
• 出版时间: 2026年3月3日
• PDF: 下载 PDF