ORB5X v1.0: a performance-portable global electromagnetic gyrokinetic PIC code in C++/Kokkos built using Agentic AI

La Grande Caccia alla Fusione e la Barriera Linguistica

Immaginate di cercare di costruire una stella proprio qui sulla Terra. Questo è il sogno della fusione nucleare, il processo che alimenta il sole, che promette un'energia pulita e illimitata. Per far sì che ciò accada, gli scienziati intrappolano il plasma super-caldo — una zuppa vorticosa di particelle cariche — in gabbie magnetiche chiamate tokamak. Ma questo plasma è caotico; è come cercare di prevedere il percorso di un miliardo di piccole api iperattive che si scontrano costantemente tra loro e cambiano direzione. Per capire come mantenere questo plasma stabile e abbastanza caldo da fondersi, gli scienziati utilizzano massicce simulazioni al computer. Queste simulazioni fungono da laboratori virtuali, tracciando come le particelle si muovano e come i campi elettromagnetici si torcano e si snodino.

Uno degli strumenti più affidabili per questo compito è un codice chiamato ORB5. Pensate a ORB5 come a uno chef esperto, un vecchio maestro che cucina la stessa ricetta perfetta per la fusione da decenni. Tuttavia, questo chef parla una sola lingua: il Fortran, un linguaggio di programmazione inventato negli anni '50. Sebbene il Fortran sia eccellente per la matematica, sta diventando un po' come un'auto d'epoca: difficile da riparare, difficile da aggiornare e difficile da guidare sulle strade moderne (come i supercomputer più recenti e veloci dotati di potenti schede grafiche). Mentre il mondo dell'informatica si sposta verso nuovi hardware, mantenere in funzione questa "auto d'epoca" in modo efficiente sta diventando una sfida. La domanda non è se la ricetta sia buona — lo è — ma come possiamo tradurre quella ricetta in un linguaggio moderno affinché possa girare sui motori più veloci di oggi senza cambiare il sapore del piatto?

Il Progetto di Traduzione: ORB5X

Questo articolo presenta ORB5X, una nuova e audace versione di quel famoso codice di fusione. Il team non ha cercato di inventare una nuova ricetta o di cambiare la fisica del plasma. Invece, hanno agito come maestri traduttori. Hanno preso l'originale codice Fortran e, con l'aiuto dell' "IA Agente" (assistenti informatici intelligenti che scrivono e correggono il codice), hanno riscritto l'intero programma in C++, un linguaggio di programmazione moderno.

Perché questo è importante? Immaginate di avere una brillante planimetria di una casa, ma che è disegnata su un pezzo di carta che sta sbiadendo e che si adatta solo a un tipo specifico di tavolo da disegno. ORB5X è come ridisegnare esattamente quella stessa planimetria su un tablet digitale che funziona su qualsiasi computer, qualsiasi tablet o persino su un supercomputer nello spazio. L'obiettivo era preservare l' "anima matematica" del codice originale rendendolo "performance-portable" (prestazionale-portabile). Ciò significa che il nuovo codice può girare altrettanto bene su un normale laptop, un potente desktop o un enorme supercomputer con migliaia di schede grafiche, senza la necessità di una riscrittura completa per ogni macchina.

Come ci sono riusciti: Il Co-pilota IA

La traduzione non è stata un semplice lavoro di copia-incolla; è stata una ricostruzione attenta, passo dopo passo. I ricercatori hanno utilizzato un flusso di lavoro in cui esperti umani e agenti IA hanno lavorato insieme. L'IA ha agito come un co-pilota instancabile: analizzava un frammento del vecchio codice Fortran, suggeriva come scriverlo in moderno C++ e segnalava potenziali trappole dove la matematica avrebbe potuto sbagliare. Gli scienziati umani hanno poi revisionato ogni suggerimento, assicurandosi che il nuovo codice si comportasse esattamente come quello vecchio.

Hanno mantenuto le due versioni affiancate, come un dizionario bilingue. Se il vecchio codice aveva un file chiamato fields.F90, il nuovo ne aveva uno chiamato fields.cpp. Ciò ha permesso loro di controllare ogni singola operazione. Hanno utilizzato un toolkit moderno chiamato Kokolis, che funge da adattatore universale. Esso permette al codice di comunicare con diversi tipi di processori informatici (come le CPU Intel o le GPU NVIDIA) automaticamente, senza che il programmatore debba scrivere istruzioni separate per ciascuno di essi.

I Risultati: Accuratezza di Macchina

Il team ha messo alla prova ORB5X utilizzando tre famosi "stress test" che simulano diverse turbolenze del plasma:

1. ITG (Gradiente di Temperatura Ionica): Simulando come il calore si muove attraverso il plasma.
2. ITPA: Un test complesso che coinvolge particelle veloci e onde magnetiche.
3. Chirping: Osservando come la frequenza delle onde cambia mentre interagisce con particelle energetiche.

I risultati sono stati sbalorditivi. Quando hanno confrontato l'output del nuovo codice C++ con l'originale codice Fortran, i numeri corrispondevano con accuratezza di macchina. Ciò significa che la differenza tra i due era così minuscola da essere essenzialmente nulla, limitata solo dalla precisione della matematica del computer. Che eseguissero la simulazione su un laptop personale o su un enorme supercomputer come LUMI o CINECA, la fisica rimaneva identica.

Perché questo è un grande passo avanti

L'articolo conferma che ORB5X è un successo, ma stabilisce anche aspettative realistiche. Il nuovo codice non è necessariamente più veloce della versione Fortran originale su ogni singola macchina; anzi, il codice originale è ancora leggermente più veloce in alcuni casi perché è stato perfezionato nel corso di decenni. Tuttavia, la vera vittoria è la portabilità. Il codice Fortran originale sta diventando un sistema "legacy" (ereditato) difficile da eseguire su nuovi hardware. ORB5X, costruito con strumenti moderni, può essere compilato ed eseguito facilmente su laptop personali (anche sui Mac con chip Apple M4) e sui supercomputer più avanzati del mondo.

Gli autori sottolineano con cautela che questa è la versione 1.0. È una base solida, non un prodotto finito. Programmano di renderlo ancora più veloce in futuro riducendo il tempo necessario per spostare i dati tra la memoria del computer e il suo processore, e sperano di usarlo eventualmente per simulare forme ancora più complesse di gabbie magnetiche (come gli stellarator). Ma per ora, hanno dimostrato con successo che è possibile prendere un codice scientifico di decenni fa, altamente complesso, tradurlo in un linguaggio moderno con l'aiuto dell'IA e mantenere la fisica perfettamente intatta. È un ponte costruito tra il passato e il futuro della ricerca sulla fusione, garantendo che la ricerca di energia pulita possa continuare a correre sui motori più veloci che il mondo possa offrire.