AMD e Intel adotam oneDNN para acelerar IA no PyTorch
A AMD e a Intel decidiram unir esforços na otimização do framework PyTorch para tarefas de inteligência artificial em processadores com arquitetura x86. A base comum será a biblioteca de código aberto de aprendizado profundo oneDNN, desenvolvida com o apoio da fundação Unified Acceleration Foundation (UXL), que se dedica a tecnologias abertas de aceleração de computação.
O acordo foi alcançado levando em conta a recomendação do grupo consultivo x86 Ecosystem Advisory Group. Segundo o entendimento, a oneDNN será utilizada como principal biblioteca de código aberto para a aceleração de hardware de IA no PyTorch em processadores x86. Com isso, os desenvolvedores poderão contar com uma base de software unificada, em vez de mecanismos de otimização separados para o hardware de cada fabricante.
De acordo com as informações divulgadas, a AMD já começou a transferir otimizações de sua própria biblioteca ZenDNN para a base de código da oneDNN. Além disso, a empresa passou a integrar o comitê diretor da UXL Foundation. Participantes do mercado esperam que essa medida reduza a fragmentação do ecossistema de software x86.
A próxima etapa da colaboração será o desenvolvimento das AI Compute Extensions, um conjunto de extensões do conjunto de instruções x86 destinado a acelerar a multiplicação de matrizes, isto é, operações amplamente utilizadas no treinamento e na execução de redes neurais. O suporte às novas instruções deverá ser integrado ao PyTorch por meio de atualizações da biblioteca oneDNN.
No entanto, a tecnologia ainda está em fase de desenvolvimento. O comunicado oficial não esclarece quando as extensões estarão prontas nem quando chegarão aos processadores de produção em série.
A Intel observou que um caminho de software único evitará que os desenvolvedores tenham de escolher diferentes ferramentas de otimização conforme o fabricante do processador. A AMD destacou que a cooperação favorece o desenvolvimento de um ecossistema de software aberto e facilita a criação de aplicações de IA de alto desempenho. Um representante da Red Hat, que participa da comunidade PyTorch, também apoiou a unificação das ferramentas.
Vale ressaltar que o acordo diz respeito principalmente à infraestrutura de software e ao seu desenvolvimento futuro. Não se trata do lançamento de um novo processador nem de um acelerador já pronto. A biblioteca comum deverá simplificar o uso dos recursos já existentes da AMD e da Intel e, após a chegada das AI Compute Extensions, oferecer uma forma unificada de conectar as novas funções de hardware ao PyTorch.
