[論文レビュー] 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 ネイティブなギャロアクイネティクス乱流コードであり、擬スペクトル的ラゲル・エルミート速度空間形式と径方向フラックスチューブ幾何学を用いて、1 枚の 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 の正確性と性能をトカマクおよびステラレーター構成において検証すること。
• 炉関連の乱流研究に適したオープンソースでプロダクション運用可能なツールを提供すること。

提案手法

• 全位相空間における擬スペクトル的離散化を採用し、速度空間にラゲル・エルミートスペクトル基底を用いて、滑らかな解像度スケーリングを可能にする。
• 径方向局所的フラックスチューブ近似を採用し、ρ∗ が小さい（ρi/R ≪ 1）という条件を活用して、幾何学を単純化し、計算効率を向上させる。
• CUDA/ROCm をネイティブにサポートし、大規模並列処理とメモリコalescing を最適化したアルゴリズム設計を採用する。
• 非線形シミュレーションの安定化を図りながらスペクトル収束性を保持するため、ハイパービスクオシティおよびハイパーカリオンを実装する。
• スペクトル的速度空間形式と互換性のあるモデル衝突演算子を実装し、計算効率を高める。
• ボルツマンおよびキネティック電子モデルの両方をサポートし、正規化およびフラックススケーリングを一貫させる。

実験結果

リサーチクエスチョン

• RQ1 GPU ネイティブなギャロアクイネティクスコードは、最小限のレイテンシでトカマクおよびステラレーターにおける乱流の高性能シミュレーションを達成できるか？
• RQ2 ラゲル・エルミートスペクトル的速度空間形式は、従来のグリッドベースの方法と比較して、非線形乱流問題においてどのように性能を発揮するか？
• RQ3 GPU アクcelレーションは、炉関連の構成において正確性を維持したまま、シミュレーション時間をどの程度短縮できるか？
• RQ4 線形および非線形ベンチマークにおいて、GX は GS2 および stella コードの既存結果をどの程度正確に再現できるか？
• RQ5 GX は、核融合炉の設計および最適化パイプラインに適した、迅速で高精度な乱流シミュレーションを可能にするか？

主な発見

• 低解像度のケースでは 1 枚の GPU で数分のシミュレーションターンアラウンドを達成し、複数の GPU を用いることで高精度な結果が数時間で得られる。
• ラゲル・エルミートスペクトル的速度空間形式は、非線形乱流問題において良好な収束特性を示す。
• トカマクおよびステラレーターの両幾何形状における線形および非線形ケースのベンチマーク結果は、GS2 および stella コードと優れた一致を示す。
• W7-X ステラレーターのケースでは、128 個の並列グリッドポイントおよび 192 個のバイノーマルグリッドポイントを用いて、参照シミュレーションと同等の非線形乱流レベルを正確に再現した。
• スケーリングスタディーにより、強力な GPU スケーリング効率が確認され、パフォーマンス向上は GPU 数に比例し、最適なメモリアクセスパターンが得られている。
• GX はオープンソースであり、プロダクション運用可能であり、炉規模の設計研究およびパラメータ空間の探索に完全に対応している。