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

Dieses Paper präsentiert eine für das DIII-D-Plasmakontrollsystem entwickelte echtzeitfähige, sichere Multi-Threading-Bibliothek, welche die Ausführung der Physik-Codes TORBEAM und STRIDE erfolgreich auf unter 20 ms bzw. 100 ms optimiert und damit entscheidende Berechnungen zur Ausbreitung von Elektronen-Zyklotron-Wellen sowie zur Stabilitätsgrenze für zukünftige Fusionskraftwerke ermöglicht.

Das Wesentliche

Stellen Sie sich einen Fusionsreaktor wie den DIII-D Tokamak als einen riesigen, superheißen, wirbelnden Sturm aus Elektrizität (Plasma) vor, der in einem magnetischen Flaschenverschluss perfekt ruhig gehalten werden muss. Wenn der Sturm zu wild wird, kann er gegen die Wände prallen und die Maschine zerstören. Um ihn sicher zu halten, fungiert ein „Plasma Control System“ (PCS) als Pilot, der ständig winzige Anpassungen vornimmt.

Der Sturm verändert sich jedoch schneller, als ein Mensch reagieren könnte. Der Pilot benötigt ein super-schnelles Computergehirn, das das Verhalten des Sturms vorhersagen und Anpassungen im Bruchteil einer Sekunde vorschlagen kann. Hier kommt das Paper ins Spiel.

Das Problem: Der „Ein-Arbeiter“-Engpass

Stellen Sie sich vor, Sie sind ein Koch, der versucht, ein riesiges Bankett zu kochen. Sie haben ein Rezept (einen Physik-Code), das Ihnen sagt, wie Sie das Essen zubereiten sollen. Aber Sie haben nur einen einzigen Koch (einen einzelnen Computerprozessorkern), der das gesamte Hacken, Rühren und Backen erledigt. Wenn das Rezept zu komplex ist, ist der Koch überfordert, das Essen verbrennt und das Bankett scheitert.

In der Welt der Fusion sind diese „Rezepte“ komplexe Physik-Simulationen (wie TORBEAM und STRIDE), die berechnen, wie das Plasma erhitzt werden muss oder ob es kurz davor steht, instabil zu werden. Traditionell waren diese Berechnungen zu langsam, um in Echtzeit zu laufen, da sie versuchten, alles mit nur einem einzigen „Koch“ zu erledigen.

Die Lösung: Ein „Echtzeit-sicheres“ Team von Köchen

Die Autoren haben ein neues System entwickelt, um diesen einzelnen Koch in ein Team von Köchen zu verwandeln, die in perfekter Synchronisation arbeiten.

1. Der Manager und die Arbeiter: Sie haben eine spezielle Bibliothek (einen Satz von Regeln) erstellt, die wie ein Manager fungiert. Der Manager verteilt kleine, unabhängige Aufgaben an eine Gruppe von Worker-Threads (andere Computerkerne).
2. Kein Chaos, nur Ordnung: In normalen Computerprogrammen können Probleme auftreten, wenn man mehr Arbeiter hinzufügt – sie könnten verwirrt werden, zu lange aufeinander warten oder das System zum Absturz bringen, wenn einer einen Fehler macht. Das System der Autoren ist „echtzeit-sicher“. Es ist wie eine militärische Einheit, in der jeder Soldat genau weiß, wann er sich bewegen und wann er anhalten muss. Sie verwenden einen speziellen „Handschlag“ (atomare Variablen), um zu sagen: „Ich bin bereit“, „Ich bin fertig“ und „Lass uns die nächste Runde starten“.
3. Deterministisches Timing: Der wichtigste Teil ist, dass dieses System garantiert, seine Arbeit innerhalb eines strengen Zeitlimits abzuschließen. Es spielt keine Rolle, ob der Computer mit anderen Aufgaben beschäftigt ist; dieses Team ist isoliert und wird immer rechtzeitig fertig. Dies ist entscheidend, denn wenn der Computer zu lange braucht, könnte das Plasma bereits kollabiert sein.

