CATAPULT: A CUDA-Accelerated Timestepper for Alpha Particles Using Local Tricubics

Le papier présente CATAPULT, un intégrateur temporel accéléré par CUDA pour le confinement des particules alpha dans les stellarators, qui offre des performances supérieures aux implémentations CPU existantes tout en gérant les champs magnétiques d'équilibre et les ondes d'Alfvén.

L'essentiel

🚀 CATAPULT : Le Super-Héros de la Fusion Nucléaire

Imaginez que vous essayez de construire une mini-étoile sur Terre (c'est ce qu'on appelle la fusion nucléaire) pour produire une énergie propre et infinie. Le problème ? Pour que cette étoile fonctionne, il faut garder des particules ultra-chaudes (des ions) emprisonnées dans un piège magnétique invisible.

Dans les réacteurs de type "stellarator" (une forme de réacteur très complexe), ces particules sont comme des balles de tennis lancées à une vitesse folle dans un labyrinthe magnétique. Si elles touchent les murs, elles perdent leur énergie et le réacteur s'éteint.

Les scientifiques doivent simuler des millions de ces balles pour voir si elles restent dans le jeu ou si elles s'échappent. C'est là qu'intervient CATAPULT.

1. Le Problème : Un Calcul Trop Lent

Avant CATAPULT, les scientifiques utilisaient des ordinateurs classiques (des processeurs CPU) pour simuler le trajet de ces particules.

• L'analogie : Imaginez que vous devez vérifier si un million de personnes peuvent traverser une ville sans se perdre. Avec les anciens ordinateurs, c'est comme si vous aviez 128 détectives qui marchent lentement, un par un, pour vérifier chaque itinéraire. C'est précis, mais cela prend des jours, voire des semaines.

2. La Solution : CATAPULT (Le GPU)

Les auteurs ont créé un nouveau logiciel appelé CATAPULT. Il est conçu pour fonctionner sur des cartes graphiques (GPU), comme celles utilisées pour les jeux vidéo ultra-réalistes.

• L'analogie : Au lieu d'avoir 128 détectives, CATAPULT utilise une armée de 10 000 petits robots qui travaillent tous en même temps.
• Le résultat : Là où l'ancien système prenait 10 heures, CATAPULT le fait en quelques minutes. C'est 5 à 60 fois plus rapide !

3. Comment ça marche ? (La Magie des "Tricubics")

Pour prédire où ira une particule, l'ordinateur doit connaître la forme du champ magnétique (le "labyrinthe").

• L'ancien système : Il regardait le labyrinthe par petits morceaux, comme si on lisait une carte avec des cases carrées. C'était un peu grossier.
• CATAPULT : Il utilise une technique appelée "Tricubics". Imaginez que vous avez une carte papier, mais au lieu de la voir comme une grille rigide, vous la transformez en une pâte à modeler lisse et flexible.
  • Quand une particule bouge, CATAPULT "lisse" la route devant elle en temps réel, même si elle passe d'une zone à l'autre. Cela permet de suivre les balles avec une précision extrême, même quand elles font des virages serrés ou qu'elles s'approchent des bords du piège.

4. Pourquoi c'est génial ?

Grâce à cette vitesse fulgurante, les scientifiques peuvent faire des choses qu'ils ne pouvaient pas faire avant :

1. Plus de précision : Ils peuvent simuler des millions de particules au lieu de quelques milliers. C'est comme passer d'une photo floue à une image 4K ultra-nette.
2. Des scénarios complexes : Ils peuvent maintenant ajouter des "vagues" dans le champ magnétique (comme des vagues dans l'océan) pour voir comment les particules réagissent. C'est crucial pour comprendre comment stabiliser la fusion.
3. Économiser de l'argent : Au lieu de construire des réacteurs physiques coûteux pour tester des idées, ils peuvent tout tester virtuellement et très vite.

5. En Résumé

CATAPULT, c'est comme donner un système de navigation GPS en temps réel à une armée de balles de tennis dans un labyrinthe magnétique géant.

• Avant : On regardait la carte lentement, un par un.
• Aujourd'hui : On a une vue satellite en direct, traitée par une armée de robots, qui dit exactement où chaque balle va atterrir.

C'est un pas de géant vers la réalisation d'une énergie de fusion propre et illimitée pour notre planète. Et le meilleur ? Le code de ce logiciel est gratuit et ouvert à tous ! 🌟⚡

Résumé technique

Titre

CATAPULT : Un intégrateur temporel accéléré par CUDA pour les particules alpha utilisant des tricubiques locales.

1. Problématique

Dans les dispositifs de confinement magnétique pour la fusion nucléaire, en particulier les stellarators, il est crucial de comprendre le confinement des particules alpha (ions de haute température). Une perte excessive de ces particules peut refroidir le plasma et endommager les parois du réacteur.

L'évaluation de ce confinement repose généralement sur des méthodes de Monte Carlo couplées à des intégrateurs temporels (solveurs d'EDP) pour tracer les trajectoires des particules. Les défis majeurs sont :

