联邦学习中DMA连接建立方法.pdf

(19)中华人民共和国国家知识产权局(12)发明专利申请(10)申请公布号CN113900976A(43)申请公布日2022.01.07(21)申请号202111087461.3(22)申请日2021.09.16(71)申请人深圳致星科技有限公司地址518000广东省深圳市南山区粤海街道大冲社区深南大道9676号大冲商务中心（一期）2栋3号楼18C-1(72)发明人张翔宇孙军欢陈沫(74)专利代理机构深圳市恒程创新知识产权代理有限公司44542代理人苗广冬(51)Int.Cl.G06F13/28(2006.01)G06N20/20(2019.01)权利要求书3页说明书15页附图3页(54)发明名称联邦学习中DMA连接建立方法(57)摘要本申请公开了联邦学习中DMA连接建立方法，所述联邦学习中DMA连接建立方法包括：生成当前需要建立DMA连接的发送方计算进程的目标进程标识，并传递至发送方转发进程；其中，所述目标进程标识用于指定需要与所述发送方计算进程建立DMA连接的目标计算进程；通过所述发送方转发进程获取所述目标进程标识对应的目标计算进程的DMA通信地址信息；依据所述DMA通信地址信息，建立所述发送方计算进程与所述目标计算进程之间的DMA连接。本申请解决了现有技术中联邦学习网络传输效率低的技术问题。CN113900976ACN113900976A权利要求书1/3页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通信地址信息，2CN113900976A权利要求书2/3页并将各所述接收方进程标识以及各所述接收方DMA通信地址信息对应存放至发送方本地缓存。6.一种联邦学习中DMA连接建立方法，其特征在于，应用于数据接收方，所述联邦学习中DMA连接建立方法包括：接收数据发送方发送的通信地址请求，所述通信地址请求携带发送方计算进程产生的目标进程标识；通过接收方转发进程将所述目标进程标识对应的目标计算进程的DMA通信地址信息反馈至所述数据发送方；其中，所述目标计算进程的DMA通信地址信息用于所述发送方计算进程与所述目标计算进程之间建立DMA连接。7.如权利要求6所述联邦学