CATAPULT: A CUDA-Accelerated Timestepper for Alpha Particles Using Local Tricubics

Dit artikel introduceert CATAPULT, een door CUDA-versnelling aangedreven tijdstapcode voor Monte Carlo-simulaties van alfadeeltjesconfinement in stellaratoren, die aanzienlijk sneller is dan bestaande CPU-implementaties en evenwichtsmagnetische velden evenals Shear Alfven-golven kan verwerken.

De kern

Stel je voor dat je een gigantisch, futuristisch magnetisch kussen bouwt om de warmte van de zon op aarde te vangen. Dit is wat wetenschappers doen met een apparaat dat een stellarator heet. Het doel is om energie op te wekken door atoomkernen te laten versmelten (fusie).

Het probleem? De deeltjes die bij deze versmelting ontstaan (de "alfa-deeltjes") zijn razendsnel en willen graag ontsnappen. Als ze ontsnappen, koelt het plasma af en stopt de energieproductie. Om te weten of een stellarator goed werkt, moeten wetenschappers simuleren hoe miljarden van deze deeltjes zich gedragen.

Hier komt CATAPULT in beeld. Laten we uitleggen wat dit is, zonder ingewikkelde wiskunde.

1. Het Probleem: Het Spooktrein-Experiment

Stel je voor dat je een spooktrein hebt met 32.000 passagiers (de deeltjes). Je wilt weten hoeveel passagiers uit de trein vallen voordat de rit voorbij is.

• De oude manier (CPU): Je hebt 128 mensen nodig die elk een passagier volgen. Ze lopen allemaal in een rij en noteren waar de passagier is. Dit gaat redelijk snel, maar het kost veel tijd en energie.
• Het probleem: De treinbaan (het magnetische veld) is niet perfect glad; hij heeft bochten, gaten en soms zelfs trillingen. De passagiers moeten heel precies worden gevolgd, anders val je eruit.

2. De Oplossing: CATAPULT (De Super-Snelle GPU)

CATAPULT is een nieuwe software die draait op GPU's (de krachtige grafische kaarten die je ook in gaming-computers vindt, maar dan in datacenters).

In plaats van 128 mensen die één voor één lopen, zet CATAPULT duizenden kleine robotjes (de "threads" op de GPU) aan het werk.

• De Analogie: Stel je voor dat de oude manier een lange rij mensen is die een touw vasthouden. CATAPULT is als een zwerm bijen die tegelijkertijd duizenden bloemen bezoekt.
• Het resultaat: CATAPULT is 5 tot 60 keer sneller dan de oude methoden. Wat voor de oude computer een hele dag duurde, doet CATAPULT in een uur.

3. Hoe werkt het? (De "Lokale Tricubics")

Om te weten waar de deeltjes naartoe gaan, moet de computer de vorm van het magnetische veld kennen.

• Het Net: De wetenschappers maken een 3D-rooster (een soort digitaal gaas) rondom de stellarator.
• De Vulling: In elk vakje van dit rooster weten ze hoe het magnetische veld eruitziet.
• De Kunstgreep: Als een deeltje van het ene vakje naar het andere springt, moet de computer snel schatten wat er tussenin gebeurt. CATAPULT gebruikt een slimme wiskundige truc (lokale "tricubische" interpolatie).
  • Vergelijking: Het is alsof je een kaart hebt met alleen de hoogte van de bergen op de hoekpunten. Als je ergens in het midden loopt, rekent CATAPULT heel snel en nauwkeurig uit hoe steil de helling is, zodat je niet struikelt.

4. Waarom is dit zo belangrijk?

Omdat CATAPULT zo snel is, kunnen wetenschappers twee dingen doen die eerder te duur of te langzaam waren:

1. Meer details (Hogere Fideliteit): Ze kunnen het rooster veel fijner maken. In plaats van een ruwe schets van de magnetische baan, krijgen ze een HD-foto. Hierdoor zien ze kleine foutjes in het ontwerp die de oude computer over het hoofd zag.
2. Meer deeltjes: Ze kunnen niet 32.000, maar miljoenen deeltjes tegelijk volgen. Dit geeft een veel betrouwbaarder antwoord op de vraag: "Vallen er veel passagiers uit de trein?"

