[发明专利]一种人工智能芯片、数据传输方法及数据传输系统

说明书

本发明公开了一种人工智能芯片、数据传输方法及数据传输系统，涉及芯片技术领域，该芯片包括：控制引擎模块，用于将待搬运的第一目标数据组装得到至少一个第一数据包，并将至少一个第一数据包发送给路由功能模块；其中，第一数据包附加有第一目标地址；路由功能模块，用于根据与当前数据传输模式匹配的路由策略，在全部通信接口中确定至少一个第一通信接口，并通过第一通信接口发出第一数据包。本发明实施例的技术方案，使得AI芯片本身具备了路由转发功能，进而以多样化网络模式互联的AI芯片可以满足不同业务场景下的数据搬运需求，避免了构建数据传输系统时对数据交换芯片的依赖，提高了构建完成的数据传输系统的数据搬运速度。

技术领域

本发明涉及芯片技术领域，尤其涉及一种人工智能芯片、数据传输方法及数据传输系统。

背景技术

随着多节点大规模并行计算需求的产生，AI（Artificial Intelligence，人工智能）芯片之间的高速互联通信，成为了芯片技术领域的重要研究课题。

传统的AI芯片的互联方案如图1所示，一个中央处理器下的各个AI芯片，通过PCIe（peripheral component interconnect express，高速串行计算机扩展总线标准）总线的Switch（转换）芯片连接；如图2所示，英伟达（NVIDIA）也提供了基于专用转换芯片（即NV_Switch芯片）的互联方案；如图3所示，寒武纪提供一种基于MLU-Link桥接卡的8芯片互联方案。

然而，图1所示的互联方案，受限于PCIe总线协议以及PCIe拓扑结构的限制，数据带宽往往无法满足实际的负载需求；而图2所示的互联方式，同样需要配置专用转换芯片，不但需要耗费较高的硬件成本，而且大部分芯片系统的互联规模达不到上述规模，常常发生芯片功能冗余现象；图3所示的互联规模只能在8芯片以内，扩展性不足，进而影响芯片系统的性能。

发明内容

本发明提供了一种人工智能芯片、数据传输方法及数据传输系统，以解决人工智能芯片在构建大规模网络时对数据交换芯片的依赖问题。

根据本发明的一方面，提供了一种人工智能芯片，包括：控制引擎模块、路由功能模块以及用于与其他硬件设备互联的多个通信接口；

所述控制引擎模块，用于将待搬运的第一目标数据组装得到至少一个第一数据包，并将所述至少一个第一数据包发送给所述路由功能模块；其中，所述第一数据包附加有第一目标地址；

所述路由功能模块，用于根据与当前数据传输模式匹配的路由策略，在全部通信接口中确定至少一个第一通信接口，并通过所述第一通信接口发出所述第一数据包。

根据本发明的另一方面，提供了一种数据传输方法，应用于本发明任一实施例所述的人工智能芯片，包括：

控制引擎模块将待搬运的第一目标数据组装得到至少一个第一数据包，并将所述至少一个第一数据包发送给路由功能模块；其中，所述第一数据包附加有第一目标地址；

所述路由功能模块基于交叉开关矩阵模式，通过拥塞算法获取匹配的至少一个第一通信接口，并通过所述第一通信接口发出所述第一数据包。

根据本发明的另一方面，提供了一种数据传输方法，应用于本发明任一实施例所述的人工智能芯片，包括：

控制引擎模块将待搬运的第一目标数据组装得到至少一个第一数据包，并将所述至少一个第一数据包发送给路由功能模块；其中，所述第一数据包附加有第一目标地址；

所述路由功能模块基于无线网络网格模式，根据所述第一目标地址和当前人工智能芯片的地址信息获取第一传输路径，并获取所述第一传输路径中第一相邻人工智能芯片对应的第一通信接口；

所述路由功能模块通过所述第一通信接口向所述第一相邻人工智能芯片发出所述第一数据包。