联邦学习中DMA连接建立方法技术

技术编号:31793405 阅读:23 留言:0更新日期:2022-01-08 10:52
本申请公开了联邦学习中DMA连接建立方法,所述联邦学习中DMA连接建立方法包括:生成当前需要建立DMA连接的发送方计算进程的目标进程标识,并传递至发送方转发进程;其中,所述目标进程标识用于指定需要与所述发送方计算进程建立DMA连接的目标计算进程;通过所述发送方转发进程获取所述目标进程标识对应的目标计算进程的DMA通信地址信息;依据所述DMA通信地址信息,建立所述发送方计算进程与所述目标计算进程之间的DMA连接。本申请解决了现有技术中联邦学习网络传输效率低的技术问题。技术中联邦学习网络传输效率低的技术问题。技术中联邦学习网络传输效率低的技术问题。

【技术实现步骤摘要】
联邦学习中DMA连接建立方法


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

技术介绍

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

技术实现思路

[0003]本申请的主要目的在于提供一种DMA(Direct Memory Access,远程直接数据存取)连接建立方法、装置、电子设备、可读存储介质及产品,旨在解决现有技术中联邦学习网络传输效率低的技术问题。
[0004]为实现上述目的,本申请提供一种联邦学习中DMA连接建立方法,所述联邦学习中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连接建立...

【专利技术属性】
技术研发人员:张翔宇孙军欢陈沫
申请(专利权)人:深圳致星科技有限公司
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1