⚛️ nuclear theory

HPRMAT: A high-performance R-matrix solver with GPU acceleration for coupled-channel problems in nuclear physics

Este artigo apresenta o HPRMAT, um solver de matriz R de alto desempenho e acelerado por GPU que utiliza aritmética de precisão mista e álgebra linear otimizada para alcançar acelerações significativas em relação aos códigos legados, permitindo, assim, cálculos de física nuclear de canais acoplados em larga escala em estações de trabalho convencionais de consumo.

Autores originais: Jin Lei

Publicado 2026-09-07

📖 5 min de leitura🧠 Leitura aprofundada

CC BY 4.0

Autores originais: Jin Lei

Artigo original sob licença CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/). ✨ Esta é uma explicação gerada por IA do artigo abaixo. Não foi escrita nem endossada pelos autores. Para precisão técnica, consulte o artigo original. Ler aviso legal completo

No mundo microscópico do núcleo atômico, as partículas não simplesmente ricocheteiam umas nas outras como bolas de bilhar; elas interagem através de uma complexa teia de forças que podem uni-las ou dispersá-las. Para compreender essas interações, os físicos utilizam um arcabouço matemático conhecido como método R-matrix. Essa abordagem divide o espaço ao redor de um núcleo em duas regiões: uma zona interna, onde as partículas estão densamente compactadas e interagindo, e uma zona externa, onde elas se movem livremente. O desafio reside em calcular exatamente como as partículas se comportam nessa zona interna, onde a matemática se torna incrivelmente densa e difícil de resolver. Durante décadas, pesquisadores confiaram em métodos computacionais padrão para decifrar essas equações, mas, à medida que o número de partículas em interação e os estados de energia aumentam, os cálculos tornam-se tão massivos que desaceleram até quase parar, muitas vezes exigindo acesso a supercomputadores caros e especializados que estão fora do alcance de muitos laboratórios universitários.

Um pesquisador da Universidade de Tongji desenvolveu agora uma nova ferramenta chamada HPRMAT, que acelera dramaticamente esses cálculos, tornando a física nuclear de alto nível acessível em computadores desktop comuns. O cerne dessa inovação é uma mudança na forma como o computador resolve os sistemas massivos de equações necessários para descrever o espalhamento nuclear. Em vez de usar o método tradicional e lento de inverter uma matriz gigante — um processo semelhante a tentar resolver um quebra-cabeça trabalhando de trás para frente a partir da imagem final —, o novo software utiliza uma abordagem mais direta que atravessa o problema de forma eficiente. Ao combinar esse método direto com unidades de processamento gráfico modernas, os tipos de chips poderosos encontrados em computadores de jogos de alto desempenho, o software alcança um nível de velocidade que era anteriormente considerado impossível para este tipo de trabalho.

O software foi testado em uma placa de vídeo de consumo, especificamente uma NVIDIA RTX 3090, que é amplamente disponível e custa uma fração do preço dos servidores especializados de data center tipicamente usados para esse tipo de tarefa pesada. Em testes envolvendo uma matriz de tamanho 25.600, uma escala comum para reações nucleares complexas, o novo solver alimentado por GPU completou o cálculo em apenas 3,5 segundos. Em contraste, o melhor método tradicional rodando em um processador multi-core potente levou 52 segundos, e o código legado antigo, que muitos físicos ainda utilizam, levou mais de dois minutos. Isso representa uma aceleração de aproximadamente 15 vezes em comparação com o método moderno otimizado de CPU e 41 vezes em comparação com o código antigo. Os pesquisadores demonstraram que essa velocidade não vem à custa da precisão; os resultados corresponderam aos dados de referência estabelecidos com um grau de precisão que está bem dentro dos limites do erro experimental, o que significa que as previsões físicas permanecem confiáveis.

Um dos obstáculos mais significativos que o software supera é a limitação das placas de vídeo de consumo, que são projetadas para serem incrivelmente rápidas em cálculos de precisão simples, mas são intencionalmente desaceleradas para o trabalho de precisão dupla, que geralmente é necessário para a precisão científica. A nova ferramenta navega habilmente por isso, realizando a maior parte do trabalho pesado no modo de precisão simples rápida e, em seguida, utilizando uma etapa de refinamento para corrigir quaisquer pequenos erros, garantindo que o resultado final seja tão preciso quanto se tivesse sido calculado inteiramente no modo mais lento e preciso. Essa estratégia permite que um único computador desktop lide com problemas que anteriormente exigiam um cluster de supercomputadores. Para os casos raros em que um problema é grande demais até para a maior placa de consumo, o software pode distribuir o trabalho entre múltiplas placas, expandindo o limite do que pode ser resolvido em um desktop para tamanhos de matriz de 76.800.

