买专利卖专利找龙图腾，真高效！ 查专利查商标用IPTOP,全免费！专利年费监控用IP管家,真方便！

申请/专利权人：杭州质算科技有限公司;杭州幻方人工智能基础研究有限公司;宁波积幂信息科技有限公司;杭州幻方科技有限公司;上海积幂科技有限公司

摘要：本发明涉及计算机技术领域，尤其涉及一种多GPU集群网络的RDMA并行数据传输方法，包括：发送节点和接收节点上，参与发送与接收的GPU都注册内存缓存区并交换信息，并在对应的参与发送GPU和参与接收GPU之间一一建立RDMA连接；主发送GPU调用接口发送，并确定报文切分大小并切片；部分报文切片先由主发送GPU转发到辅助发送GPU；报文切片通过所有参与发送的GPU传输到所有参与接收的GPU，每对RDMA发送连接需要控制发送在途报文切片数量小于等于预设在途最大切片数；辅助接收GPU将报文切片转发到主接收GPU；主接收GPU确认完成。本发明的有益效果：充分利用多张网卡的带宽，提高网络传输效率，减少拥塞。

主权项：1.一种多GPU集群网络的RDMA并行数据传输方法，其特征在于包括以下步骤：S101：发送节点和接收节点上，参与本轮报文数据发送与接收的GPU都注册内存缓存区，交换所述内存缓存区的信息，并在对应的参与发送GPU和参与接收GPU之间一一建立RDMA连接；S102：主发送GPU调用接口发送，并确定报文切分大小，并对发送报文进行切片，得到若干大小相同的报文切片；S103：发送过程开始，部分报文切片先由主发送GPU转发到辅助发送GPU；S104：报文切片通过所有参与发送的GPU传输到所有参与接收的GPU，每对RDMA发送连接需要控制发送在途报文切片数量小于等于预设在途最大切片数；S105：辅助接收GPU将报文切片转发到主接收GPU；S106：主接收GPU确认接收完成；返回发送完成。

全文数据：

权利要求：

百度查询： 杭州质算科技有限公司 杭州幻方人工智能基础研究有限公司 宁波积幂信息科技有限公司 杭州幻方科技有限公司 上海积幂科技有限公司 一种多GPU集群网络的RDMA并行数据传输方法