Die zwei Hauptrezepte, die sie gekocht haben

Das Team nutzte dieses neue „Multi-Koch“-System, um zwei spezifische Physik-Codes zu beschleunigen:

1. TORBEAM: Der Laserstrahl-Guide

• Was es tut: Stellen Sie sich vor, Sie versuchen, ein winziges, sich bewegendes Ziel in einem dunklen Raum mit einem Laser zu treffen. Das Plasma ist der Raum, und der „Laser“ ist ein Energiestrahl (Elektronen-Zyklotron-Heizung), der verwendet wird, um die Stabilität des Plasmas zu steuern.
• Die Herausforderung: Der Computer muss genau berechnen, welchen Pfad der Laserstrahl durch das Plasma nehmen wird, um die richtige Stelle zu treffen.
• Das Ergebnis: Da jeder Laserstrahl (von verschiedenen Maschinen namens Gyrotronen) unabhängig voneinander reist, ermöglichte das neue System den „Arbeitern“, die Pfade für alle Strahlen gleichzeitig zu berechnen.
• Die Geschwindigkeit: Sie bekamen die Berechnung in weniger als 20 Millisekunden abgeschlossen. Dies ist schnell genug, um die Laser in Echtzeit zu steuern und das Plasma stabil zu halten.

2. STRIDE: Der Stabilitäts-Checker

• Was es tut: Stellen Sie sich einen Seiltänzer vor. STRIDE ist der Sicherheitsinspektor, der ständig prüft, ob der Seiltänzer kurz davor ist, zu fallen. Er berechnet einen „Stabilitätswert“, um zu sehen, ob das Plasma kurz davor steht, instabil zu werden und abzustürzen.
• Die Herausforderung: Diese Berechnung ist sehr rechenintensiv und dauert normalerweise zu lange für die Echtzeitnutzung.
• Der Trick: Die Autoren erkannten, dass sie die Sicherheitsprüfung in viele kleine, unabhängige Teile zerlegen konnten (wie das Überprüfen verschiedener Abschnitte des Seils). Sie schickten diese Teile an die „Arbeiter“, um sie simultan zu lösen, und kombinierten dann die Antworten.
• Die Geschwindigkeit: Sie reduzierten die Berechnungszeit auf etwa 100 Millisekunden. Dies ist schnell genug, um das Steuerungssystem zu warnen, bevor ein Desaster passiert.

Das Fazit

Das Paper zeigt, dass wir durch den Bau eines spezialisierten, hochdisziplinierten Teams von Computerprozessoren (einer „echtzeit-sicheren Multi-Threading-Bibliothek“) komplexe Physik-Simulationen schnell genug ausführen können, um einen Fusionsreaktor tatsächlich während des Betriebs zu steuern.

• TORBEAM (Lasersteuerung) läuft in ~20 ms.
• STRIDE (Stabilitätsprüfung) läuft in ~100 ms.

Oh\ne dieses neue „Teamwork“-System wären diese Berechnungen zu langsam, um einen Fusionsreaktor während des Betriebs sicher und stabil zu halten. Diese Arbeit beweist, dass wir den „Piloten“ eines Fusionsreaktors intelligent genug machen können, um mit der extremen Geschwindigkeit des Plasma-Sturms Schritt zu halten.

Technische Zusammenfassung

Technische Zusammenfassung: Parallelisierte Echtzeit-Physik-Codes zur Plasmakontrolle auf DIII-D

