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

Este artículo presenta una biblioteca de multihilos segura en tiempo real desarrollada para el sistema de control de plasma DIII-D que optimiza con éxito la ejecución de los códigos de física TORBEAM y STRIDE a menos de 20 ms y 100 ms, respectivamente, permitiendo cálculos cruciales de propagación de ondas de ciclotrón electrónico y de límite de estabilidad para futuras centrales de fusión.

Lo esencial

Imagina un reactor de fusión como el DIII-D, un tokamak, como una tormenta gigante y supercaliente de electricidad (plasma) que debe mantenerse perfectamente quieta dentro de una botella magnética. Si la tormenta se vuelve demasiado salvaje, puede chocar contra las paredes y destruir la máquina. Para mantenerla segura, un "Sistema de Control de Plasma" (PCS) actúa como el piloto, realizando ajustes diminutos constantemente.

Sin embargo, la tormenta cambia más rápido de lo que un humano puede reaccionar. El piloto necesita un cerebro informático superrápido que pueda predecir el comportamiento de la tormenta y sugerir ajustes en un abrir y cerrar de ojos. Aquí es donde entra en juego el artículo.

El Problema: El cuello de botella del "Trabajador Único"

Imagina que eres un chef intentando cocinar un banquete masivo. Tienes una receta (un código de física) que te dice cómo cocinar la comida. Pero solo tienes un chef (un único núcleo de procesador informático) haciendo todo el picar, remover y hornear. Si la receta es demasiado compleja, el chef se ve abrumado, la comida se quema y el banquete fracasa.

En el mundo de la fusión, estas "recetas" son simulaciones físicas complejas (como TORBEAM y STRIDE) que calculan cómo calentar el plasma o comprueban si está a punto de volverse inestable. Tradicionalmente, estos cálculos eran demasiado lentos para ejecutarse en tiempo real porque intentaban hacerlo todo con un solo "chef".

La Solución: Un equipo de chefs "Seguro en Tiempo Real"

Los autores construyeron un nuevo sistema para convertir ese único chef en un equipo de chefs trabajando en perfecta sincronía.

1. El Gestor y los Trabajadores: Crearon una biblioteca especial (un conjunto de reglas) que actúa como un Gestor. El Gestor reparte tareas pequeñas e independientes a un grupo de hilos Trabajadores (otros núcleos informáticos).
2. Sin Caos, Solo Orden: En los programas informáticos normales, cuando añades más trabajadores, estos pueden confundirse, esperar demasiado entre sí o colapsar el sistema si uno comete un error. El sistema de los autores es "seguro en tiempo real". Es como una unidad militar donde cada soldado sabe exactamente cuándo moverse y cuándo detenerse. Utilizan un "apretón de manos" especial (variables atómicas) para decir: "Estoy listo", "He terminado" y "Comencemos la siguiente ronda".
3. Sincronización Determinista: La parte más importante es que este sistema garantiza que terminará su trabajo dentro de un límite de tiempo estricto. No importa si el ordenador está ocupado con otras cosas; este equipo está aislado y siempre terminará a tiempo. Esto es crucial porque, si el ordenador tarda demasiado, el plasma podría haber chocado ya.

Las Dos Recetas Principales que Cocinaron

El equipo utilizó este nuevo sistema de "multi-chef" para acelerar dos códigos físicos específicos:

1. TORBEAM: El Guía del Haz Láser

• Qué hace: Imagina intentar golpear un objetivo diminuto y en movimiento dentro de una habitación oscura con un láser. El plasma es la habitación, y el "láser" es un haz de energía (Calentamiento de Ciclotrón de Electrones) utilizado para controlar la estabilidad del plasma.
• El Desafío: El ordenador tiene que calcular la trayectoria exacta que seguirá el haz láser a través del plasma para dar en el punto correcto.
• El Resultado: Debido a que cada haz láser (proveniente de diferentes máquinas llamadas girotrones) viaja de forma independiente, el nuevo sistema permitió que los "trabajadores" calcularan las trayectorias de todos los haces al mismo tiempo.
• La Velocidad: Lograron completar el cálculo en menos de 20 milisegundos. Esto es lo suficientemente rápido para dirigir los láseres en tiempo real, manteniendo el plasma estable.

2. STRIDE: El Verificador de Estabilidad

