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

Die große Suche nach der Fusion und die Sprachbarriere

Stellen Sie sich vor, Sie versuchen, einen Stern direkt hier auf der Erde zu bauen. Das ist der Traum der Kernfusion – jener Prozess, der die Sonne antreibt und das Versprechen sauberer, unbegrenzter Energie in sich trägt. Um dies zu verwirklichen, sperren Wissenschaftler superheißes Plasma – eine wirbelnde Suppe aus geladenen Teilchen – in magnetische Käfige, sogenannte Tokamaks. Doch dieses Plasma ist chaotisch; es ist, als versuche man, den Pfad von einer Milliarde winziger, hyperaktiver Bienen vorherzusagen, die ständig gegeneinanderstoßen und die Richtung ändern. Um zu verstehen, wie man dieses Plasma stabil und heiß genug halten kann, um eine Fusion zu ermöglichen, nutzen Wissenschaftler massive Computersimulationen. Diese Simulationen fungieren als virtuelle Laboratorien, die verfolgen, wie sich Teilchen bewegen und wie sich elektromagnetische Felder verdrehen und wenden.

Eines der vertrauenswürdigsten Werkzeuge für diese Aufgabe ist ein Code namens ORB5. Stellen Sie sich ORB5 als einen hochqualifizierten, altmodischen Chefkoch vor, der seit Jahrzehnten dasselbe perfekte Fusionsrezept kocht. Dieser Koch spricht jedoch nur eine Sprache: Fortran, eine Programmiersprache, die in den 1950er Jahren erfunden wurde. Obwohl Fortran hervorragend für die Mathematik ist, wird es allmählich wie ein Oldtimer: schwer zu reparieren, schwer aufzurüsten und schwierig, auf modernen Straßen zu fahren (wie den neuesten, schnellsten Supercomputern mit leistungsstarken Grafikkarten). Während sich die Computerwelt in Richtung neuer Hardware verschiebt, wird es zu einem Kampf, diesen „Oldtimer“ effizient am Laufen zu halten. Die Frage ist nicht, ob das Rezept gut ist – das ist es –, sondern wie wir dieses Rezept in eine moderne Sprache übersetzen, damit es auf den heutigen schnellsten Motoren laufen kann, ohne den Geschmack des Gerichts zu verändern.

Das Übersetzungsprojekt: ORB5X

Dieses Paper stellt ORB5X vor, eine kühne neue Version dieses berühmten Fusionscodes. Das Team hat nicht versucht, ein neues Rezept zu erfinden oder die Physik des Plasmas zu verändern. Stattdessen agierten sie als Meisterübersetzer. Sie nahmen den ursprünglichen Fortran-Code und schrieben das gesamte Programm mithilfe von „Agentic AI“ (intelligenten Computer-Assistenten, die Code schreiben und debuggen) in C++ um, eine moderne Programmiersprache.

Warum ist das wichtig? Stellen Sie sich vor, Sie haben einen brillanten Bauplan für ein Haus, aber er ist auf einem Stück Papier gezeichnet, das verblasst und nur auf einen ganz bestimmten Typ von Zeichentisch passt. ORB5X ist wie das exakt gleiche Neuzeichnen dieses Bauplans auf einem digitalen Tablet, das auf jedem Computer, jedem Tablet oder sogar einem Supercomputer im Weltraum funktioniert. Das Ziel war es, die „mathematische Seele“ des ursprünglichen Codes zu bewahren und ihn gleichzeitig „performance-portabel“ zu machen. Das bedeutet, dass der neue Code genauso gut auf einem Standard-Laptop, einem leistungsstarken Desktop oder einem massiven Supercomputer mit Tausenden von Grafikkarten laufen kann, ohne dass er für jede Maschine komplett neu geschrieben werden muss.

Wie sie es gemacht haben: Der KI-Co-Pilot

