远程直接内存访问的数据重传方法及相关设备与流程

所属的技术人员能够理解，本公开的各个方面可以实现为系统、方法或程序产品。因此，本公开的各个方面可以具体实现为以下形式，即：完全的硬件实施方式、完全的软件实施方式(包括固件、微代码等)，或硬件和软件方面结合的实施方式，这里可以统称为“电路”、“模块”或“系统”。下面参照图10来描述根据本公开的这种实施方式的电子设备1000。图10显示的电子设备1000仅仅是一个示例，不应对本公开实施例的功能和使用范围带来任何限制。如图10所示，电子设备1000以通用计算设备的形式表现。电子设备1000的组件可以包括但不限于：上述至少一个处理单元1010、上述至少一个存储单元1020、连接不同系统组件(包括存储单元1020和处理单元1010)的总线1030。其中，存储单元存储有程序代码，程序代码可以被处理单元1010执行，使得处理单元1010执行本说明书上述“示例性方法”部分中描述的根据本公开各种示例性实施方式的步骤。示例性的，处理单元1010可以执行如图2中的方法实施例的如下步骤：获取多条待重传任务，多条待重传任务的元数据存储于未完成请求表ort中；根据各条待重传任务的元数据确定对应的重传数据包的数据量；根据多个重传数据包的数据量确定重传组合，重传组合中包括至少两个重传子组，各个重传子组内包括至少一个重传数据包，不同重传子组内的重传数据包的数据量满足预设条件；按照预设规则对至少两个重传子组的重传数据包执行数据重传。存储单元1020可以包括易失性存储单元形式的可读介质，例如随机存取存储单元(ram)10201和/或高速缓存存储单元10202，还可以进一步包括只读存储单元(rom)10203。存储单元1020还可以包括具有一组(至少一个)程序模块10205的程序/实用工具10204，这样的程序模块10205包括但不限于：操作系统、一个或者多个应用程序、其它程序模块以及程序数据，这些示例中的每一个或某种组合中可能包括网络环境的实现。总线1030可以为表示几类总线结构中的一种或多种，包括存储单元总线或者存储单元控制器、外围总线、图形加速端口、处理单元或者使用多种总线结构中的任意总线结构的局域总线。电子设备1000也可以与一个或多个外部设备1040(例如键盘、指向设备、蓝牙设备等)通信，还可与一个或者多个使得用户能与该电子设备1000交互的设备通信，和/或与使得该电子设备1000能与一个或多个其它计算设备进行通信的任何设备(例如路由器、调制解调器等等)通信。这种通信可以通过输入/输出(i/o)接口1050进行。并且，电子设备1000还可以通过网络适配器1060与一个或者多个网络(例如局域网(lan)，广域网(wan)和/或公共网络，例如因特网)通信。如图10所示，网络适配器1060通过总线1030与电子设备1000的其它模块通信。应当明白，尽管图中未示出，可以结合电子设备1000使用其它硬件和/或软件模块，包括但不限于：微代码、设备驱动器、冗余处理单元、外部磁盘驱动阵列、raid系统、磁带驱动器以及数据备份存储系统等。通过以上的实施方式的描述，本领域的技术人员易于理解，这里描述的示例实施方式可以通过软件实现，也可以通过软件结合必要的硬件的方式来实现。因此，根据本公开实施方式的技术方案可以以软件产品的形式体现出来，该软件产品可以存储在一个非易失性存储介质(可以是cd-rom，u盘，移动硬盘等)中或网络上，包括若干指令以使得一台计算设备(可以是个人计算机、服务器、终端装置、或者网络设备等)执行根据本公开实施方式的方法。在本公开的示例性实施例中，还提供了一种计算机可读存储介质，该计算机可读存储介质可以是可读信号介质或者可读存储介质。该计算机可读存储介质上存储有能够实现本公开上述方法的程序产品。在一些可能的实施方式中，本公开的各个方面还可以实现为一种程序产品的形式，其包括程序代码，当程序产品在网络设备上运行时，程序代码用于使网络设备执行本说明书上述“示例性方法”部分中描述的根据本公开各种示例性实施方式的步骤。本公开中的计算机可读存储介质的更具体的例子可以包括但不限于：具有一个或多个导线的电连接、便携式计算机磁盘、硬盘、随机访问存储器(ram)、只读存储器(rom)、可擦式可编程只读存储器(eprom或闪存)、光纤、便携式紧凑磁盘只读存储器(cd-rom)、光存储器件、磁存储器件、或者上述的任意合适的组合。在本公开中，计算机可读存储介质可以包括在基带中或者作为载波一部分传播的数据信号，其中承载了可读程序代码。这种传播的数据信号可以采用多种形式，包括但不限于电磁信号、光信号或上述的任意合适的组合。可读信号介质还可以是可读存储介质以外的任何可读介质，该可读介质可以发送、传播或者传输用于由指令执行系统、装置或者器件使用或者与其结合使用的程序。可选地，计算机可读存储介质上包含的程序代码可以用任何适当的介质传输，包括但不限于无线、有线、光缆、rf等等，或者上述的任意合适的组合。在具体实施时，可以以一种或多种程序设计语言的任意组合来编写用于执行本公开操作的程序代码，程序设计语言包括面向对象的程序设计语言—诸如java、c++等，还包括常规的过程式程序设计语言—诸如“c”语言或类似的程序设计语言。程序代码可以完全地在用户计算设备上执行、部分地在用户设备上执行、作为一个独立的软件包执行、部分在用户计算设备上部分在远程计算设备上执行、或者完全在远程计算设备或服务器上执行。在涉及远程计算设备的情形中，远程计算设备可以通过任意种类的网络，包括局域网(lan)或广域网(wan)，连接到用户计算设备，或者，可以连接到外部计算设备(例如利用因特网服务提供商来通过因特网连接)。应当注意，尽管在上文详细描述中提及了用于动作执行的设备的若干模块或者单元，但是这种划分并非强制性的。实际上，根据本公开的实施方式，上文描述的两个或更多模块或者单元的特征和功能可以在一个模块或者单元中具体化。反之，上文描述的一个模块或者单元的特征和功能可以进一步划分为由多个模块或者单元来具体化。此外，尽管在附图中以特定顺序描述了本公开中方法的各个步骤，但是，这并非要求或者暗示必须按照该特定顺序来执行这些步骤，或是必须执行全部所示的步骤才能实现期望的结果。附加的或备选的，可以省略某些步骤，将多个步骤合并为一个步骤执行，以及/或者将一个步骤分解为多个步骤执行等。通过以上实施方式的描述，本领域的技术人员易于理解，这里描述的示例实施方式可以通过软件实现，也可以通过软件结合必要的硬件的方式来实现。因此，根据本公开实施方式的技术方案可以以软件产品的形式体现出来，该软件产品可以存储在一个非易失性存储介质(可以是cd-rom，u盘，移动硬盘等)中或网络上，包括若干指令以使得一台计算设备(可以是个人计算机、服务器、移动终端、或者网络设备等)执行根据本公开实施方式的方法。本领域技术人员在考虑说明书及实践这里公开的发明后，将容易想到本公开的其它实施方案。本公开旨在涵盖本公开的任何变型、用途或者适应性变化，这些变型、用途或者适应性变化遵循本公开的一般性原理并包括本公开未公开的本中的公知常识或惯用技术手段。说明书和实施例仅被视为示例性的，本公开的真正范围和精神由所附的权利要求指出。