• Qué hace: Imagina a un funambulista. STRIDE es el inspector de seguridad que comprueba constantemente si el funambulista está a punto de caer. Calcula una "puntuación de estabilidad" para ver si el plasma está a punto de volverse inestable y colapsar.
• El Desafío: Este cálculo es muy pesado y normalmente tarda demasiado para ser útil en tiempo real.
• El Truco: Los autores se dieron cuenta de que podían dividir la comprobación de seguridad en muchas piezas pequeñas e independientes (como revisar diferentes secciones de la cuerda floja). Enviaron estas piezas a los "trabajadores" para resolverlas simultáneamente, y luego combinaron las respuestas.
• La Velocidad: Redujeron el tiempo de cálculo a unos 100 milisegundos. Esto es lo suficientemente rápido para avisar al sistema de control antes de que ocurra un desastre.

La Conclusión

El artículo demuestra que, al construir un equipo de procesadores informáticos especializado y altamente disciplinado (una "biblioteca de multihilos segura en tiempo real"), podemos ejecutar simulaciones físicas complejas lo suficientemente rápido como para controlar realmente un reactor de fusión mientras está en funcionamiento.

• TORBEAM (Dirección de láser) se ejecuta en ~20ms.
• STRIDE (Comprobación de estabilidad) se ejecuta en ~100ms.

Sin este nuevo sistema de "trabajo en equipo", estos cálculos serían demasiado lentos para ser útiles para mantener un reactor de fusión seguro y estable. Este trabajo demuestra que podemos hacer que el "piloto" de un reactor de fusión sea lo suficientemente inteligente como para manejar la velocidad extrema de la tormenta de plasma.

Resumen técnico

Resumen Técnico: Códigos de Física en Tiempo Real Paralelizados para el Control de Plasma en DIII-D

Planteamiento del Problema
Los sistemas de control de plasma (PCS) en tiempo real para tokamaks, como los previstos para futuros reactores de fusión como ITER y SPARC, requieren que los códigos de física basados en modelos operen con una latencia determinista y una alta robustez. Aunque existen códigos de física fuera de línea para el trazado de rayos de calentamiento por ciclotrón de electrones (ECH) (TORBEAM) y el análisis de estabilidad magnetohidrodinámica (MHD) ideal (STRIDE), sus implementaciones en tiempo real suelen enfrentar cuellos de botella de rendimiento en núcleos de CPU individuales. Las soluciones de multiprocesamiento existentes, como OpenMP, son inadecuadas para los entornos de PCS de fusión porque carecen de garantías de tiempo real estricto, dependen de comportamientos definidos por la implementación e introducen un jitter de latencia no determinista debido a las interacciones con el kernel. Además, la aceleración por GPU suele ser poco práctica para estos códigos específicos debido a los altos costos de arranque y a una granularidad de paralelismo insuficiente, lo que requiere una estrategia de paralelización basada en CPU que sea robusta y garantice un determinismo a escala de microsegundos.

Metodología
Los autores desarrollaron e implementaron una biblioteca de multiprocesamiento personalizada y segura para tiempo real en el PCS de DIII-D para permitir la ejecución paralela de los códigos TORBEAM y STRIDE.

• Arquitectura de la Biblioteca: La biblioteca utiliza un patrón "Manager-Worker" (Gestor-Trabajador) implementado en C11. Un único hilo Manager (que coincide con el hilo principal del PCS) asigna unidades de trabajo a múltiples hilos Worker. La comunicación se logra exclusivamente mediante variables atómicas de bajo nivel utilizando el modelo de memoria acquire-release de C11, evitando abstracciones de alto nivel como los semáforos que interactúan con el kernel.
• Mecanismo de Sincronización: El sistema emplea un protocolo de sincronización bidireccional inspirado en la característica "rendezvous" de Ada. El Manager y los Workers se notifican mutuamente de su estado de forma independiente mediante banderas atómicas (Ready y Complete). Esto asegura que todos los hilos alcancen los puntos de sincronización al inicio y al final de las secciones críticas sin comportamiento indefinido. El diseño prioriza el determinismo sobre el rendimiento máximo, aceptando que los hilos puedan girar ociosamente (spin) mientras esperan a que otros se pongan al día.
• Entorno de Hardware: La implementación se ejecutó en sistemas con núcleos de CPU aislados para evitar la interferencia de otros procesos e interrupciones del kernel. La implementación de STRIDE utilizó un sistema operativo de tiempo real comercial (RedHawk) para minimizar el jitter de latencia, mientras que TORBEAM se ejecutó en un entorno CentOS 7 estándar con aislamiento de núcleos.
• Paralelización Específica del Código:
  • rt-TORBEAM: Se paralelizaron los cálculos de trazado de rayos para los girotrones independientes. El código se simplificó de un modelo de haz gaussiano a un trazado de rayo único para calcular la ubicación máxima de absorción $(R, Z)$, que luego se convierte a una coordenada de flujo normalizada ($\rho$) para el direccionamiento de los espejos.
  • rt-STRIDE: El problema de estabilidad ideal se reformuló como una ecuación de Riccati en lugar de una ecuación de Newcomb. Esto permite que la Matriz de Transición de Estado (STM) se calcule mediante un método de "disparo" (shooting method) donde el dominio de integración se divide en intervalos. Estos intervalos se distribuyen entre los hilos Worker basándose en una métrica de "rigidez" (stiffness) (mayor densidad de pasos cerca de las superficies racionales), y los STM resultantes son combinados por el Manager para calcular el parámetro de estabilidad final ($\delta W$).

