AMD e Intel planejam reunir no oneDNN o trabalho de otimização do PyTorch para CPUs x86. A AMD vai integrar otimizações do projeto ZenDNN à biblioteca de código aberto, criando um caminho comum de software para processadores das duas empresas. Por enquanto, trata-se de um plano: o anúncio não informa prazo de conclusão nem apresenta resultados de desempenho.
2
3
18
O que foi anunciado
Após uma recomendação do x86 Ecosystem Advisory Group, AMD e Intel escolheram o oneDNN como biblioteca de código aberto padrão para aceleração x86 no PyTorch. A AMD pretende incorporar ao projeto as otimizações do ZenDNN e, ao mesmo tempo, continuar ajustando seus processadores.
2
18
A ideia é facilitar o desenvolvimento e a otimização de cargas de trabalho do PyTorch em CPUs AMD e Intel, sem exigir que desenvolvedores mantenham caminhos de software separados para cada fabricante. Jeff McVeigh, da Intel, resumiu o objetivo dizendo que eles não deveriam precisar lidar com rotas diferentes para aproveitar o desempenho das CPUs x86.
1
O anúncio disponível não detalha os critérios técnicos usados pelo grupo consultivo para recomendar o oneDNN. As empresas apresentam a biblioteca compartilhada como uma forma de simplificar o desenvolvimento e reduzir a fragmentação do ecossistema.
1
2
Onde entra o ZenDNN
A Intel começou a desenvolver o oneDNN, que hoje é governado pela Unified Acceleration Foundation (UXL). Já o ZenDNN reúne otimizações voltadas à arquitetura Zen, da AMD. A integração planejada leva esse trabalho específico da AMD para o projeto comum — mas não significa que os ajustes próprios para cada processador vão deixar de existir.
2
6
13
18
A AMD também passou a integrar a UXL como membro do comitê diretor. Isso dá à empresa um papel na governança da fundação, embora as fontes disponíveis não especifiquem seus direitos de voto nem o impacto exato de sua participação nas decisões do projeto.
2
9
Prazo e AI Compute Extensions ainda estão em aberto
As empresas não divulgaram uma data para concluir a integração do ZenDNN, nem indicaram uma versão específica do PyTorch ou do oneDNN para recebê-la. Também não anunciaram ganhos de desempenho medidos.
3
4
AMD e Intel planejam futuras atualizações do oneDNN para oferecer suporte às AI Compute Extensions (ACE), projetadas para acelerar multiplicações de matrizes. Uma versão candidata do oneDNN 3.14 já inclui suporte inicial às instruções ACE, mas ele não é ativado por padrão: é preciso habilitá-lo com ONEDNN_MAX_CPU_ISA=AVX10_2_ACE. Esse suporte preliminar não informa quando as ACE estarão disponíveis por meio do PyTorch.
7
10
Para quem desenvolve com PyTorch, a notícia aponta para uma biblioteca comum e um caminho de integração planejado — não para uma mudança já implementada em todas as instalações. O cronograma e o desempenho que essa colaboração poderá entregar ainda não foram definidos.