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

Dit artikel presenteert een real-time veilige multi-threading bibliotheek ontwikkeld voor het DIII-D plasma controlesysteem die de uitvoering van de TORBEAM en STRIDE physics codes succesvol optimaliseert naar respectievelijk minder dan 20 ms en 100 ms, wat cruciale berekeningen van elektron cyclotron golfvoortplanting en stabiliteitslimieten mogelijk maakt voor toekomstige fusiekrachtcentrales.

De kern

Stel je een fusiereactor zoals de DIII-D tokamak voor als een gigantische, gloeiend hete, kolkende storm van elektriciteit (plasma) die perfect binnen een magnetische fles moet worden gehouden. Als de storm te wild wordt, kan hij tegen de wanden botsen en de machine vernietigen. Om hem veilig te houden, fungeert een "Plasma Control System" (PCS) als de piloot die constant kleine aanpassingen maakt.

Echter, de storm verandert sneller dan een mens kan reageren. De piloot heeft een super-snel computerbrein nodig dat het gedrag van de storm kan voorspellen en in een oogwenk aanpassingen kan voorstellen. Hier komt het artikel om de hoek kijken.

Het Probleem: De "Eén-Werker"-bottleneck

Stel je voor dat je een chef bent die probeert een enorm banket te bereiden. Je hebt een recept (een natuurkundige code) dat je vertelt hoe je het eten moet koken. Maar je hebt slechts één chef (een enkele computerprocessor-kern) die al het snijden, roeren en bakken doet. Als het recept te complex is, raakt de chef overweldigd, het eten verbrandt en mislukt het banket.

In de wereld van fusie zijn deze "recepten" complexe natuurkundige simulaties (zoals TORBEAM en STRIDE) die berekenen hoe het plasma moet worden verhit of controleren of het op het punt staat instabiel te worden. Traditioneel waren deze berekeningen te traag om in real-time te draaien omdat ze alles probeerden te doen met slechts één "chef".

De Oplossing: Een "Real-Time Veilige" Team van Chefs

De auteurs hebben een nieuw systeem gebouwd om die enkele chef te veranderen in een team van chefs die in perfecte synchronisatie werken.

1. De Manager en de Werkers: Ze hebben een speciale bibliotheek (een set regels) gemaakt die fungeert als een Manager. De Manager verdeelt kleine, onafhankelijke taken onder een groep Worker-threads (andere computerkernen).
2. Geen Chaos, Alleen Orde: In normale computerprogramma's kunnen er, wanneer je meer werkers toevoegt, problemen ontstaan waarbij ze in de war raken, te lang op elkaar wachten of het systeem laten crashen als één van hen een fout maakt. Het systeem van de auteurs is "real-time veilig". Het is als een militaire eenheid waar elke soldaat precies weet wanneer hij moet bewegen en wanneer hij moet stoppen. Ze gebruiken een speciale "handdruk" (atomaire variabelen) om te zeggen: "Ik ben klaar", "Ik ben klaar met mijn taak" en "Laten we de volgende ronde starten".
3. Deterministische Timing: Het belangrijkste deel is dat dit systeem garandeert dat het zijn werk binnen een strikte tijdslimiet voltooit. Het maakt niet uit of de computer druk is met andere zaken; dit team is geïsoleerd en zal altijd op tijd klaar zijn. Dit is cruciaal, want als de computer te lang doet, kan het plasma al gecrasht zijn.

De Twee Belangrijkste Recepten die Ze Kookten

Het team gebruikte dit nieuwe "multi-chef"-systeem om twee specifieke natuurkundige codes te versnellen:

1. TORBEAM: De Gids voor de Laserstraal

• Wat het doet: Stel je voor dat je met een laserstraal een klein, bewegend doelwit in een donkere kamer probeert te raken. Het plasma is de kamer, en de "laser" is een energiebundel (Electron Cyclotron Heating) die wordt gebruikt om de stabiliteit van het plasma te controleren.
• De Uitdaging: De computer moet exact berekenen welk pad de laserstraal door het plasma zal afleggen om de juiste plek te raken.
• Het Resultaat: Omdat elke laserstraal (van verschillende machines die gyrotronnen worden genoemd) onafhankelijk reist, liet het nieuwe systeem de "workers" de paden voor alle bundels tegelijkertijd berekenen.
• De Snelheid: Ze kregen de berekening gedaan in minder dan 20 milliseconden. Dit is snel genoeg om de lasers in real-time te sturen en het plasma stabiel te houden.

2. STRIDE: De Stabiliteitscontroleur

