神经网络处理单元的控制方法及装置、电子设备

本公开提供了一种神经网络处理单元的控制方法及装置、电子设备，涉及机器学习技术领域。该方法应用于拓扑结构相同的多核神经网络处理单元，该方法包括：将神经网络处理单元中各处理核心的拓扑结构划分为相同的多个阶段；每个阶段包括至少一个分支，每个分支包括至少一个节点；确定各处理核心的每个阶段包括的每个分支的执行顺序；按照执行顺序执行各处理核心的每个阶段包括的每个分支。本公开解决了多核NPU的多个核在执行过程中没有考虑任务节点之间共享权重数据的因素，没有按照相似的执行顺序执行任务节点，导致执行过程中重复加载权重数据，占用较多带宽，影响处理器性能的技术问题。