Parallelized Real-time Physics Codes for Plasma Control on DIII-D

Cet article présente une bibliothèque de multi-threading sûre en temps réel développée pour le système de contrôle du plasma DIII-D qui optimise avec succès l'exécution des codes de physique TORBEAM et STRIDE à moins de 20 ms et 100 ms, respectivement, permettant des calculs cruciaux de propagation d'ondes cyclotroniques électroniques et de limite de stabilité pour les futures centrales à fusion.

L'essentiel

Imaginez un réacteur de fusion comme le DIII-D, un tokamak, comme une tempête géante et super chaude d'électricité (plasma) qui doit être maintenue parfaitement immobile à l'intérieur d'une bouteille magnétique. Si la tempête devient trop sauvage, elle peut percuter les parois et détruire la machine. Pour la garder en sécurité, un « Système de Contrôle du Plasma » (PCS) agit comme le pilote, effectuant des ajustements minuscules de manière constante.

Cependant, la tempête change plus vite qu'un humain ne peut réagir. Le pilote a besoin d'un cerveau informatique super rapide capable de prédire le comportement de la tempête et de suggérer des ajustements en un clin d'œil. C'est là que l'article intervient.

Le Problème : Le goulot d'étranglement de l'« Ouvrier Unique »

Imaginez que vous soyez un chef essayant de cuisiner un banquet massif. Vous avez une recette (un code de physique) qui vous dit comment cuisiner la nourriture. Mais vous n'avez qu'un seul chef (un seul cœur de processeur informatique) qui s'occupe de tout le hachage, du mélange et de la cuisson. Si la recette est trop complexe, le chef est submergé, la nourriture brûle et le banquet échoue.

Dans le monde de la fusion, ces « recettes » sont des simulations physiques complexes (comme TORBEAM et STRIDE) qui calculent comment chauffer le plasma ou vérifient s'il est sur le point de devenir instable. Traditionnellement, ces calculs étaient trop lents pour être exécutés en temps réel car ils essayaient de tout faire avec un seul « chef ».

La Solution : Une équipe de chefs « Sécurisée en Temps Réel »

Les auteurs ont construit un nouveau système pour transformer ce chef solitaire en une équipe de chefs travaillant en parfaite synchronisation.

