[发明专利]联邦学习中DMA连接建立方法

说明书

本申请公开了联邦学习中DMA连接建立方法，所述联邦学习中DMA连接建立方法包括：生成当前需要建立DMA连接的发送方计算进程的目标进程标识，并传递至发送方转发进程；其中，所述目标进程标识用于指定需要与所述发送方计算进程建立DMA连接的目标计算进程；通过所述发送方转发进程获取所述目标进程标识对应的目标计算进程的DMA通信地址信息；依据所述DMA通信地址信息，建立所述发送方计算进程与所述目标计算进程之间的DMA连接。本申请解决了现有技术中联邦学习网络传输效率低的技术问题。

技术领域

本申请涉及通信技术领域，尤其涉及一种联邦学习中DMA连接建立方法、装置、电子设备、可读存储介质及产品。

背景技术

随着人工智能技术的不断发展，数据传输的方式也在不断换代更新，目前在联邦学习框架通常使用传统的TCP(Transmission Control Protocol，传输控制协议)通信进行不同参与方之间的通信，但是TCP通信过程中通常会在发送侧以及接收侧发生数据拷贝，并且TCP通信过程中稳定传输的功能实现依赖于软件，实际传输速度远低于物理带宽，网络传输效率低下，影响了联邦学习网络传输的效率。

发明内容

本申请的主要目的在于提供一种DMA(Direct Memory Access，远程直接数据存取)连接建立方法、装置、电子设备、可读存储介质及产品，旨在解决现有技术中联邦学习网络传输效率低的技术问题。

为实现上述目的，本申请提供一种联邦学习中DMA连接建立方法，所述联邦学习中DMA连接建立方法应用于数据发送方，所述数据发送方包括至少一个发送方计算进程，所述联邦学习中DMA连接建立方法包括：

生成当前需要建立DMA连接的发送方计算进程的目标进程标识，并传递至发送方转发进程；其中，所述目标进程标识用于指定需要与所述发送方计算进程建立DMA连接的目标计算进程；

通过所述发送方转发进程获取所述目标进程标识对应的目标计算进程的DMA通信地址信息；

依据所述DMA通信地址信息，建立所述发送方计算进程与所述目标计算进程之间的DMA连接。

可选地，所述通过所述发送方转发进程获取所述目标进程标识对应的目标计算进程的DMA通信地址信息的步骤包括：

通过所述发送方转发进程依据所述目标进程标识，确定具备所述目标计算进程的目标数据接收方；

通过所述发送方转发进程将携带所述目标进程标识的通信地址请求发送至所述目标数据接收方；

通过所述发送方转发进程接收所述目标数据接收方发送的DMA通信地址信息，其中，所述目标计算进程的DMA通信地址信息是所述目标数据接收方基于所述所述目标进程标识查询得到的。

可选地，所述通过所述发送方转发进程获取所述目标进程标识对应的目标计算进程的DMA通信地址信息的步骤包括：

通过所述发送方转发进程将携带所述目标进程标识的通信地址请求广播至全部数据接收方，以供各所述数据接收方确定是否存在所述目标进程标识对应的目标计算进程；

通过所述发送方转发进程接收确定存在所述目标计算进程的数据接收方发送的DMA通信地址信息。

可选地，所述通过所述发送方转发进程获取所述目标进程标识对应的目标计算进程的DMA通信地址信息的步骤包括：

通过所述发送方转发进程在本地缓存中查询所述目标进程标识对应的DMA通信地址信息；

若查询成功，则将所述DMA通信地址信息由所述发送方转发进程反馈至所述发送方计算进程；

若查询失败，则依据所述目标进程标识，查找所述发送方计算进程对应的目标计算进程的DMA通信地址信息。