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

A Grande Caçada pela Fusão e a Barreira Linguística

Imagine tentar construir uma estrela bem aqui na Terra. Esse é o sonho da fusão nuclear, o processo que alimenta o sol, prometendo energia limpa e ilimitada. Para fazer isso acontecer, cientistas aprisionam plasma superquente — uma sopa giratória de partículas carregadas — em gaiolas magnéticas chamadas tokamaks. Mas esse plasma é caótico; é como tentar prever o caminho de um bilhão de pequenas abelhas hiperativas que constantemente colidem umas com as outras e mudam de direção. Para entender como manter esse plasma estável e quente o suficiente para fundir, os cientistas usam simulações computacionais massivas. Essas simulações atuam como laboratórios virtuais, rastreando como as partículas se movem e como os campos eletromagnéticos giram e se retorcem.

Uma das ferramentas mais confiáveis para este trabalho é um código chamado ORB5. Pense no ORB5 como um chef mestre altamente qualificado e de estilo clássico, que tem cozinhado a mesma receita perfeita de fusão há décadas. No entanto, este chef fala apenas uma língua: Fortran, uma linguagem de programação inventada na década de 1950. Embora o Fortran seja excelente para a matemática, ele está se tornando um pouco como um carro vintage: difícil de consertar, difícil de atualizar e difícil de dirigir em estradas modernas (como os supercomputadores mais novos e rápidos com placas de vídeo poderosas). À medida que o mundo da computação se desloca para novos hardwares, manter este "carro vintage" funcionando de forma eficiente está se tornando uma luta. A questão não é se a receita é boa — ela é — mas como traduzimos essa receita para uma linguagem moderna para que possa rodar nos motores mais rápidos de hoje sem mudar o sabor do prato?

O Projeto de Tradução: ORB5X

Este artigo apresenta o ORB5X, uma nova e ousada versão desse famoso código de fusão. A equipe não tentou inventar uma nova receita ou mudar a física do plasma. Em vez disso, eles atuaram como mestres tradutores. Eles pegaram o código Fortran original e, com a ajuda de "IA Agêntica" (assistentes de computador inteligentes que escrevem e depuram código), reescreveram o programa inteiro em C++, uma linguagem de programação moderna.

Por que isso importa? Imagine que você tem a planta brilhante de uma casa, mas ela está desenhada em um pedaço de papel que está desbotando e só cabe em um tipo específico de mesa de desenho. O ORB5X é como redesenhar essa exata mesma planta em um tablet digital que funciona em qualquer computador, qualquer tablet ou até mesmo um supercomputador no espaço. O objetivo era preservar a "alma matemática" do código original enquanto o tornava "desempenho-portátil" (performance-portable). Isso significa que o novo código pode rodar tão bem em um laptop padrão, em um desktop poderoso ou em um supercomputador massivo com milhares de placas de vídeo, sem precisar de uma reescrita completa para cada máquina.

Como Eles Fizeram: O Copiloto de IA

A tradução não foi apenas um trabalho simples de copiar e colar; foi uma reconstrução cuidadosa, passo a passo. Os pesquisadores usaram um fluxo de trabalho onde especialistas humanos e agentes de IA trabalharam juntos. A IA atuou como um copiloto incansável: ela analisava um bloco do antigo código Fortran, sugeria como escrevê-lo em C++ moderno e sinalizava armadilhas potenciais onde a matemática poderia falhar. Os cientistas humanos então revisavam cada sugestão, garantindo que o novo código se comportasse exatamente como o antigo.

Eles mantiveram as duas versões lado a lado, como um dicionário bilíngue. Se o código antigo tinha um arquivo chamado fields.F90, o novo tinha fields.cpp. Isso permitiu que verificassem cada um dos cálculos. Eles usaram um conjunto de ferramentas moderno chamado Kokkos, que atua como um adaptador universal. Ele permite que o código converse com diferentes tipos de processadores de computador (como CPUs Intel ou GPUs NVIDIA) automaticamente, sem que o programador precise escrever instruções separadas para cada um.

Os Resultados: Precisão de Máquina

A equipe submeteu o ORB5X a três "testes de estresse" famosos que simulam diferentes turbulências de plasma:

1. ITG (Gradiente de Temperatura de Íons): Simulando como o calor se move através do plasma.
2. ITPA: Um teste complexo envolvendo partículas de alta velocidade e ondas magnéticas.
3. Chirping: Observando como a frequência das ondas muda conforme elas interagem com partículas energéticas.

Os resultados foram impressionantes. Quando compararam a saída do novo código C++ contra o código Fortran original, os números coincidiram com precisão de máquina. Isso significa que a diferença entre os dois era tão minúscula que era essencialmente zero, limitada apenas pela precisão da matemática do computador. Quer tenham executado a simulação em um laptop pessoal ou em um supercomputador massivo como o LUMI ou CINECA, a física permaneceu idêntica.

Por Que Isso é Importante

O artigo confirma que o ORB5X é um sucesso, mas também estabelece expectativas realistas. O novo código não é necessariamente mais rápido que a versão Fortran original em todas as máquinas ainda; de fato, o código original ainda é ligeiramente mais rápido em alguns casos porque foi refinado ao longo de décadas. No entanto, a verdadeira vitória é a portabilidade. O código Fortran original está se tornando um sistema "legado", difícil de rodar em novos hardwares. O ORB5X, construído com ferramentas modernas, pode ser compilado e executado facilmente em laptops pessoais (incluindo Macs com chips Apple M4) e nos computadores superavançados do mundo.

Os autores ressaltam cuidadosamente que esta é a versão 1.0. É uma base sólida, não um produto finalizado. Eles planejam torná-lo ainda mais rápido no futuro, reduzindo o tempo necessário para mover dados entre a memória do computador e seu processador, e esperam eventualmente usá-lo para simular formas ainda mais complexas de gaiolas magnéticas (como stellarators). Mas, por enquanto, eles provaram com sucesso que é possível pegar um código científico de décadas, altamente complexo, traduzi-lo para uma linguagem moderna com a ajuda da IA e manter a física perfeitamente intacta. É uma ponte construída entre o passado e o futuro da pesquisa de fusão, garantindo que a busca por energia limpa possa continuar rodando nos motores mais rápidos que o mundo tem a oferecer.