[发明专利]一种基于硬件拓扑的AI服务器HCA卡性能测试方法及系统

说明书

本发明提供一种基于硬件拓扑的AI服务器HCA卡性能测试方法及系统，包括：获取HCA卡插槽以及上行PCIe的设备信息、带宽速率；根据所述上行PCIe的设备信息识别PCIe的switch芯片类型；根据CPU直连PCIe卡槽的延时标准以及厂商提供的PCIe switch芯片的延时数据计算实际延时；根据所述上行PCIe的带宽速率和实际延时重新制定HCA卡测试通过标准；以所述HCA卡测试通过标准为测试标准，进行所述硬件拓扑的AI服务器HCA卡性能测试。本发明结合HCA卡上行PCIe的情况，重新制定延时新标准，避免标准差异造成测试结果的误判，对测试人员对测试结果的分析提供了新的依据，有利于整体测试工作质量的提升。

技术领域

本发明属于芯片测试技术领域，具体涉及一种基于硬件拓扑的AI服务器HCA卡性能测试方法及系统。

背景技术

HCA卡作为AI服务器的关键部件，可满足多种领域的计算需求，对于AI服务器来说，HCA卡不仅能实现服务器之间的通信，也能实现RDMA技术，因此，HCA卡带宽延时性能是AI服务器测试人员重要的关注点，也是判断HCA卡兼容性测试是否通过的主要依据。

目前通用的HCA卡延时标准是CPU 直连 PCIe 卡槽的标准，但AI服务器机型PCIe拓扑结构复杂，各个槽位实际与CPU连接链路不尽相同，有些槽位实际与CPU直连，没有PCIeswitch芯片带来的延时影响，有些槽位可能经多个PCIe switch芯片，影响HCA卡的带宽及延时，通用的延时标准无法适用于AI服务器HCA卡性能测试，因此针对AI服务器HCA卡的测试标准需要结合AI机型实际硬件拓扑进行判断 。与此同时，通用的HCA卡兼容性测试中，对于PCIe标准槽位速率检测往往只考量HCA卡所在插槽，对于AI服务器，除了测试HCA卡所在插槽的PCIe带宽速率，也需要对上游PCIe的带宽速率进行测试，而这项测试也需基于硬件拓扑，故现有的测试方式具有局限性和片面性。

发明内容

针对现有技术的上述不足，本发明提供一种基于硬件拓扑的AI服务器HCA卡性能测试方法及系统，以解决上述技术问题。

第一方面，本发明提供一种基于硬件拓扑的AI服务器HCA卡性能测试方法，包括：

获取HCA卡插槽以及上行PCIe的设备信息、带宽速率；

根据所述上行PCIe的设备信息识别PCIe的switch芯片类型；

根据CPU直连PCIe卡槽的延时数据以及厂商提供的PCIe switch芯片的延时数据计算实际延时；

根据所述上行PCIe的带宽速率和实际延时重新制定HCA卡测试通过标准；

以所述HCA卡测试通过标准为测试标准，进行所述硬件拓扑的AI服务器HCA卡性能测试。

进一步的，所述方法还包括：

采用shell脚本的形式获取硬件拓扑的PCIe树形结构；

通过所述PCIe树形结构中各节点的配置可以获取上行PCIe的设备信息。

进一步的，所述获取HCA卡插槽以及上行PCIe的设备信息、带宽速率，包括：

遍历HCA卡，抓取HCA卡的BUS_ID以及所在PCIe的带宽速率

获取上行PCIe的设备信息；

调用函数获取BUS_ID对应的上行PCIe的带宽速率；

进一步的，所述上行PCIe的设备信息包括PCIe的总线、桥和设备类型。

进一步的，所述实际延时的计算公式为：实际延时=CPU直连PCIe卡槽的延时+（PCIe switch芯片的延时）*2。

进一步的，所述方法还包括：