Компанията AMD разкри кода на драйвера за NPU, базиран на архитектурата XDNA

Компания AMD публикува изходния код на драйвера за карти с архитектура на базата XDNA, който предоставя средства за ускорение на изчисления, свързани с машинното обучение и обработка на сигнали (NPU, Neural Processing Unit). NPU на базата на архитектурата XDNA се предлага в сериите 7040 и 8040 на процесорите AMD Ryzen, ускорителите AMD Alveo V70 и SoC AMD Versal. Кодът е написан на езиците C и C++, и е достъпен под лиценз GPLv2. За работа на драйвера е необходимо Linux ядро 6.7 с поддръжка на IOMMU SVA (Shared Virtual Addressing).

Публикуваният код включва драйвер за Linux ядрото (amdxdna.ko) и runtime библиотека (плъгин „xrt_plugin*-amdxdna“) за използване на интерфейса XRT (Xilinx Runtime Library), който позволява на приложенията да се обръщат към обработващите единици (kernel), изпълнявани на страната на хардуерния ускорител. XRT позволява да се използва NPU AMD в приложения на обикновени програмни езици. Предоставят се различни нива на абстракция, от нискоуровневи API за C/C++ до високоуровневи взаимодействия за Python и компоненти за интеграция с TensorFlow, PyTorch и Caffe.

Източник: opennet.ru

Купете надежден хостинг за сайтове със защита от DDoS, VPS и VDS сървъри 🔥 Купете надежден хостинг за сайтове със защита от DDoS, VPS и VDS сървъри | ProHoster