Parallelized Real-time Physics Codes for Plasma Control on DIII-D

Este artigo apresenta uma biblioteca de multithreading segura em tempo real desenvolvida para o sistema de controle de plasma DIII-D que otimiza com sucesso a execução dos códigos de física TORBEAM e STRIDE para menos de 20 ms e 100 ms, respectivamente, permitindo cálculos cruciais de propagação de ondas de ciclo eletrônico e limites de estabilidade para futuras usinas de fusão.

O essencial

Imagine um reator de fusão como o DIII-D tokamak, que é uma tempestade gigante e superquente de eletricidade (plasma) que precisa ser mantida perfeitamente imóvel dentro de uma garrafa magnética. Se a tempestade ficar selvagem demais, ela pode colidir com as paredes e destruir a máquina. Para mantê-la segura, um "Sistema de Controle de Plasma" (PCS) atua como o piloto, fazendo pequenos ajustes constantes.

No entanto, a tempestade muda mais rápido do que um humano consegue reagir. O piloto precisa de um cérebro de computador superveloz que possa prever o comportamento da tempestade e sugerir ajustes num piscar de olhos. É aqui que entra o artigo.

O Problema: O Gargalo do "Trabalhador Único"

Imagine que você é um chef tentando cozinhar um banquete massivo. Você tem uma receita (um código de física) que lhe diz como cozinhar a comida. Mas você tem apenas um chef (um único núcleo de processador de computador) fazendo todo o picar, mexer e assar. Se a receita for muito complexa, o chef ficará sobrecarregado, a comida queimará e o banquete falhará.

No mundo da fusão, essas "receitas" são simulações de física complexas (como TORBEAM e STRIDE) que calculam como aquecer o plasma ou verificar se ele está prestes a se tornar instável. Tradicionalmente, esses cálculos eram lentos demais para serem executados em tempo real porque tentavam fazer tudo com apenas um "chef".

A Solução: Uma Equipe de Chefs "Segura em Tempo Real"

Os autores construíram um novo sistema para transformar esse chef individual em uma equipe de chefs trabalhando em perfeita sincronia.

1. O Gerente e os Trabalhadores: Eles criaram uma biblioteca especial (um conjunto de regras) que atua como um Gerente. O Gerente distribui tarefas pequenas e independentes para um grupo de threads de Trabalhadores (outros núcleos de computador).
2. Sem Caos, Apenas Ordem: Em programas de computador normais, quando você adiciona mais trabalhadores, eles podem ficar confusos, esperar demais uns pelos outros ou travar o sistema se um deles cometer um erro. O sistema dos autores é "seguro em tempo real". É como uma unidade militar onde cada soldado sabe exatamente quando se mover e quando parar. Eles usam um "aperto de mão" especial (variáveis atômicas) para dizer: "Estou pronto", "Terminei" e "Vamos começar a próxima rodada".
3. Tempo Determinístico: A parte mais importante é que este sistema garante que terminará seu trabalho dentro de um limite de tempo rigoroso. Não importa se o computador está ocupado com outras coisas; esta equipe é isolada e sempre terminará a tempo. Isso é crucial porque, se o computador demorar demais, o plasma já poderá ter colidido.

As Duas Principais Receitas que Eles Cozinharam

A equipe usou este novo sistema de "múltiplos chefs" para acelerar dois códigos de física específicos:

1. TORBEAM: O Guia do Feixe de Laser

• O que faz: Imagine tentar atingir um alvo minúsculo e móvel dentro de uma sala escura com um laser. O plasma é a sala, e o "laser" é um feixe de energia (Aquecimento de Ciclotron de Elétrons) usado para controlar a estabilidade do plasma.
• O Desafio: O computador tem que calcular a trajetória exata que o feixe de laser percorrerá através do plasma para atingir o ponto certo.
• O Resultado: Como cada feixe de laser (vindo de diferentes máquinas chamadas girotrons) viaja de forma independente, o novo sistema permitiu que os "trabalhadores" calculassem as trajetórias de todos os feixes ao mesmo tempo.
• A Velocidade: Eles conseguiram realizar o cálculo em menos de 20 milissegundos. Isso é rápido o suficiente para direcionar os lasers em tempo real, mantendo o plasma estável.

2. STRIDE: O Verificador de Estabilidade