Problemstellung
Echtzeit-Plasmakontrollsysteme (PCS) für Tokamaks, wie sie für zukünftige Fusionsreaktoren wie ITER und SPARC geplant sind, erfordern physikbasierte Codes, die mit deterministischer Latenz und hoher Robustheit operieren können. Während Offline-Physik-Codes für die Elektronen-Zyklotron-Heizung (ECH) Strahlverfolgung (TORBEAM) und ideale magnetohydrodynamische (MHD) Stabilitätsanalyse (STRIDE) existieren, stoßen deren Echtzeit-Implementierungen oft auf Leistungsengpässe auf einzelnen CPU-Kernen. Bestehende Multi-Threading-Lösungen, wie OpenMP, sind für Fusions-PCS-Umgebungen ungeeignet, da sie keine Hard-Real-Time-Garantien bieten, auf implementierungsdefinierten Verhaltensweisen beruhen und eine nicht-deterministische Latenz-Jitter durch Kernel-Interaktionen verursachen. Eine GPU-Beschleunigung ist für diese spezifischen Codes oft unpraktisch aufgrund hoher Startkosten und unzureichender Parallelisierungsgranularität, was eine robuste CPU-basierte Parallelisierungsstrategie erforderlich macht, die eine Determinisierung im Mikrosekundenbereich gewährleistet.

Methodik
Die Autoren entwickelten und implementierten eine benutzerdefinierte, Echtzeit-sichere Multi-Threading-Bibliothek auf dem DIII-D PCS, um die parallele Ausführung der TORBEAM- und STRIDE-Codes zu ermöglichen.

• Bibliotheksarchitektur: Die Bibliothek nutzt ein „Manager-Worker“-Muster, das in C11 implementiert wurde. Ein einzelner Manager-Thread (der mit dem Haupt-PCS-Thread zusammenfällt) weist Arbeitseinheiten an mehrere Worker-Threads zu. Die Kommunikation erfolgt ausschließlich über Low-Level-Atomvariablen unter Verwendung des C11 Acquire-Release-Speichermodells, wodurch höherwertige Abstraktionen wie Semaphore vermieden werden, die mit dem Kernel interagieren.
• Synchronisationsmechanismus: Das System verwendet ein bidirektionales Synchronisationsprotokoll, das vom „Rendezvous“-Feature von Ada inspiriert ist. Der Manager und die Worker benachrichtigen sich gegenseitig über den Status mittels atomarer Flags (Ready und Complete). Dies stellt sicher, dass alle Threads an den Synchronisationspunkten zu Beginn und am Ende kritischer Abschnitte eintreffen, ohne undefiniertes Verhalten zu erzeugen. Das Design priorisiert Determinisierung gegenüber maximalem Durchsatz und nimmt in Kauf, dass Threads im Leerlauf rotieren (Spin-Waiting), während sie darauf warten, dass andere aufholen.
• Hardware-Umgebung: Die Implementierung lief auf Systemen mit isolierten CPU-Kernen, um Interferenzen durch andere Prozesse und Kernel-Interrupts zu verhindern. Die STRIDE-Implementierung nutzte ein kommerzielles Echtzeitbetriebssystem (RedHawk), um Latenz-Jitter zu minimieren, während TORBEAM in einer Standard-CentOS-7-Umgebung mit Kernisolierung lief.
• Kodenspezifische Parallelisierung:
  • rt-TORBEAM: Die Strahlverfolgung-Berechnungen für unabhängige Gyrotronen wurden parallelisiert. Der Code wurde von einem Gaußschen Strahlmodell zu einem Einzelstrahl-Modell vereinfacht, um den Ort der maximalen Absorption $(R, Z)$ zu berechnen, welcher dann in eine normalisierte Flusskoordinate ($\rho$) für die Spiegelsteuerung umgewandelt wird.
  • rt-STRIDE: Das ideale Stabilitätsproblem wurde als Riccati-Gleichung statt als Newcomb-Gleichung formuliert. Dies ermöglicht die Berechnung der Zustandsübergangsmatrix (State Transition Matrix, STM) mittels einer „Shooting“-Methode, bei der das Integrationsintervall in Intervalle unterteilt wird. Diese Intervalle werden basierend auf einer „Steifigkeits“-Metrik (höhere Schrittdichte nahe rationaler Oberflächen) an die Worker-Threads verteilt, und die resultierenden STMs werden vom Manager kombiniert, um den finalen Stabilitätsparameter ($\delta W$) zu berechnen.

Wesentliche Beiträge

