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

Um das Universum auf seiner fundamentalsten Ebene zu verstehen, wenden Physiker oft eine Methode an, bei der Raum und Zeit nicht als glatter, kontinuierlicher Stoff, sondern als ein Gitter aus winzigen, diskreten Punkten behandelt werden. Stellen Sie sich ein riesiges, unsichtbares Schachbrett vor, das sich durch den Kosmos erstreckt, wobei jeder Schnittpunkt eine Information über die Kräfte enthält, die die Materie zusammenhalten. Dies ist der Kern der Gittereichtheorie (Lattice Gauge Theory), ein leistungsfähiges Werkzeug, das verwendet wird, um das Verhalten von subatomaren Teilchen und die sie steuernden Kräfte zu simulieren, insbesondere die starke Kraft, die Atomkerne zusammenhält. Indem Wissenschaftler das Universum in diese handhabbaren Stücke zerlegen, können sie massenhafte Computer nutzen, um die Pfade von Teilchen nachzuverfolgen und zu berechnen, wie sie interagieren – indem sie die Gesetze der Physik effektiv in umgekehrter Richtung laufen lassen, um zu sehen, was daraus entsteht. Die Hardware, die zur Ausführung dieser Simulationen verwendet wird, ist jedoch unglaublich vielfältig geworden. Heutige Supercomputer sind ein Flickenteppich aus verschiedenen Prozessoren, von Standard-Computerchips bis hin zu spezialisierten Grafikkarten, von denen jede eine andere technische Sprache spricht und ihren eigenen einzigartigen Satz von Anweisungen benötigt, um effizient zu arbeiten.

Diese Vielfalt stellt Forscher vor ein erhebliches Problem. Um die bestmögliche Leistung aus einer bestimmten Maschine herauszuholen, müssen sie ihren Simulationscode oft von Grund auf neu für jeden anderen Typ von Prozessor schreiben, den sie verwenden möchten. Das Pflegen separater Versionen desselben Algorithmus für verschiedene Computer ist langsam, teuer und fehleranfällig. Die Frage, vor der die wissenschaftliche Gemeinschaft steht, lautet, ob es möglich ist, eine einzige Version dieses komplexen Simulationscodes zu schreiben, der auf jedem modernen Computer – von einem einzelnen Prozessor bis hin zu einem massiven Supercomputer – effizient laufen kann, ohne dabei an Geschwindigkeit einzubüßen. Wei Sun, ein Forscher am Institut für Hochenergiephysik in Peking, hat diese Frage mit einem neuen Softwarewerkzeug namens kwqft beantwortet. Dieses Programm demonstriert erfolgreich, dass es möglich ist, einen einzigen, universellen Code zur Simulation der starken Kraft zu erstellen, der auf jeder Hardware, auf der er läuft, genauso gut funktioniert wie spezialisierter, maschinenspezifischer Code.

Der Kern dieser Errungenschaft liegt in einem klugen Ansatz des Softwaredesigns, der die Architektur des Computers als eine Variable und nicht als eine feste Beschränkung behandelt. Die Forscher bauten ihre Simulation unter Verwendung eines modernen Programmierrahmens (Framework), der als universeller Übersetzer fungiert. Anstatt separate Anweisungen für eine Grafikkarte, einen Standardprozessor oder einen spezialisierten Chip zu schreiben, verfassten sie einen Satz von Anweisungen, den das Framework automatisch an die Maschine anpasst, auf der er läuft. Dies ermöglicht es demselben Code, auf einer Vielzahl von Systemen zu laufen, einschließlich solcher von verschiedenen Herstellern, ohne dass die Forscher den Quellcode erneut anfassen müssen. Die Software ist so konzipiert, dass sie flexibel ist, sodass Wissenschaftler die Anzahl der Dimensionen in ihrer Simulation oder die spezifischen Eigenschaften der untersuchten Teilchen ändern können, indem sie lediglich einige Einstellungen vor dem Start des Programms anpassen, anstatt das gesamte Programm neu zu schreiben.

Um zu beweisen, dass dieser universelle Ansatz funktioniert, unterzog das Team ihre Software strengen Tests auf einer breiten Palette von Hardware. Sie führten Simulationen auf einer leistungsstarken Grafikkarte von NVIDIA, einem spezialisierten Chip von Hygon und verschiedenen Typen von Zentralprozessoren durch, darunter fortschrittliche Modelle von Huawei und Intel. In jedem Fall lieferte die Software Ergebnisse, die den bekannten, exakten Lösungen für einfachere Versionen des Problems entsprachen und mit zuvor veröffentlichten Daten für komplexere Szenarien übereinstimmten. Die Forscher verifizierten, dass die Simulation das Verhalten der starken Kraft in zwei, drei und vier Dimensionen sowie für verschiedene Arten von Teilchengruppen korrekt reproduzierte, wodurch bestätigt wurde, dass der universelle Code im Prozess der Portabilität keine Genauigkeit verliert.

Die Leistungsergebnisse waren besonders beeindruckend. Beim Betrieb auf einer High-End-Grafikkarte erreichte die universelle Software nahezu die gleiche Geschwindigkeit wie eine Version des Codes, die speziell für diese Karte optimiert wurde, wobei sie bei größeren Simulationen über 90 Prozent der Effizienz des spezialisierten Codes erreichte. Auf einem massiven Supercomputer in Shenzhen, bekannt als LineShine, skalierte die Software hoch, um gleichzeitig über 512 Knoten, oder einzelne Recheneinheiten, zu laufen. Während die Forscher mehr Rechenleistung hinzufügten, beschleunigte sich die Simulation signifikant, was zeigte, dass die Software die komplexe Kommunikation bewältigen kann, die erforderlich ist, um tausende von Prozessoren in Harmonie zusammenarbeiten zu lassen. Darüber hinaus war die Software auf der neuesten Generation von Prozessoren, die für Hochgeschwindigkeits-Vektorberechnungen konzipiert sind, in der Lage, mehrere Punkte auf dem Gitter gleichzeitig zu verarbeiten, was die Simulation im Vergleich zu einer Standardversion desselben Codes um den Faktor drei beschleunigte.

Die Auswirkungen dieser Arbeit reichen weit über einen einzelnen erfolgreichen Test hinaus. Indem sie bewiesen haben, dass ein einziger Codebase eine wettbewerbsfähige Leistung über eine so vielfältige Hardware-Landschaft hinweg liefern kann, haben die Forscher eine große Barriere für den wissenschaftlichen Fortschritt beseitigt. Wissenschaftler müssen nicht mehr zwischen dem Schreiben von leicht zu wartendem Code und schnellem Code wählen; sie können nun einmal schreiben und überall ausführen, in der Gewissheit, dass die Software sich an die jeweilige Maschine anpassen wird. Dies öffnet die Tür für komplexere Simulationen des Universums und ermöglicht es Forschern, Fragen über die frühen Momente des Kosmos und die Natur der Materie mit größerer Leichtigkeit und in einem größeren Maßstab als je zuvor zu erforschen. Der Erfolg von kwqft legt nahe, dass die Zukunft des Hochleistungsrechnens in der Physik möglicherweise nicht darin liegt, ständig Code für neue Maschinen neu zu schreiben, sondern in der Schaffung flexibler, intelligenter Werkzeuge, die in der Lage sind, die sich verändernde technologische Landschaft eigenständig zu navigieren.