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

Questo articolo presenta una libreria di multi-threading sicura in tempo reale sviluppata per il sistema di controllo del plasma DIII-D che ottimizza con successo l'esecuzione dei codici fisici TORBEAM e STRIDE a meno di 20 ms e 100 ms, rispettivamente, consentendo calcoli cruciali sulla propagazione delle onde dell'elettrone circolare e sul limite di stabilità per le future centrali a fusione.

L'essenziale

Immaginate un reattore a fusione come il DIII-D tokamak come una gigantesca, caldissima e vorticosa tempesta di elettricità (plasma) che deve essere tenuta perfettamente ferma all'interno di una bottiglia magnetica. Se la tempesta diventa troppo selvaggia, può schiantarsi contro le pareti e distruggere la macchina. Per mantenerla al sicuro, un "Sistema di Controllo del Plasma" (PCS) agisce come il pilota, effettuando continui piccoli aggiustamenti.

Tuttavia, la tempesta cambia più velocemente di quanto un essere umano possa reagire. Il pilota ha bisogno di un cervello informatico super-veloce capace di prevedere il comportamento della tempesta e suggerire aggiustamenti in un battito di ciglia. È qui che entra in gioco il documento.

Il Problema: Il collo di bottiglia dell' "Unico Lavoratore"

Immaginate di essere uno chef che cerca di cucinare un banchetto massiccio. Avete una ricetta (un codice fisico) che vi dice come cucinare il cibo. Ma avete solo uno chef (un singolo core di un processore informatico) che si occupa di tutto il taglio, la mescolatura e la cottura. Se la ricetta è troppo complessa, lo chef viene sopraffatto, il cibo brucia e il banchetto fallisce.

Nel mondo della fusione, queste "ricote" sono simulazioni fisiche complesse (come TORBEAM e STRIDE) che calcolano come riscaldare il plasma o verificare se sta per diventare instabile. Tradizionalmente, questi calcoli erano troppo lenti per essere eseguiti in tempo reale perché cercavano di fare tutto con un solo "chef".

La Soluzione: Una squadra di chef "Real-Time Safe"

Gli autori hanno costruito un nuovo sistema per trasformare quello chef singolo in una squadra di chef che lavorano in perfetta sincronia.

1. Il Manager e i Lavoratori: Hanno creato una libreria speciale (un insieme di regole) che funge da Manager. Il Manager assegna piccoli compiti indipendenti a un gruppo di thread Worker (altri core informatici).
2. Niente Caos, Solo Ordine: Nei normali programmi informatici, quando si aggiungono più lavoratori, questi potrebbero confondersi, aspettarsi troppo l'un l'altro o far crashare il sistema se uno commette un errore. Il sistema degli autori è "real-time safe". È come un'unità militare dove ogni soldato sa esattamente quando muoversi e quando fermarsi. Utilizzano una speciale "stretta di mano" (variabili atomiche) per dire: "Sono pronto", "Ho finito" e "Iniziamo il prossimo round".
3. Tempistica Deterministica: La parte più importante è che questo sistema garantisce di completare il proprio lavoro entro un limite di tempo rigoroso. Non importa se il computer è occupato con altre cose; questa squadra è isolata e finirà sempre in tempo. Questo è cruciale perché se il computer impiega troppo tempo, il plasma potrebbe essersi già schiantato.

Le due Ricette Principali che hanno Cotto

Il team ha utilizzato questo nuovo sistema "multi-chef" per velocizzare due codici fisici specifici:

1. TORBEAM: La Guida del Raggio Laser

• Cosa fa: Immaginate di cercare di colpire un bersaglio minuscolo e in movimento all'interno di una stanza buia con un laser. Il plasma è la stanza, e il "laser" è un raggio di energia (Riscaldamento Ciclotronico degli Elettroni) usato per controllare la stabilità del plasma.
• La Sfida: Il computer deve calcolare l'esatto percorso che il raggio laser percorrerà attraverso il plasma per colpire il punto giusto.
• Il Risultato: Poiché ogni raggio laser (proveniente da diverse macchine chiamate gyrotron) viaggia indipendentemente, il nuovo sistema ha permesso ai "lavoratori" di calcolare i percorsi per tutti i raggi contemporaneamente.
• La Velocità: Hanno completato il calcolo in meno di 20 millisecondi. Questo è abbastanza veloce per dirigere i laser in tempo reale, mantenendo stabile il plasma.

2. STRIDE: Il Controllore di Stabilità