• Wat het doet: Stel je een koorddanser voor. STRIDE is de veiligheidsinspecteur die constant controleert of de wandelaar op het punt staat te vallen. Het berekent een "stabiliteitsscore" om te zien of het plasma op het punt staat instabiel te worden en te crashen.
• De Uitdaging: Deze berekening is erg zwaar en duurt normaal gesproken te lang voor real-time gebruik.
• De Truc: De auteurs realiseerden zich dat ze de veiligheidscontrole konden opdelen in veel kleine, onafhankelijke stukjes (zoals het controleren van verschillende secties van het koord). Ze stuurden deze stukjes naar de "workers" om ze simultaan op te lossen, en combineerden vervolgens de antwoorden.
• De Snelheid: Ze brachten de berekeningstijd terug naar ongeveer 100 milliseconden. Dit is snel genoeg om het controlesysteem te waarschuwen voordat er een ramp gebeurt.

De Kern van het Verhaal

Het artikel demonstreert dat door een gespecialiseerd, zeer gedisciplineerd team van computerprocessoren te bouwen (een "real-time veilige multi-threading bibliotheek"), zij complexe natuurkundige simulaties snel genoeg kunnen draaien om een fusiereactor daadwerkelijk te besturen terwijl deze in bedrijf is.

• TORBEAM (Lasersturing) draait in ~20ms.
• STRIDE (Stabiliteitscontrole) draait in ~100ms.

Zonder dit nieuwe "teamwerk"-systeem zouden deze berekeningen te traag zijn om nuttig te zijn voor het veilig en stabiel houden van een fusiereactor. Dit werk bewijst dat we de "piloot" van een fusiereactor slim genoeg kunnen maken om de extreme snelheid van de plasmastorm te beheersen.

Technische samenvatting

Technische Samenvatting: Geparallelliseerde Real-time Fysica-codes voor Plasmacontrole op DIII-D

Probleemstelling
Real-time plasmacontrolesystemen (PCS) voor tokamaks, zoals gepland voor toekomstige fusiereactoren zoals ITER en SPARC, vereisen dat natuurkundige codes met deterministische latentie en hoge robuustheid opereren. Hoewel er offline fysica-codes bestaan voor elektron-cyclotronverwarming (ECH) straaltracering (TORBEAM) en ideale magnetohydrodynamische (MHD) stabiliteitsanalyse (STRIDE), kampen hun real-time implementaties vaak met prestatieknelpunten op enkele CPU-kernen. Bestaande multi-threading oplossingen, zoals OpenMP, zijn ongeschikt voor fusie PCS-omgevingen omdat ze geen harde real-time garanties bieden, afhankelijk zijn van implementatie-geïntroduceerde gedragingen en niet-deterministische latentie-jitter introduceren door interacties met de kernel. Bovendien is GPU-acceleratie vaak onpraktisch voor deze specifieke codes vanwege hoge opstartkosten en onvoldoende parallellisme-granulariteit, wat de noodzaak voor een robuuste CPU-gebaseerde parallelliseringsstrategie die microseconde-schaal determinisme garandeert, onderstreept.

Methodologie
De auteurs hebben een aangepaste, real-time veilige multi-threading bibliotheek ontwikkeld en geïmplementeerd op de DIII-D PCS om de parallelle uitvoering van de TORBEAM- en STRIDE-codes mogelijk te maken.

• Bibliotheekarchitectuur: De bibliotheek maakt gebruik van een "Manager-Worker" patroon geïmplementeerd in C11. Een enkele Manager-thread (samenvallend met de hoofd-PCS-thread) wijst werkunits toe aan meerdere Worker-threads. Communicatie vindt uitsluitend plaats via laag-niveau atomaire variabelen met behulp van het C11 acquire-release geheugenmodel, waarbij hogere abstracties zoals semaforen die met de kernel interageren, worden vermeden.
• Synchronisatiemechanisme: Het systeem maakt gebruik van een bidirectioneel synchronisatieprotocol geïnspireerd door de Ada "rendezvous" functie. De Manager en Workers geven elkaar onafhankelijk status door via atomaire vlaggen (Ready en Complete). Dit zorgt ervoor dat alle threads de synchronisatiepunten bereiken aan het begin en einde van kritieke secties zonder ongedefinieerd gedrag. Het ontwerp geeft prioriteit aan determinisme boven maximale doorvoer, waarbij geaccepteerd wordt dat threads idool blijven draaien (spinnen) terwijl ze wachten tot anderen inhalen.
• Hardware-omgeving: De implementatie werd uitgevoerd op systemen met geïsoleerde CPU-kernen om interferentie van andere processen en kernel-interrupts te voorkomen. De STRIDE-implementatie maakte gebruik van een commercieel real-time besturingssysteem (RedHawk) om latentie-jitter te minimaliseren, terwijl TORBEAM draaide op een standaard CentOS 7-omgeving met core-isolatie.
• Codespecifieke Parallellisatie:
  • rt-TORBEAM: De straaltraceringsberekeningen voor onafhankelijke gyrotronen werden geparallelliseerd. De code is vereenvoudigd van een Gaussische straalmodel naar een enkelvoudige straaltracering om de maximale absorptielocatie $(R, Z)$ te berekenen, die vervolgens wordt omgezet naar een genormaliseerde fluxcoördinaat ($\rho$) voor spiegelsturing.
  • rt-STRIDE: Het ideale stabiliteitsprobleem is geherformuleerd als een Riccati-vergelijking in plaats van een Newcomb-vergelijking. Dit maakt het mogelijk om de State Transition Matrix (STM) te berekenen via een "shooting" methode waarbij het integratiedomein wordt opgesplitst in intervallen. Deze intervallen worden verdeeld onder Worker-threads op basis van een "stijfheid"-metriek (hogere stapdichtheid nabij rationele oppervlakken), waarna de resulterende STMs door de Manager worden gecombineerd om de uiteindelijke stabiliteitsparameter ($\delta W$) te berekenen.