Die Übersetzung war nicht nur ein einfaches Kopieren und Einfügen; es war eine sorgfältige, schrittweise Rekonstruktion. Die Forscher nutzten einen Workflow, bei dem menschliche Experten und KI-Agenten zusammenarbeiteten. Die KI fungierte wie ein unermüdlicher Co-Pilot: Sie betrachtete ein Stück des alten Fortran-Codes, schlug vor, wie man diesen in modernem C++ schreibt, und markierte potenzielle Fallen, an denen die Mathematik falsch ablaufen könnte. Die menschlichen Wissenschaftler überprüften dann jeden Vorschlag, um sicherzustellen, dass sich der neue Code exakt so verhält wie der alte.

Sie hielten die beiden Versionen nebeneinander, wie ein zweisprachiges Wörterbuch. Wenn der alte Code eine Datei namens fields.F90 hatte, besaß die neue fields.cpp. Dies ermöglichte es ihnen, jede einzelne Berechnung zu überprüfen. Sie verwendeten ein modernes Toolkit namens Kokkos, das wie ein Universaladapter wirkt. Es erlaubt dem Code, mit verschiedenen Arten von Computerprozessoren (wie Intel-CPUs oder NVIDIA-GPUs) automatisch zu kommunizieren, ohne dass der Programmierer separate Anweisungen für jeden einzelnen schreiben muss.

Die Ergebnisse: Maschinengenauigkeit

Das Team unterzog ORB5X drei berühmten „Stresstests“, die unterschiedliche Plasmaturbulenzen simulieren:

1. ITG (Ion Temperature Gradient): Simulation der Wärmeübertragung durch das Plasma.
2. ITPA: Ein komplexer Test, der schnelle Teilchen und magnetische Wellen beinhaltet.
3. Chirping: Beobachtung, wie sich die Frequenz von Wellen ändert, während sie mit energetischen Teilchen interagieren.

Die Ergebnisse waren beeindruckend. Als sie die Ausgabe des neuen C++-Codes mit dem ursprünglichen Fortran-Code verglichen, stimmten die Zahlen mit Maschinengenauigkeit überein. Das bedeutet, dass der Unterschied zwischen den beiden so winzig war, dass er im Wesentlichen Null war, begrenzt nur durch die Präzision der Mathematik des Computers. Ob sie die Simulation auf einem persönlichen Laptop oder einem massiven Supercomputer wie LUMI oder CINECA ausführten, die Physik blieb identisch.

Warum das eine große Sache ist

Das Paper bestätigt, dass ORB5X ein Erfolg ist, setzt aber auch realistische Erwartungen. Der neue Code ist auf jeder Maschine noch nicht unbedingt schneller als die ursprüngliche Fortran-Version; tatsächlich ist der ursprüngliche Code in einigen Fällen immer noch etwas schneller, da er über Jahrzehnte feinjustiert wurde. Der wahre Sieg ist jedoch die Portabilität. Der ursprüngliche Fortran-Code wird zu einem „Legacy“-System, das auf neuer Hardware schwer auszuführen ist. ORB5X, gebaut mit modernen Werkzeugen, kann problemlos auf persönlichen Laptops (sogar Macs mit Apple M4-Chips) und den fortschrittlichsten Supercomputern der Welt kompiliert und ausgeführt werden.

Die Autoren merken vorsichtig an, dass dies Version 1.0 ist. Es ist ein solides Fundament, kein fertiges Produkt. Sie planen, es in Zukunft noch schneller zu machen, indem sie die Zeit reduzieren, die benötigt wird, um Daten zwischen dem Speicher des Computers und seinem Prozessor zu bewegen, und sie hoffen, es schließlich einzusetzen, um noch komplexere Formen magnetischer Käfige (wie Stellaratoren) zu simulieren. Aber für den Moment haben sie erfolgreich bewiesen, dass man einen jahrzehntealten, hochkomplexen wissenschaftlichen Code nehmen, mit Hilfe von KI in eine moderne Sprache übersetzen und dabei die Physik perfekt intakt halten kann. Es ist eine Brücke zwischen der Vergangenheit und der Zukunft der Fusionsforschung, die sicherstellt, dass die Suche nach sauberer Energie auf den schnellsten Motoren weiterlaufen kann, die die Welt zu bieten hat.