Исторический сдвиг: AMD вносит код в upstream
Фонд UXL Foundation анонсировал релиз oneDNN 3.14. Это не просто очередное обновление, а важный момент в экосистеме ИИ-вычислений: AMD впервые начинает вносить оптимизации для своих процессоров Zen напрямую в основную ветку oneDNN, а не через свой форк ZenDNN. Ранее Intel и AMD объявили о совместной работе над оптимизациями для PyTorch, и теперь этот процесс материализуется в коде.
Технические детали для AMD Zen
Оптимизации для процессоров AMD Zen касаются примитивов MATMUL и inner-product. Важно отметить, что поддержка не включена по умолчанию — разработчикам необходимо явно активировать её при сборке. Для этого используется флаг ONEDNN_X64_USE_ZEN=ON в CMake. На данный момент код ZenDNN все еще требуется внутри дерева oneDNN, что можно сделать вручную или через автоматическую загрузку с флагом ONEDNN_ZENDNN_FETCH=ON.
Поддержка нового железа: ACE, Xeon и GPU
Релиз также добавляет начальную поддержку инструкций AI Compute Extensions (ACE), которые появятся в будущих процессорах AMD и Intel. Кроме того, улучшена производительность для будущих серверных процессоров Intel Xeon Diamond Rapids (поддержка AVX10.2 и AMX) и графических ускорителей Intel серии D (Crescent Island), а также для Xe3P-LPG в процессорах Nova Lake P.
Сводная таблица ключевых изменений
| Компонент / Технология | Статус в oneDNN 3.14 | Требования / Примечания |
|---|---|---|
| AMD Zen CPU | Оптимизации MATMUL и inner-product | Требует сборки с ONEDNN_X64_USE_ZEN=ON и наличия кода ZenDNN |
| AI Compute Extensions (ACE) | Начальная поддержка | Для будущих CPU AMD и Intel |
| Intel Xeon Diamond Rapids | Улучшение производительности | Поддержка AVX10.2 и AMX |
| Intel GPU D-Series | Улучшение производительности | Архитектура Crescent Island |
| Intel Nova Lake P | Улучшение производительности | Графика Xe3P-LPG |
Полный список изменений и возможность загрузки доступны на GitHub проекта oneDNN. Для пользователей PyTorch, TensorFlow и OpenVINO это обновление означает более эффективное использование аппаратных ресурсов как от Intel, так и от AMD в ближайшем будущем.
Источник: Phoronix ↗