[논문 리뷰] GX: a GPU-native gyrokinetic turbulence code for tokamak and stellarator design

Noah Mandell, W. Dorland외 5명|arXiv (Cornell University)|2022. 09. 14.

Magnetic confinement fusion research인용 수 4

한 줄 요약

GX는 토카막과 스타렐레이터에서 저주파수 난류의 빠르고 고정밀 시뮬레이션을 위한 GPU 네이티브 기반의 기하학적 기반 난류 코드로, 가상의 스펙트럼 기반 라게르-에르미트 속도 공간 설정과 반경 방향의 플럭스튜브 기하학을 사용하여 단일 GPU에서 분 단위의 시뮬레이션을 가능하게 한다. 선형 및 비선형 기준 테스트에서 높은 성능와 정확한 수렴성을 확보하여 fusion 반응로 설계 및 최적화 워크플로우에 적합하다.

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를 활용한 GPU 네이티브 지원을 통해 대량 병렬 처리 및 메모리 코alescing에 최적화된 알고리즘 설계
• 비선형 시뮬레이션의 안정성을 확보하면서도 스펙트럼 수렴성을 유지하기 위해 초점성 점성 및 초점성 충돌 항목 구현
• 스펙트럼 속도 공간 설정과 호환되는 모델 충돌 연산자 구현으로 효율적 계산 지원
• 볼츠만 및 킬린 전자 모델 모두 지원하며, 정규화 및 플럭스 스케일링에서 일관성 확보

실험 결과

연구 질문

• RQ1 GPU 네이티브 기하학적 기반 난류 코드는 낮은 지연 시간으로 토카막과 스타렐레이터에서 난류 시뮬레이션을 고성능으로 수행할 수 있는가?
• RQ2 기존 그리드 기반 방법과 비교해 복잡한 비선형 난류 문제에서 라게르-에르미트 스펙트럼 속도 공간 설정은 어떤 성능을 보이는가?
• RQ3 GPU 가속화가 반응로 관련 기하학에서 정확성을 유지하면서도 시뮬레이션 시간을 얼마나 줄이는가?
• RQ4 선형 및 비선형 기준 테스트에서 GX가 GS2 및 stella 코드의 기존 결과를 얼마나 잘 재현하는가?
• RQ5 GX는 융합 반응로 설계 및 최적화 워크플로우에 적합한 고정밀 난류 시뮬레이션을 분당으로 수행할 수 있는가?

주요 결과

• 낮은 해상도 사례에서는 단일 GPU에서 분 단위로 시뮬레이션 완료 시간 확보하며, 다수 GPU를 사용해 몇 시간 내에 고정밀 결과 도출
• 비선형 난류 문제에서 라게르-에르미트 스펙트럼 속도 공간 설정이 유리한 수렴 특성 보여
• 토카막 및 스타렐레이터 기하학에서 선형 및 비선형 사례의 기준 테스트 결과는 GS2 및 stella 코드와 우수한 일치
• W7-X 스타렐레이터 사례에서 128개의 평행 그리드 포인트와 192개의 이심 방향 그리드 포인트를 사용해 기준 시뮬레이션과 유사한 비선형 난류 수준 재현 성공
• 스케일링 연구 결과 강력한 GPU 스케일링 효율성 확인, 성능 향상은 GPU 수와 비례하며 최적의 메모리 액세스 패턴 확보
• GX는 오픈소스이며 프로덕션 수준으로 사용 가능하며, 반응로 규모의 설계 연구 및 매개변수 공간 탐색에 대한 완전한 지원 제공