High-Performance Simulations of Higher Representations of Wilson Fermions

Resumen Técnico: Simulaciones de Alto Rendimiento de Representaciones Superiores de Fermiones de Wilson (HiRep v2)

Planteamiento del Problema
La teoría de campos en redes (Lattice Field Theory) es esencial para la comprensión no perturbativa de las teorías de campos fuertemente acoplados, particularmente para pruebas de precisión del Modelo Estándar (QCD). Aunque existen librerías de simulación robustas para grupos de gauge $SU(3)$, la búsqueda directa de física más allá del Modelo Estándar (BSM) requiere software capaz de simular grupos de gauge generales $SU(N_g)$ y fermiones en representaciones superiores (por ejemplo, adjunta, simétrica, antisimétrica). Las librerías de código abierto existentes a menudo carecen de la flexibilidad para soportar un $N_g$ general con representaciones superiores, o carecen de la combinación específica de características de alto rendimiento (como la aceleración por GPU y algoritmos avanzados) requeridas para simulaciones a gran escala de modelos de Higgs compuesto y tecnicolor. Existe una necesidad específica de una librería que implemente fermiones de Wilson en representaciones superiores con números generales de colores, optimizada para arquitecturas heterogéneas modernas de CPU-GPU.

Metodología
El artículo presenta HiRep v2, una suite de software de código abierto (escrita en C, CUDA C y C++) diseñada para abordar estas limitaciones. La metodología central consiste en una reingeniería integral del flujo de trabajo de simulación de la teoría de campos en redes con un enfoque en el operador de Dirac, que es el principal cuello de botella computacional.