• O que faz: Imagine um equilibrista na corda bamba. O STRIDE é o inspetor de segurança que verifica constantemente se o equilibrista está prest着 a cair. Ele calcula uma "pontuação de estabilidade" para ver se o plasma está prestes a se tornar instável e colidir.
• O Desafio: Este cálculo é muito pesado e geralmente demora demais para uso em tempo real.
• O Truque:** Os autores perceberam que poderiam dividir a verificação de segurança em muitas partes pequenas e independentes (como verificar diferentes seções da corda bamba). Eles enviaram essas partes para os "trabalhadores" resolverem simultaneamente e, depois, combinaram as respostas.
• A Velocidade: Eles reduziram o tempo de cálculo para cerca de 100 milissegundos. Isso é rápido o suficiente para alertar o sistema de controle antes que um desastre aconteça.

A Conclusão

O artigo demonstra que, ao construir uma equipe especializada e altamente disciplinada de processadores de computador (uma "biblioteca de multi-threading segura em tempo real"), podemos executar simulações de física complexas rapidamente o suficiente para realmente controlar um reator de fusão enquanto ele está operando.

• TORBEAM (direcionamento de laser) roda em ~20ms.
• STRIDE (verificação de estabilidade) roda em ~100ms.

Sem este novo sistema de "trabalho em equipe", esses cálculos seriam lentos demais para serem úteis para manter um reator de fusão seguro e estável. Este trabalho prova que podemos tornar o "piloto" de um reator de fusão inteligente o suficiente para lidar com a velocidade extrema da tempestade de plasma.

Resumo técnico

Resumo Técnico: Códigos de Física em Tempo Real Paralelizados para Controle de Plasma no DIII-D

Definição do Problema
Sistemas de controle de plasma (PCS) em tempo real para tokamaks, como os planejados para futuros reatores de fusão como o ITER e o SPARC, exigem que códigos de física baseados em modelos operem com latência determinística e alta robustez. Embora existam códigos de física offline para o rastreamento de raios de aquecimento por ciclo de elétrons (ECH) (TORBEAM) e análise de estabilidade magnetohidrodinâmica (MHD) ideal (STRIDE), suas implementações em tempo real frequentemente enfrentam gargalos de desempenho em núcleos de CPU únicos. Soluções de multithreading existentes, como o OpenMP, são inadequadas para ambientes de PCS de fusão porque carecem de garantias de tempo real estrito (hard real-time), dependem de comportamentos definidos pela implementação e introduzem jitter de latência não determinístico devido a interações com o kernel. Além disso, a aceleração por GPU é frequentemente impraticável para esses códigos específicos devido aos altos custos de inicialização e à insuficiência de granularidade de paralelismo, necessitando de uma estratégia de paralelização baseada em CPU que seja robusta e garanta determinismo na escala de microssegundos.

Metodologia
Os autores desenvolveram e implementaram uma biblioteca de multithreading personalizada e segura para tempo real no PCS do DIII-D para permitir a execução paralela dos códigos TORBEAM e STRIDE.

• Arquitetura da Biblioteca: A biblioteca utiliza um padrão "Gerente-Trabalhador" (Manager-Worker) implementado em C11. Uma única thread Gerente (coincidindo com a thread principal do PCS) atribui unidades de trabalho a múltiplas threads Trabalhadoras. A comunicação é realizada exclusivamente através de variáveis atômicas de baixo nível usando o modelo de memória acquire-release do C11, evitando abstrações de alto nível como semáforos que interagem com o kernel.
• Mecanismo de Sincronização: O sistema emprega um protocolo de sincronização bidirecional inspirado no recurso "rendezvous" da linguagem Ada. O Gerente e os Trabalhadores notificam uns aos outros de forma independente sobre o status via flags atômicas (Ready e Complete). Isso garante que todas as threads alcancem pontos de sincronização no início e no fim de seções críticas sem comportamento indefinido. O design prioriza o determinismo sobre a vazão máxima (throughput), aceitando que as threads possam girar ociosamente (spin) enquanto esperam que outras alcancem o mesmo ponto.
• Ambiente de Hardware: A implementação rodou em sistemas com núcleos de CPU isolados para evitar interferência de outros processos e interrupções de kernel. A implementação do STRIDE utilizou um sistema operacional de tempo real comercial (RedHawk) para minimizar o jitter de latência, enquanto o TORBEAM rodou em um ambiente CentOS 7 padrão com isolamento de núcleos.
• Paralelização Específica de Código:
  • rt-TORBEAM: Os cálculos de rastreamento de raios para gyrotrons independentes foram paralelizados. O código foi simplificado de um modelo de feixe Gaussiano para um rastreamento de raio único para calcular a localização de absorção máxima $(R, Z)$, que é então convertida para uma coordenada de fluxo normalizada ($\rho$) para o direcionamento de espelhos.
  • rt-STRIDE: O problema de estabilidade ideal foi reformulado como uma equação de Riccati em vez de uma equação de Newcomb. Isso permite que a Matriz de Transição de Estado (STM) seja calculada via um método de "shooting" onde o domínio de integração é dividido em intervalos. Esses intervalos são distribuídos entre as threads Trabalhadoras com base em uma métrica de "rigidez" (stiffness) (maior densidade de passos próximos a superfícies racionais), e as STMs resultantes são combinadas pelo Gerente para computar o parâmetro de estabilidade final ($\delta W$).

