上週,Alexandre Bailon 與 Bay Libre 一起發布了 AI 處理單元“APU”直接渲染管理器驅動程序,用於連接 CPU 和 AI 處理單元。希望這個 APU 驅動程序可以被各種硬件驅動程序重複使用,而最初的重點是提升 MediaTek MT8183 SoC 的 AI 功能。
雖然已經有作為 DRM 子系統的附屬的加速器“accel”框架,但 Bay Libre 開發人員的新提議是讓這個 APU 驅動程序進一步發展這個不斷發展的 AI/加速器內核驅動程序領域。 Bailon 在最初的補丁系列中寫道:
“這添加了一個 DRM 驅動程序,它實現了 CPU 和 APU 之間的通信。該驅動程序針對通常使用一些預構建模型運行推理的嵌入式設備。目標是提供可以重用以支持許多加速器的通用基礎設施。內核、用戶空間和固件都試圖使用標準和現有的來利用開發和維護工作。該系列實現了兩個平台驅動程序,一個用於模擬,另一個用於mt8183(與 mt8365 兼容)。”
提議的 DRM_APU Kconfig 幫助文本繼續添加:
“這提供了一個 DRM 驅動程序,它提供了一些與 AI 處理器單元通信的設施(APU)。該驅動程序旨在提供可用於支持許多不同 APU 的通用基礎結構。”
除了在早期代碼審查中提出的一些技術問題外,還提出了一些僅針對命名的反對意見這個驅動程序“APU”。有人建議將其稱為 apu_if 或 tiny_apu。我們將看看這項努力的結果。有興趣的人可以查看初始補丁系列以了解更多信息APU 驅動程序是不斷發展的 AI 加速器驅動程序領域的最新開源成果。
在其他正在進行的開源 AI 內核驅動程序中,Intel Habana Labs 是該領域最著名和最強大的驅動程序,其次是新的適用於 Meteor Lake 的英特爾通用處理單元 (iVPU),以及高通、東芝和其他處於不同開發階段的驅動程序。