CANN/ops-tensor算子测试编写指南

算子测试编写指南

【免费下载链接】ops-tensorops-tensor 是 CANN （Compute Architecture for Neural Networks）算子库中提供张量类计算的基础算子库，采用模块化设计，支持灵活的算子开发和管理。项目地址: https://gitcode.com/cann/ops-tensor

文件结构

写一个算子的测试需要2 个文件：

src/<op_name>/tests/ ├── <op_name>_test.h # 头文件（必需） └── <op_name>_test.cpp # 实现文件（必需）

说明：测试框架提供了ElementwiseBinaryTestConfig辅助类和ExecuteElementwiseBinaryTest通用函数，简化测试编写。

头文件（必需）

文件：<op_name>_test.h

内容：

#pragma once #include "test_common.h" namespace <OpName>Test { void run_all_tests(aclrtStream stream, OpsTensorTest::TestStats& stats); }

说明：

• 命名空间：<OpName>Test（首字母大写，用于隔离不同算子的测试）
• 函数名：run_all_tests（固定名称，测试框架会调用此函数）
• 参数说明：
  • aclrtStream stream- ACL 运行流，用于执行算子
  • TestStats& stats- 测试统计对象，用于记录测试结果

实现文件（必需）

文件：<op_name>_test.cpp

必须包含的 3 个部分：

1. 测试用例函数（必需，至少1个）

函数签名：

void test_xxx(aclrtStream stream, OpsTensorTest::TestStats& stats);

参数说明：

• stream- 传递给算子调用，用于在 NPU 上执行
• stats- 传递给所有测试宏（TEST_ASSERT、TEST_CASE_PASS 等），用于自动统计

示例：

void test_<test_case_name>(aclrtStream stream, OpsTensorTest::TestStats& stats) { TEST_CASE_BEGIN("test_<test_case_name>"); // 准备数据 std::vector<float> A{1.0f, 2.0f}; std::vector<float> C{2.0f, 3.0f}; std::vector<float> expected{3.0f, 5.0f}; std::vector<float> D(2); // 使用测试框架提供的配置类 ElementwiseBinaryTestConfig config(2); // 一维，大小为2 // 调用通用测试函数（自动处理内存管理、数据拷贝等） acltensorStatus_t result = ExecuteElementwiseBinaryTest( A.data(), C.data(), D.data(), config, ACLTENSOR_OP_<OP_NAME>, stream); // 验证结果 TEST_ASSERT(stats, result == ACLTENSOR_STATUS_SUCCESS, "failed"); TEST_ASSERT_ARRAY_NEAR(stats, D, expected, 2, 1e-6f, "mismatch"); TEST_CASE_PASS(stats, "test_<test_case_name>"); }

2. run_all_tests 实现（必需）

namespace <OpName>Test { void run_all_tests(aclrtStream stream, OpsTensorTest::TestStats& stats) { test_xxx(stream, stats); // 可以添加更多测试函数 } }

3. 自动注册（必需）

REGISTER_OP_TEST(<OpName>)

完整示例

<op_name>_test.h

#pragma once #include "test_common.h" namespace <OpName>Test { void run_all_tests(aclrtStream stream, OpsTensorTest::TestStats& stats); }

<op_name>_test.cpp

#include "cann_ops_tensor.h" #include "<op_name>_test.h" #include <vector> void test_<case_name>(aclrtStream stream, OpsTensorTest::TestStats& stats) { TEST_CASE_BEGIN("test_<case_name>"); std::vector<float> A{1.0f, 2.0f}; std::vector<float> C{2.0f, 3.0f}; std::vector<float> expected{3.0f, 5.0f}; std::vector<float> D(2); ElementwiseBinaryTestConfig config(2); acltensorStatus_t result = ExecuteElementwiseBinaryTest( A.data(), C.data(), D.data(), config, ACLTENSOR_OP_<OP_NAME>, stream); TEST_ASSERT(stats, result == ACLTENSOR_STATUS_SUCCESS, "failed"); TEST_ASSERT_ARRAY_NEAR(stats, D, expected, 2, 1e-6f, "mismatch"); TEST_CASE_PASS(stats, "test_<case_name>"); } namespace <OpName>Test { void run_all_tests(aclrtStream stream, OpsTensorTest::TestStats& stats) { test_<case_name>(stream, stats); } } REGISTER_OP_TEST(<op_name>)

可用的宏

用例控制

TEST_CASE_BEGIN("test_name") // 标记开始 TEST_CASE_PASS(stats, "test_name") // 标记通过

断言

TEST_ASSERT(stats, condition, "error") // 条件断言 TEST_ASSERT_ARRAY_EQ(stats, actual, expected, size, "error") // 精确比较 TEST_ASSERT_ARRAY_NEAR(stats, actual, expected, size, tol, "error") // 容差比较

编译运行

./build.sh --ops=<op_name> --run

参考示例

完整示例：src/add/tests/add_test.cpp

【免费下载链接】ops-tensorops-tensor 是 CANN （Compute Architecture for Neural Networks）算子库中提供张量类计算的基础算子库，采用模块化设计，支持灵活的算子开发和管理。项目地址: https://gitcode.com/cann/ops-tensor