Principais Contribuições

1. Biblioteca de Multithreading Segura para Tempo Real: O desenvolvimento de uma biblioteca de multithreading determinística e livre de kernel que permite a paralelização segura em hardware de PCS de fusão, abordando as limitações de bibliotecas padrão como o OpenMP.
2. rt-TORBEAM Paralelizado: Implementação bem-sucedida de um código de rastreamento de raios paralelizado para o direcionamento de espelhos de ECH, capaz de lidar com múltiplos gyrotrons simultaneamente.
3. rt-STRIDE Paralelizado: Implementação de um código de estabilidade MHD ideal paralelizado usando uma formulação de Riccati e divisão de intervalos, permitindo o cálculo em tempo real dos limites de estabilidade.
4. Validação Experimental: Demonstração desses códigos operando dentro das rigorosas restrições de tempo de ciclo do PCS do DIII-D durante disparos (shots) reais de plasma.

Resultos

• Desempenho do rt-TORBEAM: O código executou consistentemente em menos de 20 ms para quatro gyrotrons direcionados. O sistema demonstrou um rastreamento razoável dos valores de $\rho$ alvo ($|\rho_{target} - \rho_{torbeam}| \le 0.05$). O tempo de ciclo permaneceu constante conforme gyrotrons adicionais eram incluídos, desde que houvesse uma thread trabalhadora livre para cada um. Um pico no tempo de ciclo para ~40 ms foi observado em casos de borda específicos, atribuído ao código de física e não à biblioteca de multithreading.
• Desempenho do rt-STRIDE: O código calculou consistentemente o parâmetro de estabilidade $\delta W$ em aproximadamente 100 ms em um sistema de 72 núcleos. O perfilamento revelou que a integração da própria STM levou apenas ~20 ms, sendo que a maior parte do tempo foi consumida pelo pré-processamento de dados serial (conversão de equilíbrio para coordenadas de fluxo).
• Rastreamento e Controle: Resultados experimentais mostraram que, embora os cálculos de física fossem rápidos o suficiente para o controle, o hardware físico dos espelhos (levando ~0,5 s para se mover) e a velocidade da reconstrução de equilíbrios em tempo real limitaram o desempenho geral de rastreamento, particularmente para mudanças rápidas nos valores alvo.

Significância e Alegações
O artigo afirma que a biblioteca de multithreading desenvolvida é um habilitador crítico para dispositivos de fusão de próxima geração, permitindo que códigos de física baseados em modelos atendam aos requisitos de tempo de execução para controle em tempo real. Os autores enfatizam que, ao contrário de modelos substitutos (surrogate models) de aprendizado de máquina, que podem carecer de robustez fora de seus dados de treinamento, esses códigos de física oferecem "desempenho robusto desde o primeiro dia de operação". O trabalho demonstra que, com a paralelização adequada e multithreading determinístico, códigos de física complexos como TORBEAM e STRIDE podem ser integrados ao PCS para fornecer informações essenciais sobre deposição de ECH e limites de estabilidade ideal. Os autores sugerem que esta biblioteca pode ser aplicada a outros códigos de tempo real, incluindo análise de estabilidade vertical e previsão de perfis, e observam que esforços futuros devem focar em paralelizar as etapas de pré-processamento de dados para reduzir ainda mais a latência.