[论文解读] GX: a GPU-native gyrokinetic turbulence code for tokamak and stellarator design

Noah Mandell, W. Dorland+5 more|arXiv (Cornell University)|Sep 14, 2022

Magnetic confinement fusion research被引用 4

一句话总结

GX 是一种原生 GPU 的磁流体动力学湍流代码，专为在托卡马克和 stellarator 中实现快速、高保真的低频湍流模拟而设计，采用伪谱拉盖尔-埃尔米特速度空间格式和径向通量管几何结构，可在单个 GPU 上实现分钟级模拟。它在线性和非线性基准测试中均实现了高性能和精确的收敛性，适用于聚变反应堆设计与优化工作流程。

ABSTRACT

GX is a code for solving the nonlinear gyrokinetic system for low-frequency turbulence in magnetized plasmas, particularly tokamaks and stellarators. In GX, our primary motivation and target is a fast gyrokinetic solver that can be used for fusion reactor design and optimization along with wide-ranging physics exploration. This has led to several code and algorithm design decisions, specifically chosen to prioritize time to solution. First, we have used a discretization algorithm that is pseudo-spectral in the entire phase-space, including a Laguerre-Hermite pseudo-spectral formulation of velocity space, which allows for smooth interpolation between coarse gyrofluid-like resolutions and finer conventional gyrokinetic resolutions and efficient evaluation of a model collision operator. Additionally, we have built GX to natively target graphics processors (GPUs), which are among the fastest computational platforms available today. Finally, we have taken advantage of the reactor-relevant limit of small $ρ_*$ by using the radially-local flux-tube approach. In this paper we present details about the gyrokinetic system and the numerical algorithms used in GX to solve the system. We then present several numerical benchmarks against established gyrokinetic codes in both tokamak and stellarator magnetic geometries to verify that GX correctly simulates gyrokinetic turbulence in the small $ρ_*$ limit. Moreover, we show that the convergence properties of the Laguerre-Hermite spectral velocity formulation are quite favorable for nonlinear problems of interest. Coupled with GPU acceleration, which we also investigate with scaling studies, this enables GX to be able to produce useful turbulence simulations in minutes on one (or a few) GPUs. GX is open-source software that is ready for fusion reactor design studies.

研究动机与目标

• 开发一种快速、针对 GPU 优化的磁流体动力学求解器，用于聚变反应堆设计与优化。
• 通过优先考虑求解时间而非完整物理解释，实现快速模拟周转。
• 通过伪谱拉盖尔-埃尔米特速度空间格式，支持从磁流体动力学类到高保真磁流体动力学模拟的平滑分辨率过渡。
• 通过既定基准测试，验证 GX 在托卡马克和 stellarator 构型下的准确性与性能。
• 提供一个开源、可投入生产的工具，用于反应堆相关的湍流研究。

提出的方法

• 在全相空间中采用伪谱离散化，包括使用拉盖尔-埃尔米特谱基进行速度空间离散，以实现平滑的分辨率缩放。
• 采用径向局部通量管近似，利用小 ρ∗ 极限（ρi/R ≪ 1），简化几何结构并提升计算效率。
• 原生支持 GPU，使用 CUDA/ROCm，算法设计选择针对大规模并行计算和内存合并进行了优化。
• 实现超黏性和超碰撞项以稳定非线性模拟，同时保持谱收敛性。
• 采用与谱速度空间格式兼容的模型碰撞算子，实现高效计算。
• 支持玻尔兹曼和粒子动力学电子模型，具有统一的归一化和通量缩放。

实验结果

研究问题

• RQ1 原生 GPU 的磁流体动力学代码能否在极低延迟下实现托卡马克和 stellarator 中湍流的高性能模拟？
• RQ2 与传统网格方法相比，拉盖尔-埃尔米特谱速度空间格式在非线性湍流问题中的表现如何？
• RQ3 GPU 加速在保持准确性的同时，能在多大程度上缩短反应堆相关构型的模拟时间？
• RQ4 GX 在线性和非线性基准测试中，对 GS2 和 stella 代码的既定结果再现程度如何？
• RQ5 GX 是否能实现快速、高保真的湍流模拟，适用于聚变反应堆设计与优化工作流？

主要发现

• 在单个 GPU 上，GX 对于低分辨率情形可在几分钟内完成模拟，使用多个 GPU 时高保真结果也仅需数小时。
• 拉盖尔-埃尔米特谱速度空间格式在非线性湍流问题中表现出优异的收敛特性。
• 在托卡马克和 stellarator 几何结构下的线性和非线性基准测试中，GX 结果与 GS2 和 stella 代码高度一致。
• 在 W7-X stellarator 情形下，GX 使用 128 个平行网格点和 192 个法向网格点，成功再现了与参考模拟相当的非线性湍流水平。
• 缩放研究证实了强大的 GPU 缩放效率，性能提升与 GPU 数量成正比，并具备最优的内存访问模式。
• GX 是开源且可投入生产的，全面支持反应堆规模的设计研究和参数空间探索。