[发明专利]一种基于高速信号切换芯片的数据传输方法、装置及介质

说明书

本申请公开了一种基于高速信号切换芯片的数据传输方法、装置及存储介质，包括：获取与目标高速信号切换芯片设置有物理连接关系的第一GPU和第二GPU的GPU接口以及第一CPU和第二CPU的CPU接口；按照预设通信规则在目标高速信号切换芯片中设置第一GPU的第一GPU接口和第二CPU的第二CPU接口的通信连接关系、第二GPU的第二GPU接口和第一CPU的第一CPU接口的通信连接关系，得出对应的传输路径；当存在待传输数据时，从多条传输路径中选取目标传输路径，并利用目标传输路径传输待传输数据。本方法能够提高不同处理端中的GPU之间的数据传输效率。

技术领域

本发明涉及数据传输领域，特别涉及一种基于高速信号切换芯片的数据传输方法、装置、设备及计算机可读存储介质。

背景技术

近年来，HPC(High performance computing，高性能计算)在服务器中的应用越来越广泛。如图1为现有技术提出的一种HPC的结构示意图，在搭配NVIDIA的CUDA(computeunified Device Architecture，统一计算设备架构)中，同一处理端中的GPU通过高速信号切换芯片SW与CPU做信息交换，不同的处理端中的GPU通过不同处理端中的CPU之间的CPU传输链路做信息交换，如，GPU1～GPU4与GPU5～GPU8间的信息交换，则需要通过CPU0与CPU1之间的CPU传输链路实现数据传输。随着计算机技术的快速发展，对服务器的效能要求越来越高，现有技术中的数据传输方式已逐渐不能满足日益增长的数据传输需求。

因此，如何提高不同处理端中的GPU的数据传输效率，是本领域技术人员目前需要解决的技术问题。

发明内容

有鉴于此，本发明的目的在于提供一种基于高速信号切换芯片的数据传输方法，能够提高不同处理端中的GPU之间的数据传输效率；本发明的另一目的是提供一种基于高速信号切换芯片的数据传输装置、设备及计算机可读存储介质，均具有上述有益效果。

为解决上述技术问题，本发明提供一种基于高速信号切换芯片的数据传输方法，包括：

获取与目标高速信号切换芯片设置有物理连接关系的第一GPU和第二GPU的GPU接口以及与所述目标高速信号切换芯片设置有物理连接关系的第一CPU和第二CPU的CPU接口；其中，所述第一GPU和所述第一CPU属于第一处理端，所述第二CPU和所述第二CPU属于第二处理端；

按照预设通信规则在所述目标高速信号切换芯片中设置所述第一GPU的第一GPU接口和所述第二CPU的第二CPU接口的通信连接关系、所述第二GPU的第二GPU接口和所述第一CPU的第一CPU接口的通信连接关系，得出对应的传输路径；

当存在待传输数据时，从多条所述传输路径中选取目标传输路径，并利用所述目标传输路径传输所述待传输数据。

优选地，进一步包括：

记录预设时间段内利用各所述传输路径传输所述待传输数据的次数。

优选地，进一步包括：

在所述目标高速信号切换芯片中设置所述第一GPU接口和所述第一CPU接口的通信连接关系、所述第二GPU接口和所述第二CPU接口的通信连接关系，得出对应的传输路径。

优选地，所述第一CPU接口数量与所述第二GPU的数量相同，所述第二CPU的接口数量与所述第一GPU的数量相同。

优选地，所述当存在待传输数据时，从多条所述传输路径中选取目标传输路径，并利用所述目标传输路径传输所述待传输数据具体为：

当存在所述待传输数据时，根据所述待传输数据的数据类型从多条所述传输路径中选取所述目标传输路径，并利用所述目标传输路径传输所述待传输数据。