[发明专利]一种集群系统中的通信方法和装置

说明书

技术领域

本发明涉及计算机技术领域，具体涉及一种集群系统中的通信方法和装置。

背景技术

随着计算机技术的迅速发展，高性能集群的优势越来越明显，而GPU集群的作用也越来越突出。与CPU集群相比，GPU集群能够在使用较少操作系统镜像的情况下正常工作，降低管理复杂性，因而在高性能计算应用中非常有吸引力。

现有技术中，集群系统的通信流程包括：在设备上分配内存，从主机向设备复制数据；在设备上执行计算，得到计算结果；将计算结果从设备复制到主机；释放在设备上分配的内存。其中，从主机向设备复制数据，以及将计算结果从设备复制到主机是必需的步骤，也是传输数据最慢的部分，使得集群系统的计算时间较长，导致集群系统的整体计算性能较差。

发明内容

本发明提供了一种集群系统中的通信方法和装置，以解决集群系统的整体计算性能较差的缺陷。

本发明提供了一种集群系统中的通信方法，应用于包括主机和设备的系统中，所述方法包括以下步骤：

在所述主机的内存中为所述设备分配固定内存；

通过所述固定内存实现所述主机与所述设备之间的通信。

可选地，在所述主机的内存中为所述设备分配固定内存，具体为：

从所述主机的内存中选择固定内存，将所述固定内存直接映射到所述设备的内存空间。

可选地，通过所述固定内存实现所述主机与所述设备之间的通信，具体为：

所述设备通过对所述固定内存进行写操作，向所述主机发送数据；

和/或，

所述设备通过对所述固定内存进行读操作，获取来自所述主机的数据。

可选地，通过所述固定内存实现所述主机与所述设备之间的通信，具体为：

使用所述固定内存，在操作队列中进行异步内存复制。

可选地，在所述主机的内存中为所述设备分配固定内存之后，还包括：

释放所述固定内存。

本发明还提供了一种集群系统中的通信装置，应用于包括主机和设备的系统中，所述装置包括：

分配模块，用于在所述主机的内存中为所述设备分配固定内存；

通信模块，用于通过所述分配模块分配的所述固定内存实现所述主机与所述设备之间的通信。

可选地，所述分配模块，具体用于从所述主机的内存中选择固定内存，将所述固定内存直接映射到所述设备的内存空间。

可选地，所述通信模块，具体用于通过对所述固定内存进行读和/或写操作，实现所述主机与所述设备之间的数据收发。

可选地，所述通信模块，具体用于使用所述固定内存，在操作队列中进行异步内存复制。

可选地，所述的装置，还包括：

释放模块，用于释放所述分配模块分配的所述固定内存。

本发明通过使用固定内存传输控制主机与设备之间的数据传输，固定内存始终存在于物理内存中，能够通过DMA加速与设备端进行通信，不会被分配到低速的虚拟内存中，从而省掉了主机和设备之间进行数据拷贝的工作，实现主机和设备之间的最高带宽，消除了硬件传输对集群系统的瓶颈影响，减少集群系统的计算时间，提升集群系统的整体计算性能。

附图说明

图1为本发明实施例中的一种集群系统中的通信方法流程图；

图2为本发明实施例中的一种集群系统中的通信装置的结构示意图。

具体实施方式

下面将结合本发明实施例中的附图，对本发明实施例中的技术方案进行清楚、完整地描述，显然，所描述的实施例仅仅是本发明一部分实施例，而不是全部的实施例。基于本发明中的实施例，本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例，都属于本发明保护的范围。

需要说明的是，如果不冲突，本发明实施例以及实施例中的各个特征可以相互结合，均在本发明的保护范围之内。另外，虽然在流程图中示出了逻辑顺序，但是在某些情况下，可以以不同于此处的顺序执行所示出或描述的步骤。

本发明实施例提供了一种集群系统中的通信方法，应用于包括主机和设备的系统中，其中，主机可以是CPU及系统的内存，设备可以是GPU及GPU本身的显示内存。上述通信方法如图1所示，包括以下步骤：

步骤101，在主机的内存中为设备分配固定内存。

具体地，从主机的内存中选择固定内存，将固定内存直接映射到设备的内存空间。

其中，固定内存会始终驻留在物理内存中，不会被主机的操作系统分页，也不会被主机的操作系统交换到磁盘上，也不会被破坏或重新定位。