機器學習的軟體函式庫

Arm 運算函式庫是針對 Cortex-A CPU, Neoverse 和 Mali 繪圖處理器架構最佳化的低階機器學習函數的集合。該函式庫是在核可的 MIT 授權下可用的開源軟體。

Arm 運算函式庫提供優於其他開放原始碼替代品的效能，並立即支援新的 Arm 技術，例如 SVE2。

主要特色：

• 超過 100 個用於 CPU 和繪圖處理器的機器學習函數
• 多種卷積演算法 (GEMM、Winograd、FFT 和 Direct)
• 支援多種資料類型：FP32、FP16、int8、uint8、BFloat16
• 關鍵機器學習原語的微架構最佳化
• 高度可配置的建構選項，支援輕量級二進位檔
• 進階最佳化技術，例如核心融合、快速數學支援和紋理利用
• 使用 Open CL 調諧器 和 GEMM 最佳化的啟發式方法對裝置和工作負載進行特定調校