CP2K on the road to exascale

Immaginate di cercare di capire come si comporta una goccia d'acqua, non solo come un liquido statico, ma come un sistema vivo e pulsante dove ogni singolo atomo è in costante movimento, collisione e reazione. Per fare ciò, gli scienziati utilizzano un potente tipo di simulazione al computer che tratta la materia secondo le rigide leggi della meccanica quantistica, la fisica che governa l'infinitamente piccolo. Queste simulazioni sono incredibilmente esigenti perché devono tracciare simultaneamente il comportamento di elettroni e nuclei, un compito che richiede una potenza di calcolo immensa. Per decenni, i ricercatori si sono affidati ai supercomputer per eseguire questi calcoli, ma man mano che i sistemi che vogliono studiare diventano più grandi — passando da piccoli cluster a vasti materiali con milioni di atomi — il tempo necessario per ottenere una risposta è diventato un collo di bottiglia. L'obiettivo è sempre stato quello di raggiungere un punto in cui i computer possano risolvere questi problemi così velocemente da permettere agli scienziati di osservare l'evolversi di complessi processi chimici in tempo reale, simulando interi mondi di atomi anziché solo pochi.

Un team di ricercatori dietro il pacchetto software CP2K, uno strumento ampiamente utilizzato per questo tipo di simulazioni atomiche, ha lavorato per colmare il divario tra le capacità attuali e la prossima generazione di supercomputer, noti come macchine exascale. Questi nuovi computer sono progettati per eseguire un quint Trilione di calcoli al secondo, offrendo un salto enorme in termini di velocità. Tuttavia, non è sufficiente avere semplicemente hardware più veloce; il software deve essere riscritto per gestire il modo unico in cui operano queste macchine. Il lavoro del team si concentra sul perfezionamento degli algoritmi centrali che guidano le simulazioni, assicurando che possano scalare per utilizzare milioni di core di processori simultaneamente senza rallentare. Sono particolarmente interessati a un metodo specifico chiamato dinamica molecolare di secondo tipo Car-Parrinello, che permette al computer di simulare il movimento degli atomi su periodi molto lunghi, come i nanosecondi, mantenendo gli elettroni nel loro stato più stabile. Questo approccio è crucialo perché consente lo studio di sistemi realistici a temperatura finita, come liquidi o solidi, piuttosto che semplici istantanee congelate della materia.

I ricercatori hanno identificato tre grandi tendenze nell'informatica moderna a cui il loro software deve adattarsi: la necessità di un massiccio elaboramento parallelo, l'uso diffuso di unità di elaborazione grafica (GPU) per il calcolo, e la disponibilità di hardware che esegue i calcoli con diversi livelli di precisione. Per affrontare queste sfide, il team ha ristrutturato il modo in cui il proprio codice gestisce i dati. Hanno sviluppato nuovi modi per dividere grandi problemi matematici in blocchi più piccoli e indipendenti che diverse parti di un computer possono risolvere contemporaneamente senza dover costantemente fermarsi per controllare tra loro. Ciò si ottiene organizzando i dati in blocchi fissi assegnati a specifici thread di elaborazione, eliminando i ritardi che solitamente si verificano quando molti processori tentano di accedere alle stesse informazioni simultaneamente. Hanno anche integrato il supporto per le GPU, i potenti chip originariamente progettati per i videogiochi ma ora essenziali per il calcolo scientifico, consentendo al software di scaricare i calcoli più pesanti su questi acceleratori specializzati.

Una parte significativa del loro sforzo riguarda l'affrontare la complessità matematica delle interazioni elettroniche, che sono la parte più costosa di queste simulazioni. Tradizionalmente, calcolare come gli elettroni si respingono a vicenda in un grande sistema richiedeva un numero di passaggi che cresceva in modo esplosivo man mano che il sistema si ingrandiva, rendendo quasi impossibili le simulazioni su larga scala. Il team ha introdotto nuove tecniche per filtrare le interazioni trascurabili, ignorando efficacemente le forze minuscole che non contano, il che riduce drasticamente il costo computazionale. Hanno anche sviluppato metodi per approssimare la densità degli elettroni utilizzando matrici più piccole e facili da calcolare, correggendo i risultati in seguito con un calcolo più preciso ma meno oneroso. Ciò consente di mantenere un'alta accuratezza pur riducendo drasticamente il tempo necessario. Inoltre, stanno esplorando l'uso della "precisione mista", in cui il computer esegue la maggior parte del suo lavoro utilizzando un livello inferiore di accuratezza numerica che è più veloce da calcolare, e poi utilizza una correzione matematica specializzata per garantire che il risultato finale rimanga esatto.

I risultati di questi sviluppi sono già impressionanti. Nelle simulazioni di acqua bulk, i ricercatori hanno dimostrato che il loro codice ottimizzato poteva gestire sistemi fino a un milione di atomi su un supercomputer con decine di migliaia di core. Più recentemente, combinando il loro nuovo metodo submatrizza con lo schema di purificazione della densità elettronica, hanno ottenuto una simulazione record di un sistema contenente più di 100 milioni di atomi. Questo calcolo massiccio è stato eseguito su un supercomputer con una performance sostenuta di 324 petaflops, utilizzando più del 67 percento della potenza teorica totale della macchina. Questa efficienza dimostra che il software può sfruttare efficacemente la velocità bruta dell'hardware moderno per risolvere problemi che prima erano fuori portata.

Guardando al futuro, gli autori vedono un futuro in cui queste simulazioni classiche saranno aumentate da altre tecnologie emergenti. Suggeriscono che l'intelligenza artificiale, specificamente le reti neurali profonde, potrebbe essere utilizzata per prendere decisioni durante una simulazione, come determinare se un calcolo complesso debba essere eseguito da zero o se una rete addestrata possa fornire una risposta sufficientemente accurata. Immaginano anche approcci ibridi in cui un computer quantistico, che opera sui principi della meccanica quantistica, potrebbe lavorare insieme al supercomputer classico per risolvere parti specifiche del problema, come il calcolo del comportamento degli elettroni in un modo che è attualmente troppo difficile per le macchine classiche. Sebbene queste applicazioni future siano ancora in fase di pianificazione, l'attuale lavoro su CP2K getta le basi essenziali, assicurando che il software sia pronto a evolversi insieme all'hardware, trasformando la promessa del calcolo exascale in una realtà pratica per comprendere il mondo materiale.