背景技术：

1、随着人工智能、云计算、分布式计算等技术的快速发展，面向人工大模型训推的智算中心采用基于远程直接内存访问(remote direct memory access，rdma)的高性能网络，随着大模型的参数体量不断发展以及智算业务租户化发展，智算中心节点也呈现规模化扩张，这对rdma高性能网络的大规模组网能力也提出更高的要求，如何在大规模组网下避免网络报文乱序以及丢包，是其中的一个关键要求。

2、在相关技术中，通过rdma网卡上的报文重传机制解决包乱序以及丢包问题。然而，当频繁出现丢包、乱序包重传时，大数据包对传输通道队列对(queue pair，qp)的持续占用致使小数据包无法及时送达，影响业务。

3、需要说明的是，在上述背景技术部分公开的信息仅用于加强对本公开的背景的理解，因此可以包括不构成对本领域普通技术人员已知的现有技术的信息。

技术实现思路

1、本公开提供一种远程直接内存访问的数据重传方法及相关设备，至少在一定程度上克服相关技术中数据重传导致网络性能下降的问题。

2、本公开的其他特性和优点将通过下面的详细描述变得显然，或部分地通过本公开的实践而习得。

3、根据本公开的一个方面，提供一种远程直接内存访问的数据重传方法，包括：获取多条待重传任务，其中，所述多条待重传任务的元数据存储于未完成请求表ort中；根据各条待重传任务的元数据确定对应的重传数据包的数据量；根据多个重传数据包的数据量确定重传组合，其中，所述重传组合中包括至少两个重传子组，各个重传子组内包括至少一个重传数据包，不同重传子组内的重传数据包的数据量满足预设条件；按照预设规则对所述至少两个重传子组的重传数据包执行数据重传。

