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

权利要求书

1.一种联邦学习中DMA连接建立方法，其特征在于，应用于数据发送方，所述数据发送方包括至少一个发送方计算进程，所述联邦学习中DMA连接建立方法包括：

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

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

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

2.如权利要求1所述联邦学习中DMA连接建立方法，其特征在于，所述通过所述发送方转发进程获取所述目标进程标识对应的目标计算进程的DMA通信地址信息的步骤包括：

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

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

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

3.如权利要求1所述联邦学习中DMA连接建立方法，其特征在于，所述通过所述发送方转发进程获取所述目标进程标识对应的目标计算进程的DMA通信地址信息的步骤包括：

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

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

4.如权利要求1所述联邦学习中DMA连接建立方法，其特征在于，所述通过所述发送方转发进程获取所述目标进程标识对应的目标计算进程的DMA通信地址信息的步骤包括：

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

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

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

5.如权利要求4所述联邦学习中DMA连接建立方法，其特征在于，在所述通过所述发送方转发进程在本地缓存中查询所述目标进程标识对应的DMA通信地址信息的步骤之前，所述联邦学习中DMA连接建立方法还包括：

通过所述发送方转发进程将各所述发送方计算进程的发送方进程标识以及发送方DMA通信地址信息广播至各数据接收方，以供所述数据接收方将各所述发送方进程标识以及各所述发送方DMA通信地址信息对应存放至接收方本地缓存；

接收各所述数据接收方发送的接收方进程标识以及对应的接收方DMA通信地址信息，并将各所述接收方进程标识以及各所述接收方DMA通信地址信息对应存放至发送方本地缓存。

6.一种联邦学习中DMA连接建立方法，其特征在于，应用于数据接收方，所述联邦学习中DMA连接建立方法包括：

接收数据发送方发送的通信地址请求，所述通信地址请求携带发送方计算进程产生的目标进程标识；

通过接收方转发进程将所述目标进程标识对应的目标计算进程的DMA通信地址信息反馈至所述数据发送方；其中，所述目标计算进程的DMA通信地址信息用于所述发送方计算进程与所述目标计算进程之间建立DMA连接。

7.如权利要求6所述联邦学习中DMA连接建立方法，其特征在于，在所述通过接收方转发进程将所述目标进程标识对应的目标计算进程的DMA通信地址信息反馈至所述数据发送方的步骤之前，所述联邦学习中DMA连接建立方法还包括：

查询是否存在所述目标进程标识对应的目标计算进程；

若存在，则查询所述目标计算进程对应的DMA通信地址信息，并执行步骤：通过接收方转发进程将所述目标进程标识对应的目标计算进程的DMA通信地址信息反馈至所述数据发送方；

若不存在，则不对所述通信地址请求进行应答。