• La nécessité de tracer un grand nombre de particules pour réduire l'erreur stochastique ($O(n^{-1/2})$).
• La complexité des champs magnétiques (présence d'îlots, surfaces de flux fermées, ondes d'Alfvén).
• Le coût computationnel élevé des simulations actuelles sur CPU, limitant la fidélité des modèles (résolution de grille, tolérance d'erreur).

2. Méthodologie

Les auteurs proposent CATAPULT, une nouvelle implémentation sur GPU NVIDIA (CUDA) qui remplace les solveurs CPU parallélisés existants.

A. Modèle Physique et Dynamique

• Le mouvement des particules est décrit par le lagrangien de Littlejohn, qui inclut le mouvement de dérive dans un champ magnétique d'équilibre $B_0$.
• L'équation du mouvement est résolue dans deux systèmes de coordonnées :
  • Coordonnées de Boozer $(s, \theta, \zeta, v_\parallel)$ : adaptées à la géométrie du champ magnétique.
  • Coordonnées Cartésiennes : permettant de tracer les particules au-delà de la dernière surface de flux fermée et de gérer les îlots magnétiques.
• Le solveur utilise le schéma adaptatif Dormand-Prince d'ordre 5 (DP5), identique à celui utilisé dans le code de référence SIMSOPT.

B. Représentation du Champ Magnétique

• Le champ magnétique est représenté sur une grille régulière d'interpolation.
• Pour chaque cellule de la grille, les grandeurs magnétiques sont interpolées via une interpolation tricubique sur une grille $4 \times 4 \times 4$.
• Cette approche assure la continuité du champ entre les cellules, bien que la dérivabilité ne soit pas garantie aux interfaces (comme illustré par des splines cubiques).

C. Optimisation CUDA et Architecture

L'implémentation est optimisée pour maximiser le débit de particules (throughput) :

• Gestion de la mémoire : Les données constantes (paramètres globaux, coefficients DP5) sont placées en mémoire constante. Les données d'interpolation sont stockées en ordre row-major pour optimiser la localité spatiale.
• Parallélisation : Un bloc de 64 threads (2 warps) est assigné à un groupe de 8 particules.
• Coalescence des accès mémoire : Pour chaque étape DP5, un ensemble de $m$ threads consécutifs calcule les données d'interpolation pour une seule particule. Grâce à l'ordre de stockage, les lectures de mémoire sont coalescées (un seul accès mémoire pour des éléments consécutifs), réduisant le trafic d'instructions.
• Gestion des singularités :
  • Extrapolation : Lorsque les particules sortent du domaine d'interpolation (ex: $s > 1$), CATAPULT utilise les données de la cellule d'interpolation la plus proche, évitant les erreurs silencieuses observées dans SIMSOPT.
  • Coordonnées Pseudo-Cartésiennes : Pour éviter les erreurs numériques dues à la singularité de l'axe dans les coordonnées de Boozer, l'estimation d'erreur est calculée dans un système transformé $(s \cos \theta, s \sin \theta, \zeta, v_\parallel)$.

3. Contributions Clés

1. Accélération GPU : Développement d'un solveur entièrement CUDA qui surpasse significativement les implémentations CPU parallélisées.
2. Gestion Avancée des Champs : Support natif des champs magnétiques d'équilibre et des ondes d'Alfvén de cisaillement (Shear Alfven Waves).
3. Optimisation de la Mémoire : Partage d'une seule copie des données de la grille d'interpolation entre tous les threads GPU, contrairement aux implémentations CPU qui dupliquent ces données pour chaque thread. Cela permet d'utiliser des grilles de plus haute résolution sans épuiser la mémoire.
4. Correction Numérique : Résolution des problèmes d'extrapolation hors domaine et de singularité des coordonnées via des techniques spécifiques.

4. Résultats

Les tests ont été effectués sur des dispositifs modèles (ATEN, HSX, NCSX, QA, QH) et sur le supercalculateur Perlmutter (utilisant des GPU A100).

• Précision : Les résultats GPU sont cohérents avec les résultats CPU (SIMSOPT), montrant une convergence de la fraction de particules perdues lorsque la résolution de la grille et la tolérance d'intégration sont augmentées.
• Performance (Vitesse) :
  • Dans les configurations de vide (sans ondes), CATAPULT est 5 à 10 fois plus rapide que 128 cœurs CPU pour un nombre de particules saturant le GPU (~25 000 particules).
  • En présence d'ondes d'Alfvén de cisaillement, l'accélération est encore plus marquée, atteignant un facteur de 40 à 60 fois par rapport au nœud CPU.
• Évolutivité : Le temps d'exécution dépend principalement de la tolérance d'intégration (fidélité) plutôt que de la résolution de la grille, grâce à la localité des données en cache.
• Fidélité accrue : Grâce à la réduction des coûts de calcul, il devient possible d'augmenter la résolution des grilles et le nombre de particules, permettant des simulations de plus haute fidélité.

5. Signification et Perspectives

CATAPULT représente une avancée majeure pour la simulation des stellarators :

• Il permet d'explorer des physiques plus complexes (trajets complets des orbites, collisions) qui étaient auparavant trop coûteuses.
• Il ouvre la voie à l'optimisation directe du confinement et à l'étude détaillée des charges sur les parois (wall loads).
• Le code est intégré dans le package Python firm3d et est disponible en open source.

Les travaux futurs incluront l'adaptation à d'autres architectures GPU, l'optimisation pour les nouvelles générations de matériel, et l'amélioration des schémas d'interpolation et d'intégration pour accroître encore la précision numérique.