5. De "Truc" met de Koers

Soms komen de deeltjes heel dicht bij de rand van de trein (de "laatste gesloten flux-oppervlakte"). Op die rand wordt de wiskunde soms raar (zoals een kompas dat draait als je precies op de Noordpool staat).
CATAPULT heeft een slimme oplossing bedacht: als een deeltje de rand raakt, doet de software alsof het deeltje nog net binnenin zit, en gebruikt de laatste bekende gegevens. Dit voorkomt dat de simulatie vastloopt, net als een GPS die je blijft leiden als je even uit het bereik van de satellieten komt.

Conclusie

CATAPULT is als het geven van een supersnelheidsmotor aan een team van onderzoekers.

• Vroeger: Ze moesten wachten tot de computer langzaam rekende, en moesten vaak met grove schattingen werken.
• Nu: Met CATAPULT kunnen ze in een oogwenk zien of een nieuw ontwerp voor een fusiereactor veilig is.

Dit betekent dat we dichter bij de droom komen van oneindige, schone energie, omdat we de "trucs" van de zon beter kunnen nabootsen en optimaliseren dan ooit tevoren. De code is zelfs gratis beschikbaar voor iedereen die wil meehelpen aan deze toekomst!

Technische samenvatting

Titel

CATAPULT: Een CUDA-versnelde tijdstap-gebruiker voor alfadeeltjes met behulp van lokale tricubische interpolatie.

1. Probleemstelling

In magnetische veldbeperkingssystemen voor kernfusie (zoals stellaratoren) is het cruciaal om het gedrag van hoge-temperatuur ionen, en specifiek alfadeeltjes, te begrijpen. Deze deeltjes moeten worden ingesloten om energie binnen het plasma te houden en de temperatuur voor fusiereacties te handhaven.

• Huidige aanpak: De evaluatie van de insluiting gebeurt doorgaans met Monte Carlo-methoden gekoppeld aan ODE-tijdstappers (zoals in softwarepakketten SIMSOPT, SIMPLE, ORBIT, ASCOT, TAPaS).
• De uitdaging: Deze methoden vereisen het traceren van duizenden tot miljoenen deeltjes om statistisch betrouwbare schattingen te krijgen van verliezen (bijv. de fractie deeltjes die het plasma verlaat). Bestaande parallelle CPU-implementaties zijn vaak te traag om hoge resoluties (fijne roosters) en hoge nauwkeurigheid (kleine tijdstaptoleranties) haalbaar te maken binnen redelijke rekentijden. Dit beperkt de fideliteit van de simulaties.

2. Methodologie

De auteurs introduceren CATAPULT, een GPU-versnelde implementatie die de berekeningen verplaatst van CPU naar NVIDIA GPUs (specifiek getest op A100).

• Fysiek Model:

  • De beweging van deeltjes wordt beschreven door de Littlejohn-Lagrangiaan, die de driftbeweging in een magnetisch veld ($B_0$) modelleert.
  • De simulatie ondersteunt zowel Boozer-coördinaten (flux-coördinaten aangepast aan de magnetische geometrie) als Cartesiaanse coördinaten.
  • Deeltjes worden getraceerd totdat ze de laatste gesloten fluxoppervlak verlaten of een maximale tijd $T$ bereiken.

• Interpolatie van het Magnetische Veld:

  • Het magnetische veld wordt voorgesteld op een regelmatig rooster.
  • CATAPULT gebruikt lokale tricubische interpolatie (op een $4 \times 4 \times 4$ grid per cel) om een dichte representatie van het veld te verkrijgen.
  • Dit zorgt voor continuïteit van het veld tussen cellen, hoewel differentieerbaarheid niet gegarandeerd is aan de grenzen.

