PyTorch — um framework popular para aprendizado de máquina — foi atualizado para a versão 2.10.
Entre as principais novidades, podemos destacar:
Suporte a AMD ROCm:
- Foi introduzido suporte a GEMM agrupada.
- O suporte ao ROCm para Windows foi melhorado.
- Novos modelos de GPU GFX1150/GFX1151 (série AI 300) foram adicionados às listas de suporte hipblaslt.
- Recursos ampliados, como suporte a scaled_mm v2 e AOTriton scaled_dot_product_attention.
Suporte a GPU Intel:
- Novas APIs Torch XPU para Intel foram implementadas.
- Suporte adicional para operadores ATen.
- Otimização de desempenho para GPU Intel.
Suporte a NVIDIA CUDA:
- Recursos expandidos para a escrita de kernels template.
- Melhoria no suporte ao CUDA 13.
Suporte ao Python 3.14 para torch.compile(), além do suporte experimental para a construção do Python 3.14 sem bloqueio global do interpretador (free-threaded).
Redução do overhead no lançamento de kernels devido à fusão horizontal de kernels combinados no Torch Inductor.
A lista completa de alterações está disponível na github.
Versões prontas, compiladas para diferentes GPUs, sistemas operacionais e linguagens (Python/C++/Java) estão disponíveis em site Pytorch
Fonte: linux.org.ru