1. Le Manager et les Travailleurs : Ils ont créé une bibliothèque spéciale (un ensemble de règles) qui agit comme un Manager. Le Manager distribue des tâches petites et indépendantes à un groupe de fils (threads) Travailleurs (d'autres cœurs informatiques).
2. Pas de Chaos, Juste de l'Ordre : Dans les programmes informatiques normaux, lorsque vous ajoutez plus de travailleurs, ils peuvent se confondre, trop attendre les uns les autres ou faire planter le système si l'un d'eux commet une erreur. Le système des auteurs est « sûr en temps réel ». C'est comme une unité militaire où chaque soldat sait exactement quand bouger et quand s'arrêter. Ils utilisent un « handshake » spécial (variables atomiques) pour dire : « Je suis prêt », « J'ai fini » et « Commençons le prochain tour ».
3. Un Timing Déterministe : La partie la plus importante est que ce système garantit qu'il terminera son travail dans un délai strict. Peu importe si l'ordinateur est occupé par d'autres tâches, cette équipe est isolée et terminera toujours à temps. Cela est crucial car si l'ordinateur prend trop de temps, le plasma pourrait déjà avoir percuté les parois.

Les Deux Principales Recettes Qu'ils Ont Cuisinées

L'équipe a utilisé ce nouveau système de « multi-chefs » pour accélérer deux codes de physique spécifiques :

1. TORBEAM : Le Guide du Faisceau Laser

• Ce qu'il fait : Imaginez essayer de toucher une cible minuscule et mobile à l'intérieur d'une pièce sombre avec un laser. Le plasma est la pièce, et le « laser » est un faisceau d'énergie (chauffage cyclotronique électronique) utilisé pour contrôler la stabilité du plasma.
• Le Défi : L'ordinateur doit calculer le chemin exact que le faisceau laser empruntera à travers le plasma pour atteindre le bon endroit.
• Le Résultat : Comme chaque faisceau laser (provenant de différentes machines appelées gyrotrons) voyage indépendamment, le nouveau système a permis aux « travailleurs » de calculer les trajectoires de tous les faisceaux simultanément.
• La Vitesse : Ils ont réalisé le calcul en moins de 20 millisecondes. C'est assez rapide pour diriger les lasers en temps réel, maintenant ainsi la stabilité du plasma.

2. STRIDE : Le Vérificateur de Stabilité

• Ce qu'il fait : Imaginez un funambule. STRIDE est l'inspecteur de sécurité qui vérifie constamment si le funambule est sur le point de tomber. Il calcule un « score de stabilité » pour voir si le plasma est sur le point de devenir instable et de s'effondrer.
• Le Défi : Ce calcul est très lourd et prend généralement trop de temps pour une utilisation en temps réel.
• L'astuce : Les auteurs ont réalisé qu'ils pouvaient diviser la vérification de sécurité en de nombreuses petites parties indépendantes (comme vérifier différentes sections de la corde raide). Ils ont envoyé ces morceaux aux « travailleurs » pour les résoudre simultanément, puis ont combiné les réponses.
• La Vitesse : Ils ont réduit le temps de calcul à environ 100 millisecondes. C'est assez rapide pour avertir le système de contrôle avant qu'un désastre ne survienne.

L'Essentiel

L'article démontre qu'en construisant une équipe de processeurs informatiques spécialisée et hautement disciplinée (une « bibliothèque de multi-threading sécurisée en temps réel »), nous pouvons exécuter des simulations physiques complexes assez rapidement pour réellement contrôler un réacteur de fusion pendant qu'il fonctionne.

• TORBEAM (direction du laser) s'exécute en ~20 ms.
• STRIDE (vérification de la stabilité) s'exécute en ~100 ms.

Sans ce nouveau système de « travail d'équipe », ces calculs seraient trop lents pour être utiles afin de maintenir un réacteur de fusion sûr et stable. Ce travail prouve que nous pouvons rendre le « pilote » d'un réacteur de fusion assez intelligent pour gérer la vitesse extrême de la tempête de plasma.

Résumé technique

Résumé technique : Codes de physique en temps réel parallélisés pour le contrôle du plasma sur DIII-D

Énoncé du problème
Les systèmes de contrôle du plasma (PCS) en temps réel pour les tokamaks, tels que ceux prévus pour les futurs réacteurs de fusion comme ITER et SPARC, nécessitent que les codes de physique fonctionnent avec une latence déterministe et une grande robustesse. Bien qu'il existe des codes de physique hors ligne pour le tracé de rayons de chauffage par onde cyclotron électronique (ECH) (TORBEAM) et l'analyse de stabilité magnétohydrodynamique (MHD) idéale (STRIDE), leurs implémentations en temps réel font souvent face à des goulots d'étranglement de performance sur des cœurs de CPU uniques. Les solutions de multi-threading existantes, telles qu'OpenMP, sont inappropriées pour les environnements de PCS de fusion car elles manquent de garanties de temps réel strict, reposent sur des comportements définis par l'implémentation et introduisent une gigue de latence non déterministe due aux interactions avec le noyau (kernel). De plus, l'accélération par GPU est souvent peu pratique pour ces codes spécifiques en raison de coûts de démarrage élevés et d'une granularité de parallélisme insuffisante, nécessitant une stratégie de parallélisation robuste basée sur le CPU qui garantisse un déterminisme à l'échelle de la microseconde.

Méthodologie
Les auteurs ont développé et implémenté une bibliothèque de multi-threading personnalisée et sûre pour le temps réel sur le PCS de DIII-D afin de permettre l'exécution parallèle des codes TORBEAM et STRIDE.

• Architecture de la bibliothèque : La bibliothèque utilise un modèle « Manager-Worker » implémenté en C11. Un fil d'exécution unique (Manager) correspondant au fil principal du PCS assigne des unités de travail à plusieurs fils d'exécution (Workers). La communication est assurée exclusivement par des variables atomiques de bas niveau utilisant le modèle de mémoire acquire-release du C11, évitant ainsi les abstractions de haut niveau comme les sémaphores qui interagissent avec le noyau.
• Mécanisme de synchronisation : Le système emploie un protocole de synchronisation bidirectionnel inspiré de la fonctionnalité « rendez-vous » d'Ada. Le Manager et les Workers se notifient mutuellement de leur statut via des drapeaux atomiques (Ready et Complete). Cela garantit que tous les fils atteignent les points de synchronisation au début et à la fin des sections critiques sans comportement indéfini. La conception privilégie le déterminisme plutôt que le débit maximal, acceptant que les fils tournent inutilement (spin) en attendant que les autres les rattrapent.
• Environnement matériel : L'implémentation a été exécutée sur des systèmes avec des cœurs CPU isolés pour éviter toute interférence des autres processus et des interruptions du noyau. L'implémentation de STRIDE a utilisé un système d'exploitation en temps réel commercial (RedHawk) pour minimiser la gigue de latence, tandis que TORBEAM tournait sur un environnement CentOS 7 avec isolation de cœur.
• Parallélisation spécifique au code :
  • rt-TORBEAM : Les calculs de tracé de rayons pour les gyrotrons indépendants ont été parallélisés. Le code a été simplifié d'un modèle de faisceau gaussien à un tracé de rayon unique pour calculer la position d'absorption maximale $(R, Z)$, qui est ensuite convertie en une coordonnée de flux normalisée ($\rho$) pour l'orientation des miroirs.
  • rt-STRIDE : Le problème de stabilité idéale a été reformulé sous forme d'équation de Riccati plutôt que d'équation de Newcomb. Cela permet de calculer la matrice de transition d'état (STM) via une méthode de « tir » (shooting method) où le domaine d'intégration est divisé en intervalles. Ces intervalles sont distribués entre les fils Workers en fonction d'une métrique de « rigidité » (densité de pas plus élevée près des surfaces rationnelles), et les STM résultantes sont combinées par le Manager pour calculer le paramètre de stabilité final ($\delta W$).

Principales contributions

1. Bibliothèque de multi-threading sûre pour le temps réel : Le développement d'une bibliothèque de multi-threading déterministe et sans interaction avec le noyau, permettant une parallélisation sûre sur le matériel de PCS de fusion, répondant aux limites des bibliothèques standards comme OpenMP.
2. rt-TORBEAM parallélisé : Déploiement réussi d'un code de tracé de rayons parallélisé pour l'orientation des miroirs ECH, capable de gérer plusieurs gyrotrons simultanément.
3. rt-STRIDE parallélisé : Implémentation d'un code de stabilité MHD idéale parallélisé utilisant une formulation de Riccati et une division d'intervalles, permettant le calcul en temps réel des limites de stabilité.
4. Validation expérimentale : Démonstration de ces codes opérant dans les contraintes strictes de temps de cycle du PCS de DIII-D lors de tirages de plasma réels.

Résultats

• Performance de rt-TORBEAM : Le code s'est exécuté de manière constante en moins de 20 ms pour quatre gyrotrons orientés. Le système a démontré un suivi raisonnable des valeurs $\rho$ cibles ($|\rho_{target} - \rho_{torbeam}| \le 0,05$). Le temps de cycle est resté constant à mesure que des gyrotrons supplémentaires étaient ajoutés, à condition qu'un fil worker libre existe pour chacun. Un pic du temps de cycle à ~40 ms a été observé dans certains cas limites, attribué au code de physique plutôt qu'à la bibliothèque de threading.
• Performance de rt-STRIDE : Le code a calculé de manière constante le paramètre de stabilité $\delta W$ en environ 100 ms sur un système à 72 cœurs. Le profilage a révélé que l'intégration de la STM elle-même ne prenait que ~20 ms, la majeure partie du temps étant consommée par le prétraitement des données en série (conversion de l'équilibre en coordonnées de flux).
• Suivi et contrôle : Les résultats expérimentaux ont montré que bien que les calculs physiques soient suffisamment rapides pour le contrôle, le matériel physique des miroirs (mettant ~0,5 s pour bouger) et la vitesse de reconstruction des équilibres en temps réel limitaient les performances globales de suivi, particulièrement lors de changements rapides des valeurs cibles.

Signification et revendications
L'article affirme que la bibliothèque de multi-threading développée est un catalyseur critique pour les dispositifs de fusion de nouvelle génération, permettant aux codes de physique de respecter les exigences de temps d'exécution pour le contrôle en temps réel. Les auteurs soulignent qu'à l'inverse des modèles de substitution par apprentissage automatique (machine learning), qui peuvent manquer de robustesse en dehors de leurs données d'entraînement, ces codes basés sur la physique offrent des « garanties de performance robustes dès le premier jour d'opération ». Ce travail démontre qu'avec une parallélisation et un threading déterministes appropriés, des codes de physique complexes comme TORBEAM et STRIDE peuvent être intégrés au PCS pour fournir des informations essentielles sur le dépôt ECH et les limites de stabilité idéale. Les auteurs suggèrent que cette bibliothèque peut être appliquée à d'autres codes en temps réel, incluant l'analyse de stabilité verticale et la prédiction de profil, et notent que les efforts futurs devraient se concentrer sur la parallélisation des étapes de prétraitement des données pour réduire davantage la latence.