• Operador de Dirac y Álgebra Lineal: La implementación generaliza el operador de Wilson-Dirac a representaciones arbitrarias $R$ y grupos de gauge $SU(N_g)$. Para optimizar el rendimiento, el código emplea plantillas de macros genéricas para las operaciones de álgebra lineal y utiliza kernels "fusionados" (por ejemplo, combinaciones lineales) para reducir el movimiento de memoria.
• Aceleración por GPU y Disposición de Memoria: Una parte significativa del trabajo consiste en la portabilidad del código a GPUs NVIDIA y AMD. Los autores introducen una disposición de "geometría optimizada para la computación" donde la red local se almacena como un bloque contiguo único, separado de los buffers de envío/recepción. Esto contrasta con la disposición anterior "optimizada para la memoria". Este cambio facilita la coalescencia de memoria eficiente y los patrones de acceso con saltos (strided access) en GPUs, donde los hilos acceden a los componentes del campo en bloques (por ejemplo, 32 sitios) en lugar de sitio por sitio.
• Paralelización y Afinidad: El código soporta paralelización híbrida mediante MPI, OpenMP y hilos de GPU. Implementa comunicadores conscientes de la topología para reconfigurar las asociaciones de procesos MPI con los subbloques de la red, optimizando la comunicación intra-nodo. También aborda la afinidad NUMA (Acceso a Memoria No Uniforme) mediante la fijación de procesos a dominios específicos de CPU/GPU y el uso de políticas de "primer toque" (first touch) para asegurar que las páginas de memoria se asignen en el nodo NUMA correcto.
• Algoritmos: HiRep v2 soporta los algoritmos de Monte Carlo de Hamilton híbrido (HMC) y de Monte Carlo de Hamilton racional (RHMC). Incluye características avanzadas como la aceleración de Hasenbusch, el precondicionamiento par-impar y los inversores de gradiente conjugado de múltiples desplazamientos (multi-shift). La suite soporta varios tipos de acciones, incluyendo acciones de gauge de Symanzik mejoradas (Lüscher-Weisz), mejoras de tipo Clover y Clover exponencial, y acciones de Masa Retorcida (Twisted Mass).
• Observables: La librería incluye herramientas para espectroscopía (contribuciones conectadas y desconectadas), métodos de dispersión (método de Lüscher), renormalización no perturbativa (RI'-MOM) y suavizado de gauge (smearing) (APE, Stout, HYP, Wilson Flow).

Contribuciones Clave

1. Portabilidad y Optimización de GPU: La contribución principal es el soporte completo para la aceleración por GPU (NVIDIA y AMD) para simulaciones que involucran representaciones superiores de fermiones de Wilson. Los autores demuestran que su implementación del operador de Dirac alcanza un excelente rendimiento, llegando al 80–90% del ancho de banda teórico máximo en GPUs modernas.
2. Escalabilidad: El software exhibe excelentes propiedades de escalado fuerte y débil a través de múltiples GPUs y nodos. Las pruebas de rendimiento en el supercomputador LUMI-G muestran un escalado hasta 1024 GPUs con una pérdida mínima de eficiencia, alcanzando un ancho de banda pico de 908 TB/s.
3. Generalización a $N_g$: El código soporta números generales de colores ($N_g$) y diversas representaciones (Fundamental, Adjunta, Simétrica, Antisimétrica). Los autores introducen un "kernel mejorado para grandes $N_g$" que paraleliza sobre la dimensión del fermión ($D_R$) además de los sitios de la red. Este enfoque mitiga el desbordamiento de registros (register spilling) y la degradación del rendimiento que ocurre en los kernels estándar paralelizados por sitio a medida que $N_g$ aumenta.
4. Robustez Algorítmica: La suite integra técnicas numéricas avanzadas, incluyendo aritmética de doble-doble para reducciones de alta precisión (para mitigar la acumulación de errores en volúmenes grandes) y varios integradores (Leapfrog, Omelyan) con estructuras jerárquicas.

Resultados

• Benchmarks de Rendimiento: En GPUs individuales (V100, A100, H100, MI250x), las rutinas de álgebra lineal alcanzan hasta el 90% del ancho de banda pico. El operador de Wilson-Dirac muestra que, para valores grandes de $N_g$, el kernel paralelizado por $D_R$-sitio supera significativamente al kernel estándar paralelizado por sitio, evitando la caída de ancho de banda causada por el desbordamiento de registros.
• Escalado: Las pruebas de escalado débil en LUMI-G y JUWELS demuestran un escalado casi perfecto hasta 1024 GPUs. El escalado fuerte muestra las pérdidas de eficiencia esperadas para redes locales muy pequeñas donde la sobrecarga de comunicación no puede ser enmascarada por la computación.
• Pruebas Algorítmicas: El código pasa las pruebas algorítmicas estándar, incluyendo la igualdad de Creutz ($\langle e^{-\Delta H} \rangle = 1$) y el escalado de las violaciones del Hamiltoniano con el tamaño del paso ($\langle \Delta H \rangle \propto \delta\tau^4$). Las pruebas de reversibilidad confirman que el algoritmo es reversible hasta la precisión del inversor.
• Validación de la Física: Los autores validan el código de espectroscopía comparando los resultados de CPU y GPU para una teoría de gauge $SU(2)$ con fermiones adjuntos. Las masas efectivas y las funciones de correlación coinciden, confirmando la corrección del manejo de la representación de números reales y la implementación general.

Significancia
El artículo afirma que HiRep v2 es actualmente la suite de software de alto rendimiento de código abierto más flexible que soporta simulaciones de representaciones superiores de fermiones de Wilson con grupos de gauge generales. Su importancia radica en permitir estudios cuantitativos de la física BSM, como los modelos de Higgs compuesto y el límite de gran $N_g$, que anteriormente estaban limitados por la falta de software optimizado. Al proporcionar una herramienta robusta, escalable y acelerada por GPU, HiRep v2 facilita simulaciones a gran escala de masas fermiónicas ligeras en redes grandes, esenciales para las pruebas de precisión de teorías más allá del Modelo Estándar. Los autores señalan que el software es adecuado para su despliegue en supercomputadores de última generación y representa un paso significativo hacia adelante en las capacidades computacionales disponibles para la teoría de campos en redes.