一种基于深度优先调度技术的卷积神经网络编译器

本发明属于人工智能芯片技术领域，具体涉及一种基于深度优先调度技术的卷积神经网络编译器。本发明卷积神经网络编译器包括：配置模块、前端处理模块、层融合组划分模块、深度优先调度模块、成本模型模块、和代码生成模块；该编译器将ONNX模型转化为内部计算图，采用动态规划算法划分层融合组，对各层融合组采用新型深度优先调度技术实现算子切割及调度。本发明可有效提升神经网络编译效果，减少冗余计算，实现更充分的片上数据重用，从而提升卷积神经网络在资源受限的神经网络加速器上的推理速度。