项目简介
本项目是一个针对Arm(R) Ethos(TM)-U NPU(神经网络处理器)的驱动程序。采用CMake构建系统,支持交叉编译,适用于不同的Arm Cortex(R)-M CPU,提供对Ethos-U设备的初始化、配置、执行命令流以及释放资源等操作的支持。
项目的主要特性和功能
- 初始化(ethosu_init):设置设备基础地址、快速内存、安全状态和特权模式。
- 配置(ethosu_set_basep_cache_mask):设置基础地址的缓存刷新和无效化掩码。
- 命令流处理(ethosu_invoke_async, ethosu_invoke_v3):接收自定义数据并执行特定操作。
- 复位(ethosu_soft_reset):软复位Ethos-U设备。
- 请求/释放电源(ethosu_request_power, ethosu_release_power):在推理前后管理电源。
- 获取硬件信息(ethosu_get_hw_info):获取Ethos-U设备的硬件信息。
- 等待推理完成(ethosu_wait):等待推理完成或超时并处理结果。
- 获取驱动版本信息(ethosu_get_driver_version):获取Ethos-U驱动的版本信息。
- 注册/注销驱动(ethosu_register_driver, ethosu_deregister_driver):管理驱动在全局列表中的注册状态。
- 保留/释放驱动(ethosu_reserve_driver, ethosu_release_driver):确保每次只有一个推理任务运行。
安装使用步骤
- 环境准备:安装适用于目标平台的交叉编译工具链,确保CMake可用。
- 编译:配置CMake以匹配目标平台的CPU和NPU配置,执行构建命令生成驱动。
- 驱动使用:使用生成的驱动API进行设备初始化、配置、执行命令流等。
注意:驱动的具体使用方式依赖于应用程序需求,实际使用需参考更详细的文档和示例代码。
下载地址
点击下载 【提取码: 4003】【解压密码: www.makuang.net】