Performance Portable $\mathrm{SU}(N)$ Lattice Gauge Theory Simulation with Kokkos

Per comprendere l'universo al suo livello più fondamentale, i fisici si rivolgono spesso a un metodo che tratta lo spazio e il tempo non come un tessuto liscio e continuo, ma come una griglia di minuscoli punti discreti. Immaginate una vasta scacchiera invisibile che si estende attraverso il cosmo, dove ogni intersezione contiene un pezzo di informazione sulle forze che legano la materia. Questo è l'essenza della teoria di gauge su reticolo, uno strumento potente utilizzato per simulare il comportamento delle particelle subatomiche e delle forze che le governano, in particolare la forza forte che tiene insieme i nuclei atomici. Scomponendo l'universo in questi frammenti gestibili, gli scienziati possono usare computer massicci per tracciare i percorsi delle particelle e calcolare come interagiscono, eseguendo efficacemente le leggi della fisica al contrario per vedere cosa emerge. Tuttavia, l'hardware utilizzato per eseguire queste simulazioni è diventato incredibilmente diversificato. I supercomputer di oggi sono un mosaico di diversi processori, dai chip informatici standard alle schede grafiche specializzate, ognuno dei quali parla un linguaggio tecnico differente e richiede il proprio set unico di istruzioni per funzionare efficientemente.

Questa diversità crea un significativo mal di testa per i ricercatori. Per ottenere le migliori prestazioni da una macchina specifica, devono spesso riscrivere il codice della loro simulazione da zero per ogni diverso tipo di processore che vogliono utilizzare. Mantenere versioni separate dello stesso algoritmo per computer diversi è lento, costoso e incline agli errori. La domanda che la comunità scientifica si pone è se sia possibile scrivere una singola versione di questo complesso codice di simulazione che possa girare efficientemente su qualsiasi computer moderno, da un singolo processore a un enorme supercomputer, senza sacrificare la velocità. Wei Sun, un ricercatore dell'Istituto di Fisica delle Alte Energie di Pechino, ha risposto a questa domanda con un nuovo strumento software chiamato kwqft. Questo programma dimostra con successo che è possibile creare un codice singolo e universale per simulare la forza forte che performa altrettanto bene di un codice specializzato e specifico per la macchina, indipendentemente dall'hardware su cui è in esecuzione.

Il cuore di questo traguardo risiede in un approccio intelligente alla progettazione del software che tratta l'architettura del computer come una variabile piuttosto che come un vincolo fisso. I ricercatori hanno costruito la loro simulazione utilizzando un moderno framework di programmazione che agisce come un traduttore universale. Invece di scrivere istruzioni separate per una scheda grafica, un processore standard o un chip specializzato, hanno scritto un unico set di istruzioni che il framework adatta automaticamente alla macchina su cui è in esecuzione. Ciò consente allo stesso codice di girare su una vasta gamma di sistemi, inclusi quelli di produttori diversi, senza che i ricercatori debbano toccare nuovamente il codice sorgente. Il software è progettato per essere flessibile, permettendo agli scienziati di cambiare il numero di dimensioni nella loro simulazione o le proprietà specifiche delle particelle che stanno studiando semplicemente regolando alcune impostazioni prima dell'avvio del programma, invece di riscrivere l'intero programma.

Per provare che questo approccio universale funzioni, il team ha sottoposto il proprio software a rigorosi test su una vasta gamma di hardware. Hanno eseguito simulazioni su una potente scheda grafica NVIDIA, un chip specializzato di Hygon e diversi tipi di processori centrali, inclusi modelli avanzati di Huawei e Intel. In ogni caso, il software ha prodotto risultati che corrispondevano alle soluzioni esatte note per versioni più semplici del problema e concordavano con i dati precedentemente pubblicati per scenari più complessi. I ricercatori hanno verificato che la simulazione riproducesse correttamente il comportamento della forza forte in due, tre e quattro dimensioni, e per diversi tipi di gruppi di particelle, confermando che il codice universale non stava perdendo accuratezza nel processo di diventare portabile.

I risultati sulle prestazioni sono stati particolarmente sorprendenti. Eseguendo su una scheda grafica di fascia alta, il software universale ha raggiunto quasi la stessa velocità di una versione del codice ottimizzata appositamente per quella scheda, raggiungendo oltre il 90 percento dell'efficienza del codice specializzato nelle simulazioni più grandi. Su un enorme supercomputer a Shenzhen, noto come LineShine, il software è scalato per girare su 512 nodi, o unità di calcolo individuali, simultaneamente. Man mano che i ricercatori aggiungevano potenza di calcolo, la simulazione accelerava significativamente, dimostrando che il software poteva gestire la complessa comunicazione necessaria per mantenere migliaia di processori che lavorano insieme in armonia. Inoltre, sull'ultima generazione di processori progettati per calcoli vettoriali ad alta velocità, il software è stato in grado di elaborare più punti sulla griglia contemporaneamente, velocizzando la simulazione di un fattore di tre rispetto a una versione standard dello stesso codice.

Le implicazioni di questo lavoro vanno oltre un singolo test di successo. Dimostrando che un singolo codice sorgente può offrire prestazioni competitive in un panorama hardware così diversificato, i ricercatori hanno rimosso una barriera importante al progresso scientifico. Gli scienziati non devono più scegliere tra scrivere un codice che sia facile da mantenere e un codice che sia veloce; possono ora scrivere una volta sola e girare ovunque, certi che il software si adatterà alla macchina su cui si trova. Ciò apre la porta a simulazioni più complesse dell'universo, permettendo ai ricercatori di esplorare questioni riguardanti i primi momenti del cosmo e la natura della materia con maggiore facilità e su una scala più ampia rispetto al passato. Il successo di kwqft suggerisce che il futuro dell'informatica ad alte prestazioni in fisica potrebbe non risiedere nel riscrivere costantemente il codice per nuove macchine, ma nel creare strumenti flessibili e intelligenti capaci di navigare autonomamente nel mutevole panorama tecnologico.