HARD: A Performance Portable Radiation Hydrodynamics Code based on FleCSI Framework

HARD es un código de hidrodinámica radiativa de código abierto y portable en rendimiento, construido sobre el marco FleCSI y Kokkos, que permite simulaciones eficientes en diversas arquitecturas de hardware mientras garantiza la fiabilidad científica mediante pruebas de regresión automatizadas y desarrollo impulsado por la comunidad.

Lo esencial

Imagina que estás intentando simular un evento masivo y caótico, como la explosión de una estrella o la detonación de una bomba de fusión. Para lograrlo, necesitas un programa informático que pueda rastrear dos cosas ocurriendo simultáneamente: cómo se mueve el gas (hidrodinámica) y cómo la energía luminosa (radiación) calienta y empuja a ese gas. A esto se le llama Hidrodinámica de Radiación.

El artículo presenta una nueva herramienta de software llamada HARD (Hidrodinámica y Difusión de Radiación), diseñada para resolver estos complejos acertijos. Así es como funciona, explicado mediante analogías sencillas:

1. El "Adaptador Universal" (Portabilidad de Rendimiento)

Piensa en las supercomputadoras del mundo como diferentes tipos de vehículos: algunos son sedanes (portátiles), otros son camiones (clústeres estándar) y algunos son autos de carreras masivos y personalizados (las supercomputadoras más potentes del mundo con CPUs y GPUs).

Por lo general, el software es como un motor de automóvil diseñado solo para un vehículo específico. Si quieres ejecutarlo en un auto diferente, tienes que reconstruir el motor desde cero. HARD es diferente. Está construido sobre un "adaptador universal" llamado FleCSI.

• La Analogía: Imagina un controlador de videojuegos que se reconfigura automáticamente para funcionar con una PlayStation, una Xbox o una PC sin que tú cambies los botones. HARD hace esto por las computadoras. Escribe el código de física una sola vez y luego traduce automáticamente ese código para que se ejecute de manera eficiente en cualquier cosa, desde un portátil hasta una supercomputadora gigante, ya sea que esa máquina utilice procesadores estándar o tarjetas gráficas especializadas (GPUs).

2. El "Administrador de Tareas" (Orquestación)

Simular una explosión estelar implica millones de cálculos diminutos ocurriendo al mismo tiempo.

• La Analogía: Imagina un sitio de construcción. En lugar de que un solo capataz le diga a cada trabajador individualmente qué hacer uno por uno (lo cual es lento), HARD actúa como un gerente de proyecto inteligente. Divide el trabajo en pequeñas "tareas" (como "verter concreto aquí" o "medir esta viga") y las reparte a un equipo de trabajadores.
• La Magia: Si los trabajadores están en un mismo edificio, el gerente usa un estilo de comunicación (MPI). Si están en edificios diferentes, usa otro (Legion o HPX). Los trabajadores (los cálculos de física) no necesitan saber cómo están siendo gestionados; simplemente hacen su trabajo. Esto permite que el software se escale hacia arriba o hacia abajo instantáneamente.

3. El "Sistema de Doble Verificación" (Verificación)

En la ciencia, no puedes confiar solo en los números; debes probar que son correctos.

• La Analogía: HARD viene con un "manual de entrenamiento" y un "examen sorpresa" integrados. Ejecuta automáticamente problemas de prueba famosos y bien conocidos (como el "tubo de choque de Sod", que es como una pregunta de examen de física estándar que todos conocen la respuesta).
• El Resultado: El software compara su propia respuesta con la respuesta "correcta" conocida. Si coinciden, el software aprueba la prueba. Esto asegura que cuando los científicos lo utilicen para problemas nuevos y desconocidos, los resultados sean confiables.

4. ¿Qué Simula Realmente?

El artículo muestra a HARD funcionando en varios escenarios específicos:

• El Tubo de Choque: Como la ruptura de una presa, donde gas de alta presión se precipita hacia gas de baja presión, creando una onda de choque. HARD predijo la forma de la onda perfectamente.
• Calentamiento y Enfriamiento: Imagina una olla de agua junto a un calentador. El artículo muestra cómo HARD calcula con precisión cómo el agua se calienta hasta igualar la temperatura del calentador, y cómo se enfría si el calentador se apaga.
• El "Impulso de Radiación": En algunos escenarios, la energía luminosa es tan fuerte que crea sus propias ondas de choque. HARD demostró que, al añadir radiación, estas ondas de choque se forman más rápido y se comportan de manera diferente que cuando solo se observa el gas.
• El "Fluido Giratorio" (Inestabilidad de Kelvin-Helmholtz): Imagina dos ríos fluyendo uno junto al otro a diferentes velocidades, creando un límite turbio y giratorio. El artículo encontró que añadir radiación a esta mezcla hace que los remolinos crezcan y se vuelvan caóticos mucho más rápido que sin ella.

5. Velocidad y Escala

Los autores probaron HARD en una supercomputadora masiva llamada Chicoma.

• La Analogía: Intentaron resolver un rompecabezas agregando más y más personas al equipo.
• El Resultado: A medida que agregaban más "trabajadores" informáticos (nodos), la velocidad de la simulación aumentó casi perfectamente. No se ralentizó debido a retrasos en la comunicación.
• El Impulso de la GPU: Cuando lo probaron en computadoras con tarjetas gráficas potentes (GPUs), una sola tarjeta gráfica fue 7 veces más rápida que un procesador de computadora estándar.

Resumen

HARD es una nueva herramienta de código abierto para que los científicos simulen cómo la materia y la luz interactúan en entornos extremos. Su superpoder principal es que es portátil (se ejecuta en cualquier computadora), confiable (prueba que sus propios cálculos son correctos) y rápido (se escala hasta las supercomputadoras más grandes). Está diseñado para ayudar a los investigadores a comprender todo, desde cómo explotan las estrellas hasta cómo podríamos crear energía de fusión limpia.

Resumen técnico

Resumen Técnico: HARD – Un Código de Hidrodinámica de Radiación Portátil en Rendimiento

Planteamiento del Problema
La hidrodinámica de radiación (RHD) gobierna sistemas complejos donde el acoplamiento radiación-materia dicta la dinámica global, incluyendo la fusión por confinamiento inercial (ICF), la evolución estelar y las ondas de choque radiativas. Simular con precisión estos fenómenos requiere resolver ecuaciones diferenciales parciales no lineales fuertemente acopladas a través de escalas espaciales y temporales dispares, a menudo involucrando términos fuente rígidos. Aunque existen códigos RHD maduros, muchas implementaciones heredadas luchan con arquitecturas de computación heterogéneas modernas. A menudo sufren de una portabilidad de rendimiento limitada, modelos de datos rígidos que obstaculizan la experimentación algorítmica y dificultades para integrarse con herramientas contemporáneas para la integración continua (CI), el empaquetado y la reproducibilidad. Existe una necesidad crítica de una plataforma de software que separe limpiamente la física de los back-ends de ejecución y que escale eficientemente desde portátiles hasta supercomputadores de clase líder.

Metodología
HARD (Hidrodinámica y Difusión de Radiación) es una aplicación de código abierto diseñada para abordar estos desafíos aprovechando el marco de trabajo FleCSI (Infraestructura Flexible de Ciencias Computacionales).

• Arquitectura y Portabilidad: HARD expresa unidades computacionales como tareas orquestadas por múltiples tiempos de ejecución back-end, incluyendo Legion, MPI y HPX. El paralelismo a nivel de nodo se delega a Kokkos, permitiendo que una única base de código se ejecute eficientemente en CPUs y GPUs a través de sistemas heterogéneos. El software está implementado en C++ moderno y utiliza el marco de trabajo Singularity-EOS para ecuaciones de estado.
• Modelos Matemáticos: El código adopta una formulación de las ecuaciones RHD en un marco en movimiento (Euleriano) en el límite de difusión. Resuelve ecuaciones acopladas de hidrodinámica y difusión de radiación. Para manejar regímenes que van desde ópticamente gruesos hasta ópticamente delgados, el código emplea una ley puente con una función limitadora de flujo de energía de radiación, asegurando que el flujo de radiación alcance magnitudes correctas en el límite de flujo libre mientras recupera la aproximación de Eddington en el límite de difusión.
• Algoritmos Numéricos:
  • Advección: Resuelta utilizando un método de volúmenes finitos con un enfoque de división de operadores. El código emplea reconstrucción WENO5-Z para todas las variables, utilizando flujos HLL para hidrodinámica y flujos Lax-Friedrichs local (Rusanov) para la densidad de energía de radiación. La integración temporal se realiza utilizando el método de Heun (Runge-Kutta de segundo orden).
  • Difusión: La parte difusiva de las ecuaciones de transferencia radiativa se resuelve utilizando un solucionador Geometric Multigrid (GMG) con afinamiento regular. Actualmente, se utiliza un método de relajación simple en la cuadrícula más gruesa, con planes de integrar la biblioteca FleCSolve para solucionadores lineales más avanzados.
