AMD et Intel choisissent oneDNN pour l'IA dans PyTorch
AMD et Intel ont décidé de réunir leurs efforts pour optimiser le framework PyTorch dans les tâches d’intelligence artificielle sur les processeurs d’architecture x86. La base commune sera la bibliothèque open source d’apprentissage profond oneDNN, développée avec le soutien de la fondation Unified Acceleration Foundation (UXL), qui se consacre aux technologies ouvertes d’accélération du calcul.
L’accord a été conclu en tenant compte de la recommandation du groupe consultatif x86 Ecosystem Advisory Group. Selon cette entente, oneDNN sera utilisée comme principale bibliothèque open source pour l’accélération matérielle de l’IA dans PyTorch sur les processeurs x86. Les développeurs pourront ainsi s’appuyer sur une base logicielle unique, plutôt que sur des mécanismes d’optimisation distincts pour le matériel de chaque fabricant.
D’après les informations disponibles, AMD a déjà commencé à transférer les optimisations de sa propre bibliothèque ZenDNN vers le code de oneDNN. L’entreprise a en outre rejoint le comité de pilotage de la UXL Foundation. Les acteurs du secteur espèrent que cette démarche réduira la fragmentation de l’écosystème logiciel x86.
La prochaine étape de la collaboration sera la création des AI Compute Extensions, un ensemble d’extensions du jeu d’instructions x86 destiné à accélérer la multiplication matricielle, c’est-à-dire des opérations très utilisées pour l’entraînement et l’exécution des réseaux de neurones. La prise en charge des nouvelles instructions doit être intégrée à PyTorch par le biais de mises à jour de la bibliothèque oneDNN.
Cette technologie est toutefois encore en cours de développement. Le communiqué officiel ne précise ni la date de disponibilité des extensions ni le moment où elles apparaîtront dans les processeurs de série.
Intel a indiqué qu’une voie logicielle unique évitera aux développeurs de choisir différents outils d’optimisation selon le fabricant du processeur. AMD a souligné que cette coopération favorise le développement d’un écosystème logiciel ouvert et facilite la création d’applications d’IA hautes performances. Un représentant de Red Hat, qui participe à la communauté PyTorch, a également soutenu l’unification des outils.
Il convient de souligner que l’accord concerne principalement l’infrastructure logicielle et son évolution future. Il ne s’agit ni du lancement d’un nouveau processeur ni d’un accélérateur prêt à l’emploi. La bibliothèque commune doit simplifier l’utilisation des capacités déjà existantes d’AMD et d’Intel et, après l’arrivée des AI Compute Extensions, offrir un moyen unifié de connecter les nouvelles fonctions matérielles à PyTorch.
