基于神威处理器的算子融合方法

本发明涉及一种基于神威处理器的算子融合方法，属于算子融合技术领域，包括获取深度学习模型中计算图的算子，分析算子的数据访问模式，根据预设融合判定机制确定符合融合条件的待融合算子，并获取待融合算子的特征图，特征图利用待融合算子进行数据处理，将待融合算子的参数预先加载至神威处理器的局部数据存储器中，判断特征图的总数据流与局部数据存储器的容量的大小，通过判断结果对特征图的数据块进行划分，利用数据块完成待融合算子的运算，并输出运算后的数据块，减少了数据在处理器和外部存储器之间的频繁传输，能够充分利用神威处理器的计算资源，提升并行计算效率，优化内存访问模式，减少内存带宽的瓶颈。