Contribuciones Clave

1. Biblioteca de Multiprocesamiento Segura para Tiempo Real: El desarrollo de una biblioteca de hilos determinista y libre de kernel que permite la paralelización segura en el hardware de los PCS de fusión, abordando las limitaciones de las bibliotecas estándar como OpenMP.
2. rt-TORBEAM Paralelizado: Despliegue exitoso de un código de trazado de rayos paralelizado para el direccionamiento de espejos ECH, capaz de manejar múltiples girotrones simultáneamente.
3. rt-STRIDE Paralelizado: Implementación de un código de estabilidad MHD ideal paralelizado utilizando una formulación de Riccati y división de intervalos, permitiendo el cálculo en tiempo real de los límites de estabilidad.
4. Validación Experimental: Demostración de estos códigos operando dentro de las estrictas restricciones de tiempo de ciclo del PCS de DIII-D durante disparos de plasma reales.

Resultados

• Rendimiento de rt-TORBEAM: El código se ejecutó consistentemente en menos de 20 ms para cuatro girotrones dirigidos. El sistema demostró un seguimiento razonable de los valores de $\rho$ objetivo ($|\rho_{target} - \rho_{torbeam}| \le 0.05$). El tiempo de ciclo se mantuvo constante a medida que se añadían más girotrones, siempre que existiera un hilo worker libre para cada uno. Se observó un pico en el tiempo de ciclo a ~40 ms en casos de borde específicos, atribuido al código de física y no a la biblioteca de hilos.
• Rendimiento de rt-STRIDE: El código calculó consistentemente el parámetro de estabilidad $\delta W$ en aproximadamente 100 ms en un sistema de 72 núcleos. El perfilado reveló que la integración de la STM en sí misma solo tomó ~20 ms, consumiéndose la mayor parte del tiempo en el preprocesamiento de datos serial (conversión de equilibrio a coordenadas de flujo).
• Seguimiento y Control: Los resultados experimentales mostraron que, si bien los cálculos de física eran lo suficientemente rápidos para el control, el hardware físico de los espejos (que tarda ~0.5 s en moverse) y la velocidad de la reconstrucción de los equilibrios en tiempo real limitaron el rendimiento general del seguimiento, particularmente para cambios rápidos en los valores objetivo.

Significancia y Reivindicaciones
El artículo afirma que la biblioteca de multiprocesamiento desarrollada es un habilitador crítico para los próximos dispositivos de fusión, permitiendo que los códigos basados en física cumplan con los requisitos de tiempo de ejecución para el control en tiempo real. Los autores enfatizan que, a diferencia de los modelos sustitutos de aprendizaje automático (machine learning surrogates), que pueden carecer de robustez fuera de sus datos de entrenamiento, estos códigos basados en física ofrecen "garantías de rendimiento robustas desde el primer día de operación". El trabajo demuestra que, con una paralelización adecuada y un hilo determinista, códigos de física complejos como TORBEAM y STRIDE pueden integrarse en el PCS para proporcionar información esencial sobre la deposición de ECH y los límites de estabilidad ideal. Los autores sugieren que esta biblioteca puede aplicarse a otros códigos en tiempo real, incluyendo el análisis de estabilidad vertical y la predicción de perfiles, y señalan que los esfuerzos futuros deben centrarse en paralelizar los pasos de preprocesamiento de datos para reducir aún más la latencia.