O impacto deste trabalho estende-se além da velocidade bruta; ele democratiza o acesso à computação de alto desempenho na física nuclear. Ao tornar esses cálculos viáveis em estações de trabalho padrão, a ferramenta remove a barreira de precisar reservar tempo em instalações de supercomputação caras e centralizadas. Pesquisadores podem agora realizar simulações de grande escala de reações nucleares complexas, como a fragmentação de núcleos fracamente ligados, diretamente em seus próprios laboratórios. O software foi projetado para ser um substituto contínuo para as ferramentas existentes, permitindo que os cientistas mudem para este motor mais rápido sem reescrever seus próprios códigos de pesquisa. Ele suporta múltiplas linguagens de programação, incluindo Fortran, C, Python e Julia, garantindo que possa se integrar aos diversos fluxos de trabalho das equipes científicas modernas.

Os desenvolvedores validaram o software contra cinco casos de teste padrão utilizados na área, variando de espalhamento elástico simples a interações complexas envolvendo múltiplos canais e potenciais não locais. Em todos os casos, o novo solver reproduziu os resultados de referência com alta fidelidade. Para os cenários mais exigentes envolvendo ressonâncias estreitas ou energias profundas de sub-barreira, o software inclui um mecanismo de segurança que alterna automaticamente para um cálculo de precisão total caso o método mais rápido encontre dificuldades, garantindo que os resultados permaneçam confiáveis. Esta combinação de velocidade extrema, alta precisão e recursos de segurança robustos marca um passo significativo à frente, transformando um gargalo computacional em uma tarefa rotineira e abrindo as portas para que mais pesquisadores explorem a dinâmica intrincada do núcleo atômico.

Resumo Técnico de HPRMAT: Um Solucionador de Matriz R de Alto Desempenho com Aceleração por GPU

Definição do Problema
O método da matriz R é um pilar da teoria de reações nucleares, particularmente para cálculos de espalhamento de canais acoplados envolvendo descrições microscópicas de estados de contínuo (por exemplo, Canais Acoplados Discretizados pelo Contínuo ou CDCC). Nestes cálculos, a função de onda interna é expandida em uma base de Lagrange-Legendre, levando à solução de sistemas lineares densos e complexos de dimensão $N = n_{ch} \times n_{lag}$ , onde $n_{ch}$ é o número de canais e $n_{lag}$ é o número de funções de base.

Para aplicações modernas envolvendo fusão de íons pesados ou núcleos de halo (ex: $^{11}$ Be, $^{6}$ He), as dimensões das matrizes atingem rotineiramente $N \sim 10.000$ a $30.000$. O gargalo computacional é a solução destes sistemas lineares. Implementações tradicionais, como o pacote amplamente utilizado de Descouvemont, dependem da inversão explícita da matriz ( $x = A^{-1}b$ ). Esta abordagem é numericamente menos estável do que solucionadores diretos, requer o armazenamento da matriz inversa completa e falha em aproveitar bibliotecas de álgebra linear otimizadas ou aceleradores de hardware modernos. Além disso, os códigos de matriz R existentes na física nuclear geralmente carecem de suporte para aceleração por GPU ou aritmética de precisão mista, limitando a capacidade de realizar cálculos de grande escala em estações de trabalho padrão.

Metodologia
O artigo apresenta o HPRMAT, um framework de solucionador autossuficiente projetado para ser um substituto direto (drop-in replacement) para as rotinas de álgebra linear do pacote de Descouvemont, mantendo total compatibilidade de API. O HPRMAT aborda o gargalo computacional através de quatro backends de solução intercambiáveis:

Solucionador Denso (Tipo 1): Implementa a fatoração LU direta ( $M = P \cdot L \cdot U$ ) usando rotinas LAPACK otimizadas (especificamente ZGESV do OpenBLAS) em vez da inversão de matriz. Isto utiliza multithreading e vetorização SIMD em CPUs.
Solucionador de Precisão Mista (Tipo 2): Explora a disparidade de desempenho de ponto flutuante entre precisão simples (FP32) e dupla (FP64), particularmente em GPUs de consumo, onde a razão FP32:FP64 pode ser de 64:1. O método realiza a fatoração LU $O(N^3)$ custosa em precisão simples e recupera a precisão total de precisão dupla via refinamento iterativo (calculando o resíduo em FP64 e resolvendo para correções).
Solucionador Woodbury-Kinetic (Tipo 3): Um algoritmo especializado para potenciais locais que explora a estrutura em blocos do Hamiltoniano. Ele decompõe a matriz em termos de energia cinética de blocos diagonais e termos de acoplamento diagonais fora da diagonal. Ele constrói um sistema de complemento de Schur resolvido em precisão simples, obtendo um ganho de fator constante ao utilizar aritmética real para a inversão do bloco cinético.
Solucionador Acelerado por GPU (Tipo 4): Utiliza a biblioteca cuSOLVER da NVIDIA. Emprega uma estratégia de precisão mista onde a fatoração e as resoluções triangulares ocorrem na GPU em precisão simples. Para manter a precisão, oferece um modo "híbrido de refinamento no host": a fatoração em precisão simples roda na GPU, enquanto o cálculo do resíduo em precisão dupla e os passos de refinamento são realizados no host da CPU, evitando o desempenho limitado de FP64 em GPUs de consumo.