• Cosa fa: Immaginate un funambolo. STRIDE è l'ispettore della sicurezza che controlla costantemente se il funambolo sta per cadere. Calcola un "punteggio di stabilità" per vedere se il plasma sta per diventare instabile e schiantarsi.
• La Sfida: Questo calcolo è molto pesante e solitamente richiede troppo tempo per l'uso in tempo reale.
• Il Trucco: Gli autori hanno capito che potevano suddividere il controllo di sicurezza in molti piccoli pezzi indipendenti (come controllare diverse sezioni della corda tesa). Hanno inviato questi pezzi ai "lavoratori" per risolverli simultaneamente, per poi combinare le risposte.
• La Velocità: Hanno ridotto il tempo di calcolo a circa 100 millisecondi. Questo è abbastanza veloce per avvisare il sistema di controllo prima che avvenga un disastro.

In Sintesi

Il documento dimostra che costruendo una squadra di processori informatici specializzata e altamente disciplinata (una "libreria multi-threading real-time safe"), possiamo eseguire simulazioni fisiche complesse abbastanza velocemente da poter effettivamente controllare un reattore a fusione mentre è in funzione.

• TORBEAM (direzione del laser) gira in ~20ms.
• STRIDE (controllo della stabilità) gira in ~100ms.

Senza questo nuovo sistema di "lavoro di squadra", questi calcoli sarebbero troppo lenti per essere utili a mantenere un reattore a fusione sicuro e stabile. Questo lavoro dimostra che possiamo rendere il "pilota" di un reattore a fusione abbastanza intelligente da gestire l'estrema velocità della tempesta di plasma.

Sommario tecnico

Sintesi Tecnica: Codici di Fisica in Tempo Reale Parallelizzati per il Controllo del Plasma su DIII-D

Problema
I sistemi di controllo del plasma (PCS) in tempo reale per i tokamak, come quelli previsti per i futuri reattori a fusione come ITER e SPARC, richiedono che i codici basati sulla fisica operino con una latenza deterministica e un'elevata robustezza. Sebbene esistano codici di fisica offline per il ray tracing dell'elettronica a ciclo di riscaldamento elettronico (ECH) (TORBEAM) e l'analisi della stabilità magnetoidrodinamica (MHD) ideale (STRIDE), le loro implementazioni in tempo reale affrontano spesso colli di bottiglia nelle prestazioni su singoli core CPU. Le soluzioni di multi-threading esistenti, come OpenMP, sono inadatte agli ambienti PCS per la fusione perché mancano di garanzie real-time rigorose, si affidano a comportamenti definiti dall'implementazione e introducono jitter di latenza non deterministico dovuto alle interazioni con il kernel. Inoltre, l'accelerazione tramite GPU è spesso impraticabile per questi specifici codici a causa degli elevati costi di avvio e di una granularità di parallelismo insufficiente, rendendo necessaria una robusta strategia di parallelizzazione basata su CPU che garantisca un determinismo su scala del microsecondo.

Metodologia
Gli autori hanno sviluppato e implementato una libreria di multi-threading personalizzata e sicura per il tempo reale sul DIII-D PCS per consentire l'esecuzione parallela dei codici TORBEAM e STRIDE.

• Architettura della Libreria: La libreria utilizza un pattern "Manager-Worker" implementato in C11. Un singolo thread Manager (coincidente con il thread principale del PCS) assegna unità di lavoro a molteplici thread Worker. La comunicazione avviene esclusivamente tramite variabili atomiche di basso livello utilizzando il modello di memoria acquire-rilascia (acquire-release) di C11, evitando astrazioni di alto livello come i semafori che interagiscono con il kernel.
• Meccanismo di Sincronizzazione: Il sistema impiega un protocollo di sincronizzazione bidirezionale ispirato alla caratteristica di "rendezvous" di Ada. Il Manager e i Worker si notificano indipendentemente lo stato tramite flag atomici (Ready e Complete). Ciò assicura che tutti i thread raggiungano i punti di sincronizzazione all'inizio e alla fine delle sezioni critiche senza comportamenti indefiniti. Il design privilegia il determinismo rispetto al throughput massimo, accettando che i thread possano attendere in modalità spinning mentre aspettano che altri si riprendano.
• Ambiente Hardware: L'implementazione è stata eseguita su sistemi con core CPU isolati per prevenire interferenze da parte di altri processi e interrupt del kernel. L'implementazione di STRIDE ha utilizzato un sistema operativo real-time commerciale (RedHawk) per minimizzare il jitter della latenza, mentre TORBEAM è stato eseguito su un ambiente CentOS 7 standard con isolamento dei core.
• Parallelizzazione Specifica del Codice:
  • rt-TORBEAM: I calcoli di ray tracing per gli indipendenti gyrotron sono stati parallelizzati. Il codice è stato semplificato da un modello di fascio gaussiano a un singolo tracciamento di raggio per calcolare la posizione di massima assorbimento $(R, Z)$, che viene poi convertita in una coordinata di flusso normalizzata ($\rho$) per lo sterzo degli specchi.
  • rt-STRIDE: Il problema della stabilità ideale è stato riformulato come un'equazione di Riccati anziché un'equazione di Newcomb. Ciò consente il calcolo della Matrice di Transizione di Stato (STM) tramite un metodo di "shooting" dove il dominio di integrazione viene suddiviso in intervalli. Questi intervalli sono distribuiti tra i thread Worker in base a una metrica di "stiffness" (maggiore densità di step vicino alle superfici razionali), e le STM risultanti vengono combinate dal Manager per calcolare il parametro di stabilità finale ($\delta W$).

