AMD e Intel planean reunir sus optimizaciones de CPU x86 para PyTorch en oneDNN, una biblioteca abierta de aprendizaje profundo. AMD incorporará optimizaciones de su proyecto ZenDNN para crear una vía de software común para los procesadores de ambas compañías. Por ahora, el anuncio marca un plan, no una fecha de entrega: tampoco incluye pruebas de rendimiento.
2
3
18
Qué anunciaron las compañías
Tras una recomendación del x86 Ecosystem Advisory Group, AMD e Intel acordaron usar oneDNN como biblioteca abierta predeterminada para la aceleración x86 en PyTorch. AMD integrará en ella las optimizaciones de ZenDNN y seguirá trabajando en ajustes específicos para sus procesadores.
2
18
La meta declarada es que resulte más sencillo desarrollar y optimizar cargas de trabajo de PyTorch en CPU de AMD e Intel, sin tener que mantener rutas de software separadas para cada fabricante. Jeff McVeigh, de Intel, resumió la idea así: que los desarrolladores puedan aprovechar el rendimiento de las CPU x86 sin tener que gestionar caminos distintos.
1
El anuncio disponible no explica los criterios técnicos detallados que llevaron al grupo asesor a recomendar oneDNN. Sí presenta la biblioteca compartida como una forma de simplificar el desarrollo y reducir la fragmentación.
1
2
Qué aporta ZenDNN a oneDNN
Intel inició el desarrollo de oneDNN, que actualmente está bajo la gobernanza de la Unified Acceleration Foundation (UXL). ZenDNN, por su parte, incorpora optimizaciones para la arquitectura Zen de AMD. La integración prevista sumará ese trabajo específico de AMD al proyecto común; no implica que vayan a desaparecer los ajustes particulares para cada procesador.
2
6
13
18
AMD también se incorporó a UXL como miembro del comité directivo. Esto le da participación en la gobernanza de la fundación, aunque las fuentes disponibles no detallan sus derechos de voto ni el efecto concreto de su incorporación en las decisiones del proyecto.
2
9
Lo que se sabe sobre los plazos y las AI Compute Extensions
Las compañías no han indicado una fecha de finalización, una versión concreta de PyTorch ni una versión de oneDNN para la integración de ZenDNN. Tampoco han comunicado mejoras de rendimiento medidas.
3
4
AMD e Intel prevén futuras actualizaciones de oneDNN para las AI Compute Extensions (ACE), diseñadas para acelerar la multiplicación de matrices. La versión candidata de oneDNN 3.14 ya ofrece soporte inicial para instrucciones ACE, pero no lo activa de forma predeterminada: hay que habilitarlo mediante ONEDNN_MAX_CPU_ISA=AVX10_2_ACE. Ese soporte preliminar no permite saber cuándo estarán disponibles las ACE a través de PyTorch.
7
10
Para quienes desarrollan con PyTorch, la conclusión práctica es que se prepara una biblioteca y una ruta de integración comunes; no que el cambio ya esté disponible en todas las instalaciones. El calendario y el rendimiento que pueda ofrecer siguen sin confirmarse.