Belangrijkste Bijdragen

1. Real-Time Veilige Multi-threading Bibliotheek: De ontwikkeling van een deterministische, kernel-vrije threading-bibliotheek die veilige parallellisatie op fusie PCS-hardware mogelijk maakt, waarmee de beperkingen van standaardbibliotheken zoals OpenMP worden geadresseerd.
2. Geparallelliseerde rt-TORBEAM: Succesvolle implementatie van een geparallelliseerde straaltraceringscode voor ECH-spiegelsturing, in staat om meerdere gyrotronen gelijktijdig te verwerken.
3. Geparallelliseerde rt-STRIDE: Implementatie van een geparallelliseerde ideale MHD-stabiliteitscode met behulp van een Riccati-formulering en intervalsplitsing, wat real-time berekening van stabiliteitslimieten mogelijk maakt.
4. Experimentele Validatie: Demonstratie van deze codes die binnen de strikte cyclustijden van de DIII-D PCS opereren tijdens daadwerkelijke plasma-shots.

Resultaten

• rt-TORBEAM Prestaties: De code werd consistent uitgevoerd in minder dan 20 ms voor vier gestuurde gyrotronen. Het systeem demonstreerde een redelijke tracking van doel- $\rho$ waarden ($|\rho_{target} - \rho_{torbeam}| \le 0.05$). De cyclustijd bleef constant naarmate er meer gyrotronen werden toegevoegd, mits er een vrije worker-thread beschikbaar was voor elke gyrotron. Een piek in de cyclustijd naar ~40 ms werd waargenomen in specifieke randgevallen, wat werd toegeschreven aan de natuurkundige code in plaats van de threading-bibliotheek.
• rt-STRIDE Prestaties: De code berekende consistent de $\delta W$ stabiliteitsparameter in ongeveer 100 ms op een 72-core systeem. Profiling onthulde dat de STM-integratie zelf slechts ~20 ms duurde, waarbij de meerderheid van de tijd werd verbruikt door seriële data-pre-processing (conversie van evenwicht naar fluxcoördinaat).
• Tracking en Controle: Experimentele resultaten toonden aan dat hoewel de natuurkundige berekeningen snel genoeg waren voor controle, de fysieke spiegelhardware (die ~0.5 s nodig heeft om te bewegen) en de snelheid van de real-time evenwichtsreconstructie de algehele tracking-prestaties beperkten, met name bij snelle veranderingen in doelwaarden.

Betekenis en Claims
Het artikel beweert dat de ontwikkelde multi-threading bibliotheek een cruciale enabler is voor volgende generaties fusie-apparatuur, waardoor natuurkundige codes aan de eisen van uitvoeringstijd voor real-time controle kunnen voldoen. De auteurs benadrukken dat, in tegen tegenstelling tot machine learning surrogate modellen, die mogelijk een gebrek aan robuustheid vertonen buiten hun trainingsdata, deze natuurkundige codes "robuuste prestatiegaranties bieden op de eerste dag van de operatie". Het werk demonstreert dat met de juiste parallellisatie en deterministische threading, complexe natuurkundige codes zoals TORBEAM en STRIDE kunnen worden geïntegreerd in de PCS om essentiële informatie te leveren over ECH-depositie en ideale stabiliteitslimieten. De auteurs suggereren dat deze bibliotheek kan worden toegepast op andere real-time codes, inclusief verticale stabiliteitsanalyse en profielvoorspelling, en merken op dat toekomstige inspanningen zich moeten richten op het parallelliseren van data-pre-processing stappen om de latentie verder te verlagen.