PyTorch — популярен фреймворк за машинно обучение — обновен до версия 2.10.
Сред основните нововъведения можем да отбележим:
Поддръжка на AMD ROCm:
- Въведена е поддръжка на групирана GEMM.
- Подобрена е поддръжката на ROCm за Windows.
- Добавени са нови модели GPU GFX1150/GFX1151 (серия AI 300) в списъка с поддръжка на hipblaslt.
- Разширени са функционалностите, като поддръжката на scaled_mm v2 и AOTriton scaled_dot_product_attention.
Поддръжка на GPU Intel:
- Въведени са нови API Torch XPU за Intel.
- Поддръжка на допълнителни оператори ATen.
- Оптимизация на производителността за GPU Intel.
Поддръжка на NVIDIA CUDA:
- Разширени възможности за писане на шаблонни ядра.
- Подобрена е поддръжката на CUDA 13.
Поддръжка на Python 3.14 за torch.compile(), както и експериментална поддръжка на Python 3.14 без глобално блокиране на интерпретатора (free-threaded).
Намаляване на разходите при стартиране на ядра благодарение на хоризонталното сливане на комбо-ядра в Torch Inductor.
Пълен списък с промени е наличен на github.
Готови варианти, компилирани за различни GPU, ОС и езици (Python/C++/Java), са посочени на сайта Pytorch.
Източник: linux.org.ru