• Tijdstap-algoritme:

  • Er wordt gebruikgemaakt van de Dormand-Prince 5 (DP5) adaptieve tijdstap-methode, identiek aan de implementatie in SIMSOPT/Boost.
  • GPU-architectuur:
    • Een blok van 64 threads (2 warps op een A100) traceert een groep van 8 deeltjes.
    • Gegevens worden geoptimaliseerd voor GPU-cache: constante data (zoals DP5-coëfficiënten) wordt in constant memory geplaatst.
    • Interpolatiedata wordt in rij-volgorde (row-major) opgeslagen om gecoalesceerde geheugentoegang te maximaliseren. Dit vermindert het verkeer van instructies door het voorkomen van verspreide lezingen.
  • Foutbeheersing:
    • Voor Boozer-coördinaten wordt een pseudocartesiaanse transformatie gebruikt voor de foutschatting om singulariteiten op de as ($s=0$) te vermijden.
    • De relatieve schaling van de parallelle snelheid ($v_{\parallel}$) wordt aangepakt door de absolute tolerantie ($atol$) te vermenigvuldigen met de totale snelheid.

• Extrapolatie:

  • Wanneer deeltjes dicht bij de laatste gesloten fluxoppervlak komen ($s > 1$), gebruikt CATAPULT de dichtstbijzijnde interpolatiecel-data om de afgeleiden te schatten, in plaats van stil te vallen of fouten te genereren.

3. Belangrijkste Bijdragen

1. Versnelling: Een aanzienlijke snelheidswinst ten opzichte van bestaande parallelle CPU-implementaties door het gebruik van CUDA en GPU-parallelisme.
2. Schalbaarheid: Het vermogen om hogere roosterresoluties te hanteren zonder geheugenproblemen, omdat alle threads op de GPU één enkele kopie van het interpolatierooster delen (in tegenstelling tot CPU-implementaties die het rooster per thread kopiëren).
3. Flexibiliteit: Ondersteuning voor zowel vacuümvelden als velden met Shear Alfven-golven (SAWs), wat belangrijk is voor realistische plasma-simulaties.
4. Open Source: De code is geïntegreerd in het firm3d Python-pakket.

4. Resultaten

De auteurs hebben CATAPULT getest op verschillende stellarator-configuraties (ATEN, HSX, NCSX, QA, QH) geschaald naar ARIES-CS afmetingen.

• Snelheidswinst:
  • In vacuümconfiguraties (zowel Boozer als Cartesiaans) is CATAPULT gemiddeld minimaal 5 tot 10 keer sneller dan 128 CPU-kernen op een Perlmutter-node.
  • Bij het traceren in aanwezigheid van Shear Alfven-golven is de snelheidswinst nog groter: 40 tot 60 keer sneller dan de CPU.
• Fideliteit en Schaling:
  • De runtime schaalt lineair met het aantal deeltjes.
  • De rekentijd wordt voornamelijk bepaald door de tijdstap-tolerantie (nauwkeurigheid) en minder door de roosterresolutie, dankzij de efficiënte cache-lokalisatie van de interpolatie.
  • Bij een roosterresolutie van 20 cellen en een tolerantie van $10^{-4}$ of lager, convergeert de geschatte verliesfractie goed (met 32.768 deeltjes).
• Correctheid: De resultaten van de GPU-versie tonen een hoge consistentie met de CPU-versie (SIMSOPT), waarbij de lijnen in de convergentiegrafieken overlappen.

5. Betekenis en Toekomstperspectief

CATAPULT opent nieuwe mogelijkheden voor fusieonderzoek:

• Hogere Fideliteit: Onderzoekers kunnen nu simulaties uitvoeren met fijnere roosters en strengere toleranties zonder onrealistische rekentijden.
• Complexere Fysica: De hoge doorvoer maakt het haalbaar om complexere fysica te modelleren, zoals volledige orbit-tracing (in plaats van gidscentrum-benadering) en botsingen.
• Optimalisatie: Het stelt directe optimalisatie van de deeltjesinsluiting en het berekenen van wandbelastingen (wall-loads) voor alfadeeltjes op grotere schaal mogelijk.
• Toekomstig werk: Er is ruimte voor verdere optimalisatie voor andere GPU-architecturen, het porteren naar andere GPU-bibliotheken en het verbeteren van interpolatie- en tijdstapschema's voor nog hogere numerieke nauwkeurigheid.

Kortom, CATAPULT is een doorbraak in de computationele efficiëntie voor het modelleren van alfadeeltjes in stellaratoren, waardoor nauwkeurigere en uitgebreidere studies naar fusie-insluiting mogelijk worden.