Программы
AMD и Intel объединят оптимизации PyTorch для x86-процессоров в oneDNN
Публикация:
AMD и Intel договорились свести свои оптимизации процессоров для PyTorch в библиотеку oneDNN. Компании рассчитывают упростить разработку и перенос ИИ-приложений между x86-системами; AMD интегрирует в проект свои наработки ZenDNN.
AMD и Intel объявили о сотрудничестве в области ускорения ИИ-нагрузок на x86-процессорах. Компании планируют объединить оптимизации PyTorch в библиотеке oneDNN, которую развивает фонд Unified Acceleration Foundation. По их словам, общий программный путь должен уменьшить фрагментацию и упростить разработку, сопровождение и развёртывание приложений на процессорах обеих марок.
AMD собирается интегрировать оптимизации ZenDNN в oneDNN и присоединилась к UXL Foundation как управляющий участник. AMD и Intel также назвали oneDNN библиотекой по умолчанию для ускорения x86 в PyTorch после рекомендации x86 Ecosystem Advisory Group.
Отдельно компании продолжают совместную работу над AI Compute Extensions, набором инструкций x86 для ускорения матричных вычислений. Его поддержку планируют добавить в PyTorch через будущие обновления oneDNN. Сроки интеграции в объявлении не указаны.
Соглашение описывает дальнейшую работу над библиотеками. Компании не назвали сроков появления изменений в PyTorch и не привели замеров производительности. Практический эффект будет зависеть от поддержки конкретных процессоров и выхода обновлений oneDNN и PyTorch. Общая библиотека может упростить сопровождение ИИ-приложений на разных x86-системах.