AMGX GPU 加速代数多重网格求解器新

背景

压力-Poisson 与隐式推进主导了不可压 CFD 与耦合 FEM 的开销。代数多重网格（AMG）是最强的前置条件器家族，但 CPU 实现难以发挥现代加速器的细粒度并行性。

AMGX 提供什么

• 多 GPU 节点上的分布式 AMG 求解器与前置条件器
• 经典 Ruge-Stueben 与聚合式 AMG 变体
• 清晰的集成路径：经 PETSc 等驱动框架调用

与物理 AI 的关系

• 基础设施共用：训练代理模型的同一 GPU 集群，其经典求解器内部也跑 AMGX 内核
• 可微分求解器研究建立在快速前置线性代数之上
• 混合环参照：ML 提议场，AMGX 完成投影

适用场景

• GPU 原生 CFD 代码替换 CPU 压力求解器
• 隐式耦合步的耦合多物理求解器
• 教学：以 GPU 性能学习 AMG 结构

资源

• AMGX 仓库（含示例与 PETSc 集成说明）
• NVIDIA 开发者博客的 AMGX 与 GPU 线性代数栈文章