Intel випустила нову версію бібліотеки нейронних мереж глибокого навчання oneDNN

Intel випустила нову версію бібліотеки oneDNN з відкритим вихідним кодом (раніше відому як MKL-DNN). Використовується для глибокого вивчення нейронних мереж. Нова версія 2.1 тепер має підтримку прискорювачів на базі графічного процесора від NVIDIA, а також ряд поліпшень для роботи на майбутніх процесорах Intel.

Зазначається, що нова бібліотека оптимізована для роботи з графікою Intel Xe. Поліпшена робота з процесорами Intel Xeon з підтримкою матричних розширень AMX (вони отримають чіпи Sapphire Rapids). Також підтримуються різні оптимізації AVX-512 і так далі.

Є початкова підтримка майбутніх процесорів Alder Lake з AVX2 і DL Boost. Є навіть деякі ОПТИМІЗАЦІЇ INT8 для процесорів з SSE4.1 і ряд інших низькорівневих поліпшень. Крім того, є поліпшення архітектури AArch64. І бібліотека в цілому стала працювати швидше.

Що стосується підтримки графічного процесора NVIDIA, то вона все ще знаходиться на початковій стадії і спирається на компілятор Intel DPC ++, cuDNN і cuBLAS. Однак це важливий крок для платформи oneAPI, яка обіцяє стати універсальним інструментом розробки для різноманітних апаратних архітектур. Посилання для завантаження та всі подробиці про випуск oneDNN 2.1 можна знайти на GitHub.