• Verificación e Infraestructura: HARD incluye un conjunto de pruebas de regresión que reproduce automáticamente problemas canónicos (tubos de choque Sod y LeBlanc, onda de explosión de Sedov) y compara las soluciones numéricas con resultados analíticos. El proyecto se distribuye bajo una Licencia BSD de 3 Cláusulas, está alojado en GitHub y utiliza Spack para la configuración del entorno y compilaciones en contenedores para la reproducibilidad.

Resultados Clave
El artículo valida HARD a través de una serie de pruebas de referencia y experimentos de escalado de rendimiento:

• Verificación:
  • Tubo de Choque Sod: El código reproduce el problema estándar de tubo de choque con un error $L_1$ inferior a $5 \times 10^{-4}$, coincidiendo con soluciones analíticas para la densidad.
  • Calentamiento y Enfriamiento: Las simulaciones de calentamiento y enfriamiento de fluidos por radiación demuestran convergencia asintótica a la temperatura de radiación, coincidiendo con soluciones de referencia derivadas de la integración de EDO.
  • Choque Inducido por Temperatura: El código modela con éxito la formación de choques impulsada por la temperatura de radiación, mostrando ninguna formación de choque a $T=0$ K y estructuras de choque distintas a temperaturas no nulas.
  • Inestabilidad de Kelvin-Helmholtz (KHI): Las simulaciones 2D confirman que la presencia de un campo de radiación acelera el crecimiento de la KHI, consistente con la teoría lineal.
  • Convergencia: Las pruebas de ondas acústicas utilizando el limitador WENO5-Z demuestran los órdenes de convergencia esperados.
• Rendimiento y Escalabilidad:
  • Las pruebas se realizaron en el supercomputador LANL Chicoma (CPUs AMD EPYC).
  • Escalado Fuerte: La referencia de radiación 3D logra una aceleración casi ideal en hasta 1024 nodos utilizando configuraciones MPI y MPI+OpenMP.
  • Escalado Débil: La eficiencia paralela se mantiene consistente (manteniendo ~70% en 512 nodos) a medida que el tamaño del problema y el número de nodos aumentan proporcionalmente.
  • Aceleración GPU: Las pruebas preliminares en GPUs A100 muestran una aceleración de 7× en comparación con un nodo CPU, logrando 67,6 millones de actualizaciones de celda por segundo.

Significado y Afirmaciones
Los autores posicionan HARD como una plataforma sostenible para avanzar en la investigación de hidrodinámica de radiación. Su significado principal radica en dos áreas complementarias:

1. Portabilidad de Rendimiento: Proporciona una implementación única y portable que se mapea limpiamente a arquitecturas heterogéneas (CPUs y GPUs) a través de las abstracciones FleCSI y Kokkos, abordando las limitaciones de los códigos heredados en hardware moderno.
2. Plataforma de Investigación Modular: Al desacoplar la física de los back-ends de ejecución y proporcionar interfaces claras para la numeración (reconstrucción, solucionadores de Riemann) y la física (ecuaciones de estado, cierres de radiación), HARD sirve como banco de pruebas para desarrollar nuevos métodos RHD.

El artículo afirma que HARD es una herramienta versátil para la comunidad de Física de Alta Densidad de Energía (HEDP), capaz de validar nuevos algoritmos e investigar comportamientos fundamentales de RHD. Aunque actualmente se centra en límites de difusión, el diseño modular permite futuras extensiones, como enfoques de radiación multigrupo y esquemas de difusión de orden superior, para abordar fenómenos HEDP y astrofísicos más complejos. Los autores enfatizan que la combinación de portabilidad de rendimiento, infraestructura de verificación rigurosa y gobernanza de código abierto hace de HARD un recurso valioso para la comparación entre códigos y el desarrollo de métodos.