[发明专利]一种多通道数据传输方法、装置和存储介质

权利要求书

1.一种多通道数据传输方法，其特征在于，所述方法包括以下步骤：

将需要传输的数据划分为数据块，并将所述数据块按传输通道的带宽比例分配到不同的传输通道中；

通过不同的传输通道对需要传输的数据进行环形通信；

其中，所述传输通道包括PCIE传输通道、NVSWITCH传输通道和PCIE+QPI传输通道。

2.根据权利要求1所述的多通道数据传输方法，其特征在于，所述将需要传输的数据划分为数据块前，还包括：

确定需要进行数据传输的GPU；

确定GPU之间需要传输的数据。

3.根据权利要求1所述的多通道数据传输方法，其特征在于，所述将所述数据块按传输通道的带宽比例分配到不同的传输通道前，还包括：

执行数据传输的GPU之间建立PCIE环形通信环路和NVSWITCH环形通信环路；

其中，所述PCIE环形通信环路中的相邻两个GPU之间形成所述PCIE传输通道，所述NVSWITCH环形通信环路中的相邻两个GPU之间形成所述NVSWITCH传输通道。

4.根据权利要求3所述的多通道数据传输方法，其特征在于，所述执行数据传输的GPU之间建立PCIE环形通信环路和NVSWITCH环形通信环路，具体包括：

获取GPU的拓扑信息；

根据GPU的拓扑信息按照约束条件建立所述NVSWITCH环形通信环路；

清除GPU的拓扑信息和约束条件并建立所述PCIE环形通信环路。

5.根据权利要求4所述的多通道数据传输方法，其特征在于，所述建立所述PCIE环形通信环路，还包括：

判断GPU之间是否跨区，如果跨区，则建立PCIE+QPI环形通信环路；如果没有跨区，则建立PCIE环形通信环路；

其中，所述PCIE+QPI环形通信环路中的相邻两个GPU之间形成所述PCIE+QPI传输通道。

6.根据权利要求1所述的多通道数据传输方法，其特征在于，所述将所述数据块按传输通道的带宽比例分配到不同的传输通道中，具体包括：

计算不同的传输通道之间的带宽比例；

将所述数据块按不同的传输通道之间的带宽比例分配到不同的传输通道中。

7.根据权利要求1所述的多通道数据传输方法，其特征在于，所述通过不同的传输通道对需要传输的数据进行环形通信，还包括：

通过同步函数确保数据块在不同的传输通道中同时传输完成。

8.一种多通道数据传输装置，其特征在于，所述装置包括：

数据拆分模块，用于将需要传输的数据拆分为数据块；

数据分配模块，用于将数据块按照带宽比例分配到不同的传输通道中；

环形通信模块，用于实现GPU之间数据的环形通信；

传输通道模块，用于传输需要传输的数据；

其中，所述传输通道模块包括PCIE传输通道模块、NVSWITCH传输通道模块和PCIE+QPI传输通道模块。

9.根据权利要求8所述的多通道数据传输装置，其特征在于，所述装置还包括：

环路建立模块，用于建立PCIE环形通信环路、NVSWITCH环形通信环路和PCIE+QPI环形通信环路；

拓扑信息模块，用于获取及清除GPU的拓扑信息；

函数同步模块，用于确保数据块在不同的传输通道中同时传输完成。

10.一种计算机可读存储介质，其特征在于：所述计算机可读存储介质存储有程序，当所述程序被处理器执行时，使得所述处理器执行如权利要求1~7中任意一项所述的方法的步骤。