4、在本公开的一个实施例中，所述预设规则包括按照数据量由小到大的顺序执行数据重传；其中，所述按照预设规则对所述至少两个重传子组的重传数据包执行数据重传，包括：对于不同的重传子组，按照重传子组内的重传数据包的总数据量由小到大的顺序执行数据重传；对于同一重传子组，按照所述同一重传子组内的至少一个重传数据包的数据量由小到大的顺序执行数据重传。

5、在本公开的一个实施例中，所述预设规则包括按照重传数据包的优先级别执行数据重传；其中，所述按照预设规则对所述至少两个重传子组的重传数据包执行数据重传，包括：为所述各个重传子组分配相同的缓存空间地址，将所述缓存空间地址写入组合复用表中；根据组合复用表中各个重传子组内重传数据包的数据量之和，编排各个重传数据包的优先级别。

6、在本公开的一个实施例中，所述按照预设规则对所述至少两个重传子组的重传数据包执行数据重传，包括：根据各个重传数据包的优先级别，向发送端返回确认字符ack重传应答，以指示所述发送端重传数据包；接收所述发送端的数据包，将所述数据包写入与所述缓存空间地址对应的重传缓存区，以使所述重传缓存区将重传数据包写入应用数据缓存；向所述发送端发送重传数据完成消息。

7、在本公开的一个实施例中，在所述向所述发送端发送重传数据完成消息之前，所述方法还包括：释放所述重传缓存区的缓存空间，更新所述组合复用表和所述ort。

8、在本公开的一个实施例中，在所述获取多条待重传任务之前，包括：当检测到丢包或乱序时，将存在丢包或乱序的报文的元数据存储至所述ort中，以生成待重传任务。

9、在本公开的一个实施例中，所述获取多条待重传任务，包括：周期性扫描所述ort获得所述多条待重传任务。

10、在本公开的一个实施例中，所述根据各条待重传任务的元数据确定对应的重传数据包的数据量，包括：基于所述待重传任务的元数据，在缓存池中查找对应的工作队列元素wqe，解析查找到的weq，得到所述重传数据包的数据量。

11、在本公开的一个实施例中，所述重传组合中包括至少两个重传子组，各个重传子组内包括至少一个重传数据包，不同重传子组内的重传数据包的数据量满足预设条件，包括：将数据量最大的重传数据包作为一个重传子组，各个重传子组内的重传数据包的数据量之和小于或等于所述重传数据包的数据量最大值。

12、根据本公开的另一个方面，提供一种一种远程直接内存访问的数据重传装置，包括：任务获取模块，用于获取多条待重传任务，其中，所述多条待重传任务的元数据存储于未完成请求表ort中；信息确定模块，用于根据各条待重传任务的元数据确定对应的重传数据包的数据量；重传分组模块，用于根据多个重传数据包的数据量确定重传组合，其中，所述重传组合中包括至少两个重传子组，各个重传子组内包括至少一个重传数据包，不同重传子组内的重传数据包的数据量满足预设条件；数据重传模块，用于按照预设规则对所述至少两个重传子组的重传数据包执行数据重传。

13、根据本公开的再一个方面，提供一种电子设备，包括：处理器；以及存储器，用于存储所述处理器的可执行指令；其中，所述处理器配置为经由执行所述可执行指令来执行上述的远程直接内存访问的数据重传方法。

14、根据本公开的又一个方面，提供一种计算机可读存储介质，其上存储有计算机程序，所述计算机程序被处理器执行时实现上述的远程直接内存访问的数据重传方法。

15、根据本公开的另一个方面，提供一种计算机程序产品，所述计算机程序产品包括计算机程序或计算机指令，所述计算机程序或所述计算机指令由处理器加载并执行，以使计算机实现上述的远程直接内存访问的数据重传方法。

16、在本公开实施方式中，获取多条待重传任务，多条待重传任务的元数据存储于未完成请求表中；根据各条待重传任务的元数据确定对应的重传数据包的数据量；根据多个重传数据包的数据量确定重传组合，重传组合中包括至少两个重传子组，各个重传子组内包括至少一个重传数据包，不同重传子组内的重传数据包的数据量满足预设条件；按照预设规则对至少两个重传子组的重传数据包执行数据重传，通过对未完成请求表扫描获得批量待重传任务，根据数据包的大小切分不同的重传组合，从而解决大小包的传输问题，提升业务质量，提升重传效率，缓解重传等待对拥塞造成的不利影响。

17、应当理解的是，以上的一般描述和后文的细节描述仅是示例性和解释性的，并不能限制本公开。