[发明专利]用于多处理单元的数据协同处理方法及装置

说明书

本公开实施例公开了一种用于多处理单元的数据协同处理方法、装置、计算机可读存储介质及电子设备，其中，该方法包括：通过多处理单元中的第一处理单元对预设的神经网络中的目标子网络对应的第一组数据进行运算，得到运算结果数据；生成用于向至少一个第二处理单元发送提示本次运算结束的第一信号；检测每个第二处理单元生成的表示每个第二处理单元对目标子网络对应的第二组数据运算结束的第二信号；基于目标子网络在神经网络中的位置，生成指示第一处理单元进一步针对神经网络的操作的指令。本公开实施例实现了在多个处理单元执行神经网络的运算任务期间，自动对各个运算阶段进行协同，提高了神经网络运算的效率。

技术领域

本公开涉及计算机技术领域，尤其是一种用于多处理单元的数据协同处理方法、装置、计算机可读存储介质及电子设备。

背景技术

异构多核处理器是由主处理器和加速器两种体系结构处理器搭建而成的计算机系统。通过协调地使用性能、结构各异的处理单元，能够满足不同的计算需求，以获取最大总体性能方式完成加速计算。而神经网络由于结构复杂，数据计算量大，因此，可以设计专门用于神经网络计算的神经网络加速处理单元。集成在同一芯片内的多个神经网络加速处理单元可以并行地执行同一神经网络的计算任务，从而提高神经网络运算的速度。多个神经网络加速处理单元在对同一帧数据进行处理时，可以将对该帧的处理任务划分为多个处理阶段，各阶段之间需要由CPU(中央处理单元)与多个神经网络加速处理单元通过发送指令完成各神经网络加速处理单元的协调运行。

发明内容

本公开的实施例提供了一种用于多处理单元的数据协同处理方法、装置、计算机可读存储介质及电子设备。

本公开的实施例提供了一种用于多处理单元的数据协同处理方法，该方法包括：通过多处理单元中的第一处理单元对对预设的神经网络中的目标子网络对应的第一组数据进行运算，得到运算结果数据；生成用于向多处理单元包括的至少一个第二处理单元发送提示本次运算结束的第一信号；检测至少一个第二处理单元中的每个第二处理单元生成的表示每个第二处理单元对目标子网络对应的第二组数据运算结束的第二信号；基于目标子网络在神经网络中的位置，生成用于指示第一处理单元进一步针对神经网络的操作的指令。

根据本公开实施例的另一个方面，提供了一种用于多处理单元的数据协同处理装置，该装置包括：运算模块，用于通过多处理单元中的第一处理单元对对预设的神经网络中的目标子网络对应的第一组数据进行运算，得到运算结果数据；第一生成模块，用于生成用于向多处理单元包括的至少一个第二处理单元发送提示本次运算结束的第一信号；检测模块，用于检测至少一个第二处理单元中的每个第二处理单元生成的表示每个第二处理单元对目标子网络对应的第二组数据运算结束的第二信号；第二生成模块，用于基于目标子网络在神经网络中的位置，生成用于指示第一处理单元进一步针对神经网络的操作的指令。

根据本公开实施例的另一个方面，提供了一种计算机可读存储介质，计算机可读存储介质存储有计算机程序，计算机程序用于执行上述用于多处理单元的数据协同处理方法。

根据本公开实施例的另一个方面，提供了一种电子设备，电子设备包括：处理器；用于存储处理器可执行指令的存储器；处理器，用于从存储器中读取可执行指令，并执行指令以实现上述用于多处理单元的数据协同处理方法。