[发明专利]数据传输方法、处理器、芯片及电子设备

说明书

本公开涉及一种数据传输方法、处理器、芯片及电子设备，通过处理器分多次进行多个计算核之间的数据传输，在m个计算核之间进行数据传输的情况下，在每次发送中，将m个计算核中的每个计算核的数据分别发送给m个计算核中一个目标计算核；在m个计算核与n个计算核之间进行数据传输的情况下，根据m个计算核和n个计算核的存储部件的容量，调整m个计算核每次发送的数据量大小。根据本公开实施例的能够克服传输过程中因发送数据量过多造成的拥堵及发送数据量过少造成资源浪费，减少路由延时，提高芯片的计算效率。

技术领域

本公开涉及信息处理技术领域，尤其涉及一种数据传输方法、处理器、芯片及电子设备。

背景技术

深度神经网络(DNNs)为有很多隐藏层的神经网络，一个芯片的计算资源不可以满足大规模的深度神经网络的需求，如果将大规模的网络切分映射在单个计算核上计算的时间将非常大。在有限的众核簇情况下，每个计算核的内存资源有限，需要对神经网络的输入图像数据进行切分，将一个大规模的DNNs切分映射到芯片群上。整个切分过程中，计算核簇内、簇间存在的大量数据交换，如交叠数据，如部分和数据，如图形整合等等。

发明内容

有鉴于此，本公开提出了一种数据传输方法、处理器、芯片及电子设备。

根据本公开的一方面，提供了一种数据传输方法，所述方法用于处理器的计算核之间的数据传输，每个处理器包括多个计算核，每个处理器内部的计算核之间、以及不同处理器的计算核之间能够进行数据传输；其中，每个计算核包括存储部件，所述存储部件用于存储与其他计算核进行传输的数据；所述方法包括：分多次进行所述计算核之间的数据传输，其中，在m个计算核之间进行数据传输的情况下，在每次发送中，将m个计算核中的每个计算核的数据分别发送给m个计算核中一个目标计算核，其中，每个计算核对应的目标计算核各不相同；或者，在m个计算核与n个计算核之间进行数据传输的情况下，根据m个计算核和n个计算核的存储部件的容量，调整m个计算核每次发送的数据量大小，其中m个计算核与n个计算核存在重叠或不存在重叠。

在一种可能的实现方式中，在m个计算核之间进行数据传输的情况下，任意m-1个计算核组成的各计算核组的数据需要传输至作为目标计算核的其余的一个计算核，在每次发送中，将m个计算核中的每个计算核的数据分别发送给m个计算核中一个目标计算核，包括：在每次发送中，从各计算核组中，分别选取一个不同的计算核，将该计算核的数据发送至对应的目标计算核，共发送m-1次。

在一种可能的实现方式中，根据m个计算核和n个计算核的存储部件的容量，调整m个计算核每次发送的数据量大小，包括：根据m个计算核和n个计算核的存储部件的容量，调整m个计算核每次发送的数据量大小，使得每次发送中，m个计算核发送的数据总量最大，且m个计算核中的每一个发送的数据总量小于该计算核的存储部件的容量，n个计算核中的每一个接收的数据总量小于该计算核的存储部件的容量。

在一种可能的实现方式中，根据m个计算核和n个计算核的存储部件的容量，调整m个计算核每次发送的数据量大小，包括：确定约束条件，所述约束条件包括每次发送中，m个计算核中的每一个发送的数据总量小于该计算核的存储部件的容量，n个计算核中的每一个接收的数据总量小于该计算核的存储部件的容量；建立目标函数，所述目标函数使得每次发送中，m个计算核发送的数据总量最大；在满足所述约束条件的情况下，求解所述目标函数，获得传输方案，所述传输方案包括m个计算核中的每一个在每次传输中向n个计算核中的各计算核发送的数据量。