AMD та Intel планують об’єднати оптимізації PyTorch для процесорів x86 у бібліотеці oneDNN. AMD інтегрує туди напрацювання ZenDNN, щоб розробники могли працювати зі спільною програмною основою для процесорів обох компаній. Поки що це план: компанії не назвали дату завершення інтеграції й не навели результатів тестів продуктивності.
2
3
18
Що саме домовилися зробити
Після рекомендації дорадчої групи з екосистеми x86 AMD та Intel обрали oneDNN основною відкритою бібліотекою для прискорення обчислень на x86 у PyTorch. AMD інтегрує оптимізації ZenDNN у спільний проєкт і водночас продовжить окремо оптимізувати роботу на своїх процесорах.
2
18
PyTorch — популярний фреймворк для розробки систем штучного інтелекту, а oneDNN — відкрита бібліотека, що допомагає ефективно виконувати такі обчислення на процесорах. x86 — поширена архітектура, на якій працюють процесори AMD та Intel.
Задум полягає в тому, щоб розробникам було простіше створювати й оптимізувати програми для процесорів обох виробників, не підтримуючи окремі програмні шляхи для кожного. Представник Intel Джефф Маквей висловив цю мету так: розробникам не доводилося б окремо розбиратися, як отримати продуктивність x86-процесорів.
1
У повідомленнях про домовленість немає докладного пояснення, за якими технічними критеріями дорадча група рекомендувала саме oneDNN. Компанії наголошують на спрощенні розробки та зменшенні фрагментації програмного середовища.
1
2
Що буде із ZenDNN і роботою над процесорами AMD
oneDNN започаткувала Intel, а нині бібліотекою опікується Unified Acceleration Foundation (UXL) — фонд, що підтримує відкриті технології прискорення обчислень. ZenDNN — оптимізаційна бібліотека AMD, налаштована для її архітектури Zen. Планована інтеграція переносить ці напрацювання до спільного проєкту, але не означає відмови від оптимізацій під конкретні процесори.
2
6
13
18
AMD також приєдналася до UXL як керівний учасник. Це дає компанії роль у роботі фонду, однак доступні повідомлення не уточнюють її права голосу чи те, як саме новий статус вплине на ухвалення рішень.
2
9
Строки й підтримка AI Compute Extensions
Компанії не назвали термін завершення інтеграції ZenDNN, конкретну версію PyTorch або випуск oneDNN, у якому вона буде готова. Вони також не оприлюднили виміряних показників приросту швидкодії.
3
4
AMD та Intel планують додати до майбутніх оновлень oneDNN підтримку AI Compute Extensions (ACE) — розширень, призначених для прискорення множення матриць. Початкова підтримка ACE вже є в тестовій версії oneDNN 3.14, але за замовчуванням вона вимкнена. Щоб увімкнути її, потрібно встановити змінну середовища ONEDNN_MAX_CPU_ISA=AVX10_2_ACE.
7
10
Наявність ACE у тестовій версії бібліотеки сама по собі не означає, що функція вже доступна через PyTorch або коли саме вона там з’явиться. Для розробників практичний висновок поки такий: формується спільна програмна основа, але строки її впровадження та фактичний вплив на продуктивність залишаються невідомими.