arm-software/computelibrary 是一套面向 Arm CPU 和 Arm GPU 编写的计算机视觉与机器学习底层函数集合。它使用 C++ 编写,通过 SIMD 运算针对 Arm Cortex-A、Arm Neoverse 和 Arm Mali GPU 架构做了函数优化。该仓库的最新发布版本是 v53.3.1。
该仓库不是用于设计模型的框架,而是承载实际计算过程的底层函数集合。它同时面向 CPU 和 GPU 提供 100 多个机器学习函数以及多种卷积(convolution)算法,并支持 FP32、FP16、INT8、UINT8、BFLOAT16 数据类型。可以按精度分别挑选函数,因此在内存与算力需求不同的环境中能提供更多选择。它还提供可按架构配置的构建选项以及 OpenCL 调优工具(tuner)。README 中说明该项目以 MIT 许可证公开。该仓库于 2017 年 3 月开始,目前已积累 3196 个 star。
不过也有一些需要确认的地方。由于提供的发布信息中没有变更记录,无法得知 v53.3.1 具体改动了什么。由于优化以 Arm Cortex-A、Arm Neoverse 和 Arm Mali GPU 为前提,在其他 CPU 或 GPU 环境下能取得怎样的结果也无从得知。MIT 许可证这一说法见于 README,但仓库信息中的许可证字段是空的,实际适用条件需要查阅原文。性能数值与对比对象均未提供,因此难以判断它在真实工作负载中的速度表现。如果要处理 Arm 硬件,这是一个值得优先查看的仓库,但很难把它当作通用的加速库来看待。