O framework é implementado em Fortran 90/95 com extensões CUDA C. Inclui bindings para C, Python e Julia. O código lida automaticamente com mecanismos de fallback: se a inicialização da GPU falhar ou se uma fatoração de precisão simples encontrar uma matriz singular, ele reverte para solucionadores de precisão dupla baseados em CPU.

Resultos Principais
Benchmarks foram conduzidos em três plataformas: um Apple M3 Ultra (apenas CPU), uma estação de trabalho Intel Xeon com uma NVIDIA RTX 3090 (CPU + GPU) e um Intel i9-12900 (apenas CPU).

Desempenho:
- GPU vs. Legado: Em $N=25.600$ , o solucionador de GPU (Tipo 4) atinge um ganho de velocidade de 41× sobre o código legado baseado em inversão (referência de Descouvemont).
- GPU vs. CPU Otimizada: O solucionador de GPU atinge aproximadamente 15× de ganho de velocidade sobre o solucionador direto de CPU otimizado (Tipo 1) na RTX 3090.
- Melhorias de CPU: Mesmo sem GPUs, os solucionadores diretos (Tipos 1–3) fornecem ganhos de 5–7× sobre o código de inversão legado em CPUs multi-core modernas devido à transição de inversão para solução direta e ao uso de OpenBLAS.
- Escalabilidade: O solucionador de GPU escala como $O(N^3)$ . Uma única placa de consumo (RTX 3090, 24 GB) pode lidar com matrizes até $N \approx 25.600$ . Um backend multi-GPU (testado em quatro RTX 3090s) estende isto para $N=76.800$ ao distribuir a fatoração de precisão simples enquanto mantém a matriz de precisão dupla no host.
Precisão:
- Os solucionadores de CPU de precisão dupla reproduzem os resultados de referência com precisão de máquina ( $\sim 10^{-18}$ ).
- O solucionador de precisão simples por GPU (Tipo 4) produz erros de seção de choque de $\sim 10^{-3}$ , o que está bem dentro das incertezas experimentais típicas em física nuclear ( $\gtrsim 1\%$ ).
- O modo híbrido de refinamento no host recupera a precisão de máquina ( $\sim 10^{-16}$ ) no mesmo hardware de consumo com apenas um aumento modesto no tempo de execução (2,5–3× o tempo de precisão simples).
- A validação contra cinco casos de teste padrão (incluindo espalhamento $\alpha + ^{208}$ Pb e espalhamento núcleon-nucleão) confirma que todos os solucionadores reproduzem observáveis de referência dentro das incertezas experimentais.

Significância e Alegações
O artigo afirma que o HPRMAT representa o estado da arte atual em desempenho de solucionadores de matriz R para a física nuclear. Sua principal significância reside na democratização da computação de alto desempenho para a teoria de reações nucleares. Ao alavancar a aritmética de precisão mista e GPUs de nível de consumo (ex: RTX 3090/4090), o HPRMAT permite que pesquisadores realizem cálculos de CDCC e de canais acoplados de grande escala em estações de trabalho padrão, removendo a necessidade de GPUs de data-center caras (ex: A100/H100) que são frequentemente inacessíveis a grupos universitários.

O trabalho enfatiza que a transição de inversão de matriz para solução direta, combinada com a aceleração por GPU, torna problemas anteriormente intratáveis (como cálculos XCDCC estendidos para núcleos de halo) acessíveis. A compatibilidade "drop-in" garante que códigos de usuário existentes possam beneficiar-se imediatamente destes ganhos de desempenho sem exigir modificações no código fonte, apenas a relincagem. O artigo posiciona o HPRMAT não apenas como um solucionador mais rápido, mas como uma ferramenta prática que preenche a lacuna entre as capacidades de computação de alto desempenho e as necessidades rotineiras da comunidade de física nuclear.

Mais como este