一种面向类脑芯片的脉冲神经网络编译方法和系统

    公开(公告)号:CN119692404A

    公开(公告)日:2025-03-25

    申请号:CN202411863894.7

    申请日:2024-12-18

    Applicant: 湖南大学

    Inventor: 全哲 黄梓修 徐源

    Abstract: 本发明公开了一种面向类脑芯片的脉冲神经网络编译方法,其通过在TVM中扩展脉冲神经网络Relay算子,接入并绑定至硬件算子库,自动化实现从脉冲神经网络模型到类脑芯片适配的中间表示转换和优化,结合面向脉冲神经网络模型的算子融合优化、面向类脑芯片的数据布局转换等优化方法,充分利用硬件算子库的高性能实现;最终基于TVM基础设施、LLVM编译器生成部署文件包,高效完成脉冲神经网络模型在类脑芯片上的编译与部署。本发明能够解决现有基于DSL的神经网络编译方法和基于硬件intrinsic函数或算子API的神经网络编译方法存在的学习成本高、开发效率低的技术问题。

Patent Agency Ranking