AMD и Intel объявили о совместной работе над ускорением PyTorch на процессорах x86. Компании выбрали открытую библиотеку oneDNN в качестве основного средства такого ускорения. Об этом они сообщили 20 октября, приняв рекомендацию x86 Ecosystem Advisory Group.
AMD переносит в oneDNN оптимизации своей библиотеки ZenDNN. Компания также вошла в руководящий состав UXL Foundation, под управлением которой развивается oneDNN. Производители намерены объединить работу над оптимизациями, чтобы разработчикам приложений PyTorch не приходилось поддерживать отдельные программные решения для процессоров AMD и Intel.
oneDNN содержит оптимизированные операции для нейросетей: умножение матриц, свёртки, механизмы внимания и нормализацию. Библиотека определяет возможности процессора во время выполнения и генерирует код с учётом набора инструкций, числа ядер и размера кеша. Она уже используется в PyTorch, TensorFlow и OpenVINO.
Через будущие обновления oneDNN в PyTorch планируют добавить поддержку AI Compute Extensions — нового набора инструкций x86 для ускорения матричных вычислений. AMD и Intel разрабатывают ACE совместно.
