[发明专利]算子库的测试方法、测试系统及存储介质

说明书

一种算子库的测试方法、测试系统及存储介质，该算子库的测试方法包括：使用统一测试框架接收待测试的目标算子的测试参数信息；通过统一测试框架，根据测试参数信息，分别调用用于目标算子的多个算子过程子程序；运行多个算子过程子程序得到测试用指令包，其中，测试用指令包用于被运行以得到测试结果。该算子库的测试方法例如可以统一并自动化了算子库测试流程与测试过程，从而解决了单独测试大量算子需要重复工作的问题，减轻了例如细粒度算子库测试系统的后续开发与维护工作。

技术领域

本公开的实施例涉及一种算子库的测试方法、测试系统及存储介质。

背景技术

当前，随着人工智能技术的发展，越来越多的应用场景不仅需要软件实现的人工智能算法模型，更需要处理效率更高、专用性更强的硬件产品处理大量的任务和数据。人工智能算法模型在硬件上的实现可以通过将人工智能算法拆解为多个处理步骤，每个处理步骤对应一个算子，该算子在硬件产品上的功能或性能表现需要经过测试来验证。

发明内容

本公开一些实施例提供了一种算子库的测试方法，该方法包括：使用统一测试框架接收待测试的目标算子的测试参数信息；通过所述统一测试框架，根据所述测试参数信息，分别调用用于所述目标算子的多个算子过程子程序；运行所述多个算子过程子程序得到测试用指令包，其中，所述测试用指令包用于被运行以得到测试结果。

例如，在本公开一些实施例提供的一种算子库的测试方法中，所述运行所述多个算子过程子程序得到测试用指令包，包括：生成用于所述目标算子的标准比对数据，其中，所述标准比对数据包括标准比对输入数据和标准比对输出数据；生成与所述目标算子相对应的算子内核源程序；生成与所述目标算子相对应的算子内核描述信息；编译所述算子内核源程序并得到所述目标算子的编译后算子内核程序；根据所述标准比对数据、所述算子内核描述信息以及所述编译后算子内核程序生成所述测试用指令包。

例如，在本公开一些实施例提供的一种算子库的测试方法中，所述多个算子过程子程序包括：标准比对数据子程序，用于生成用于所述目标算子的所述标准比对数据；源程序生成子程序，用于生成与所述目标算子相对应的所述算子内核源程序；指令包生成子程序，用于编译所述算子内核源程序并得到所述目标算子的所述编译后算子内核程序，根据所述标准比对数据、所述算子内核描述信息以及所述编译后算子内核程序生成所述测试用指令包。

例如，在本公开一些实施例提供的一种算子库的测试方法中，所述运行所述多个算子过程子程序得到测试用指令包，包括：使用测试工具运行所述测试用指令包以得到测试结果。

例如，在本公开一些实施例提供的一种算子库的测试方法中，所述使用测试工具运行所述测试用指令包以得到测试结果，包括：向所述编译后算子内核程序输入所述标准比对输入数据；基于所述算子内核描述信息，执行所述编译后算子内核程序并获得执行结果；以及将所述执行结果与所述标准比对输出数据进行比较，并得到所述目标算子的测试结果。

例如，在本公开一些实施例提供的一种算子库的测试方法中，所述测试参数信息包括：测试用例描述信息，用于描述所述测试用例对应的算子的属性；标准比对数据配置信息，用于配置用于生成标准比对数据的参数；算子内核执行参数配置信息，用于配置生成所述算子内核的源程序的参数以及配置生成所述测试用指令包的参数。

例如，在本公开一些实施例提供的一种算子库的测试方法中，所述目标算子的类型包括：张量类、向量类或标量类。

例如，在本公开一些实施例提供的一种算子库的测试方法中，所述使用统一测试框架接收待测试的目标算子的测试参数信息，包括：将所述测试参数信息记载于同一配置文件中；使用所述统一测试框架读取所述配置文件，以接收待测试的目标算子的测试参数信息。