Contributi Chiave

1. Libreria di Multi-threading Sicura per il Tempo Reale: Lo sviluppo di una libreria di threading deterministica e priva di kernel che consente la parallelizzazione sicura sull'hardware PCS per la fusione, affrontando i limiti delle librerie standard come Openby.
2. rt-TORBEAM Parallelizzato: Implementazione riuscita di un codice di ray tracing parallelizzato per lo sterzo degli specchi ECH, capace di gestire simultaneamente molteplici gyrotron.
3. rt-STRIDE Parallelizzato: Implementazione di un codice di stabilità MHD ideale parallelizzato utilizzando una formulazione di Riccati e la suddivisione degli intervalli, consentendo il calcolo in tempo reale dei limiti di stabilità.
4. Validazione Sperimentale: Dimostrazione di questi codici operanti entro i rigorosi vincoli di tempo di ciclo del DIII-D PCS durante cariche di plasma effettive.

Risultati

• Performance di rt-TORBEAM: Il codice è stato eseguito costantemente in meno di 20 ms per quattro gyrotron sterzati. Il sistema ha dimostrato un ragionevole inseguimento dei valori target di $\rho$ ($|\rho_{target} - \rho_{torbeam}| \le 0.05$). Il tempo di ciclo è rimasto costante all'aggiunta di ulteriori gyrotron, purché fosse disponibile un thread worker libero per ciascuno. Un picco nel tempo di ciclo a circa 40 ms è stato osservato in specifici casi limite, attribuito al codice di fisica piuttosto che alla libreria di threading.
• Performance di rt-STRIDE: Il codice ha calcolato costantemente il parametro di stabilità $\delta W$ in circa 100 ms su un sistema a 72 core. Il profiling ha rivelato che l'integrazione della STM stessa richiedeva solo circa 20 ms, con la maggior parte del tempo consumata dal pre-processing seriale dei dati (conversione dall'equilibrio alle coordinate di flusso).
• Inseguimento e Controllo: I risultati sperimentali hanno mostrato che, sebbene i calcoli fisici fossero sufficientemente veloci per il controllo, l'hardware fisico degli specchi (che impiega circa 0,5 s per muoversi) e la velocità della ricostruzione degli equilibri in tempo reale limitavano le prestazioni complessive di inseguimento, particolarmente durante cambiamenti rapidi dei valori target.

Significatività e Rivendicazioni
Il documento afferma che la libreria di multi-threading sviluppata è un abilitatore critico per i dispositivi a fusione di prossima generazione, permettendo ai codici basati sulla fisica di soddisfare i requisiti di tempo di esecuzione per il controllo in tempo reale. Gli autori sottolineano che, a differenza dei modelli surrogati basati su machine learning, che possono mancare di robustezza al di fuori dei loro dati di addestramento, questi codici basati sulla fisica offrono "prestazioni robuste dal primo giorno di operatività". Il lavoro dimostra che, con una corretta parallelizzazione e un threading deterministico, codici fisici complessi come TORBEAM e STRIDE possono essere integrati nel PCS per fornire informazioni essenziali su deposizione ECH e limiti di stabilità ideale. Gli autori suggeriscono che questa libreria può essere applicata ad altri codici in tempo reale, inclusa l'analisi della stabilità verticale e la predizione dei profili, e notano che gli sforzi futuri dovrebbero concentrarsi sulla parallelizzazione dei passaggi di pre-processing dei dati per ridurre ulteriormente la latenza.