AMD et Intel prévoient de réunir dans oneDNN leurs travaux d’optimisation de PyTorch pour les processeurs x86. AMD doit y intégrer les optimisations de son projet ZenDNN afin de proposer une base logicielle commune aux processeurs des deux fabricants. L’annonce fixe une orientation, pas une date de livraison, et ne présente aucun résultat de performance.
2
3
18
Ce que les deux entreprises ont annoncé
À la suite d’une recommandation du groupe consultatif de l’écosystème x86, AMD et Intel ont choisi oneDNN comme bibliothèque open source de référence pour l’accélération x86 dans PyTorch. AMD prévoit d’y intégrer les optimisations de ZenDNN, tout en continuant à optimiser ses propres processeurs.
2
18
L’objectif affiché est de faciliter le développement et l’optimisation de charges de travail PyTorch sur les CPU AMD comme Intel, plutôt que d’imposer aux développeurs des parcours logiciels distincts selon le fabricant. Jeff McVeigh, d’Intel, a résumé cette ambition : les développeurs ne devraient pas avoir à gérer des chemins différents pour tirer parti des performances des processeurs x86.
1
L’annonce ne détaille pas les critères techniques qui ont conduit le groupe consultatif à recommander oneDNN. Elle présente toutefois cette bibliothèque partagée comme un moyen de simplifier le développement et de limiter la fragmentation.
1
2
La place de ZenDNN dans oneDNN
Intel a commencé le développement de oneDNN, une bibliothèque open source désormais gouvernée par la Unified Acceleration Foundation (UXL). ZenDNN, de son côté, propose des optimisations pour l’architecture Zen d’AMD. L’intégration envisagée ferait entrer ce travail spécifique à AMD dans le projet commun : elle ne signifie pas que les optimisations propres à chaque processeur vont disparaître.
2
6
13
18
AMD a également rejoint l’UXL en tant que membre du comité de pilotage. Cela lui donne un rôle dans la gouvernance de la fondation, mais les sources disponibles ne précisent ni ses droits de vote ni les effets concrets de cette arrivée sur les décisions du projet.
2
9
Calendrier et AI Compute Extensions : les inconnues
Les entreprises n’ont indiqué ni date d’achèvement, ni version cible de PyTorch, ni version de oneDNN pour l’intégration de ZenDNN. Elles n’ont pas non plus publié de mesures de gains de performance.
3
4
AMD et Intel annoncent par ailleurs de futures mises à jour de oneDNN pour prendre en charge les AI Compute Extensions (ACE), conçues pour accélérer les multiplications de matrices. Une prise en charge initiale des instructions ACE figure déjà dans la version candidate 3.14 de oneDNN, mais elle n’est pas activée par défaut : il faut l’autoriser avec ONEDNN_MAX_CPU_ISA=AVX10_2_ACE. Cette version candidate ne permet pas de savoir quand ACE sera disponible dans PyTorch.
7
10
Pour les développeurs, le point concret est donc un projet de bibliothèque et de voie d’intégration communes, et non un changement déjà déployé dans toutes les installations de PyTorch. Le calendrier et les performances obtenues restent à confirmer.