1. Echtzeit-sichere Multi-Threading-Bibliothek: Die Entwicklung einer deterministischen, kernel-freien Multi-Threading-Bibliothek, die eine sichere Parallelisierung auf Fusions-PCS-Hardware ermöglicht und die Einschränkungen von Standardbibliotheken wie OpenMP adressiert.
2. Parallelisiertes rt-TORBEAM: Erfolgreiche Bereitstellung eines parallelisierten Strahlverfolgung-Codes für die ECH-Spiegelsteuerung, der in der Lage ist, mehrere Gyrotronen gleichzeitig zu handhaben.
3. Parallelisiertes rt-STRIDE: Implementierung eines parallelisierten idealen MHD-Stabilitäts-Codes unter Verwendung einer Riccati-Formulierung und Intervallteilung, was die Echtzeit-Berechnung von Stabilitätsgrenzen ermöglicht.
4. Experimentelle Validierung: Demonstration dieser Codes innerhalb der strengen Zykluszeit-Beschränkungen des DIII-D PCS während tatsächlicher Plasma-Schüsse.

Ergebnisse

• rt-TORBEAM-Leistung: Der Code wurde konsistent in unter 20 ms für vier gesteuerte Gyrotronen ausgeführt. Das System zeigte ein angemessenes Tracking der Ziel-$\rho$-Werte ($|\rho_{target} - \rho_{torbeam}| \le 0,05$). Die Zykluszeit blieb konstant, wenn zusätzliche Gyrotronen hinzugefügt wurden, sofern ein freier Worker-Thread zur Verfügung stand. Ein Anstieg der Zykluszeit auf ~40 ms wurde in spezifischen Grenzfällen beobachtet, was auf den Physik-Code und nicht auf die Threading-Bibliothek zurückzuführen war.
• rt-STRIDE-Leistung: Der Code berechnete den ideale Stabilitätsparameter $\delta W$ konsistent in etwa 100 ms auf einem 72-Kern-System. Das Profiling ergab, dass die STM-Integration selbst nur ~20 ms dauerte, wobei der Großteil der Zeit durch die serielle Datenvorverarbeitung (Konvertierung vom Gleichgewicht in Flusskoordinaten) konsumiert wurde.
• Tracking und Kontrolle: Experimentelle Ergebnisse zeigten, dass, obwohl die Physik-Berechnungen schnell genug für die Kontrolle waren, die physische Spiegelhardware (die ~0,5 s zum Bewegen benötigt) und die Geschwindigkeit der Echtzeit-Gleichgewicht-Rekonstruktion die gesamte Tracking-Leistung einschränkten, insbesondere bei schnellen Änderungen der Zielwerte.

Bedeutung und Ansprüche
Das Paper behauptet, dass die entwickelte Multi-Threading-Bibliothek ein entscheidender Wegbereiter für die nächste Generation von Fusionsgeräten ist, da sie es ermöglicht, dass physikbasierte Codes die Anforderungen an die Ausführungszeit für die Echtzeitsteuerung erfüllen. Die Autoren betonen, dass im Gegensatz zu Machine-Learning-Surrogatmodellen, die möglicherweise außerhalb ihrer Trainingsdaten nicht robust sind, diese physikbasierten Codes „robuste Leistungsgarantien am ersten Tag des Betriebs“ bieten. Die Arbeit zeigt, dass mit korrekter Parallelisierung und deterministischem Threading komplexe Physik-Codes wie TORBEAM und STRIDE in das PCS integriert werden können, um essenzielle Informationen über die ECH-Deposition und ideale Stabilitätsgrenzen zu liefern. Die Autoren legen nahe, dass diese Bibliothek auch auf andere Echtzeit-Codes angewendet werden kann, einschließlich vertikaler Stabilitätsanalyse und Profilvorhersage, und merken an, dass zukünftige Bemühungen darauf abzielen sollten, die Datenvorverarbeitungsschritte zu parallelisieren, um die Latenz weiter zu reduzieren.