AMD и Intel объединят оптимизации процессоров x86 для PyTorch в библиотеке oneDNN

AMD и Intel объединят оптимизации процессоров x86 для PyTorch в библиотеке oneDNN

Arkadiy Andrienko

AMD и Intel объявили о сотрудничестве вокруг oneDNN, чтобы ускорить PyTorch на процессорах x86. По замыслу компаний, такой подход должен упростить разработку, поддержку и развёртывание ИИ-приложений на системах с x86-процессорами.

AMD перенесёт оптимизации ZenDNN в oneDNN. Сроки переноса не указаны. Единая библиотека должна сделать экосистему программ для x86 менее раздробленной. Разработчикам станет проще создавать, поддерживать и развёртывать приложения PyTorch без отдельных программных путей для разных процессоров.

oneDNN используют не только PyTorch, но и TensorFlow, ONNX Runtime и OpenVINO. Джефф Маквей (Jeff McVeigh), вице-президент и генеральный менеджер Intel по стратегическим программам центров обработки данных, объяснил, зачем нужен единый подход.

Союз во имя ИИ
Союз во имя ИИ
Разработчикам не придётся выбирать отдельные программные решения, чтобы получить максимум от процессора x86. Благодаря сотрудничеству AMD и Intel в рамках oneDNN сообщество PyTorch получит единый подход к работе с современными процессорами, а в будущем в платформу можно будет добавить новые возможности, включая AI Compute Extensions.
— Джефф Маквей, вице-президент Intel

AMD также присоединилась к UXL Foundation в качестве управляющего участника, чтобы содействовать дальнейшей совместной работе. Кроме того, компании работают над AI Compute Extensions — набором инструкций x86 для ускорения матричных умножений. Будущую поддержку этих расширений в PyTorch планируют интегрировать через обновления oneDNN.

Станет ли единая библиотека достаточным решением для упрощения разработки ИИ-приложений на x86, или разработчикам всё ещё понадобятся отдельные оптимизации?

🤯 В «Геншине» теперь можно учиться кодить на Python
    Об авторе
    Комментарии0