反向工程的瑞芯芯片NPU驱动程序达到与专有驱动程序相同的性能
开源开发者Tomeu Vizoso最近开始为一些最新的瑞芯芯片NPU创建一个开源的反向工程驱动程序。在成功为VeriSilicon NPU IP提供开源NPU驱动程序支持后,Vizoso接受了支持瑞芯微NPU的挑战。有了开源的用户空间驱动程序,他已经让他的第一个模型运行起来了。它不仅可以运行,而且性能与专有驱动程序相似。
令人惊讶的是,在开始他最新的逆向工程驱动冒险几周后,他的开源代码已经启动并在瑞芯芯片NPU上运行MobileNetV1模型。最令人兴奋的是这句话:
“它不仅运行完美,而且与blob具有相同的性能水平。”
他一直享受着快速的成功,这要归功于至少有一些来自瑞芯芯片的记录寄存器信息,他们的驱动程序设计类似于NVIDIA的NVDLA和ONNX,他可以利用文档和代码来帮助进行逆向工程。
这个瑞芯芯片的开源驱动工作也依赖于作为Vivante/Etnaviv NPU驱动工作的一部分上传到Mesa的Teflon框架。
More details on the latest open-source Rockchip NPU driver success via this blog post.