CANN算子仓CSV用例指南
ops-transformer 算子仓特性
【免费下载链接】cannbot-skillsCANNBot 是面向 CANN 开发的用于提升开发效率的系列智能体，本仓库为其提供可复用的 Skills 模块。项目地址: https://gitcode.com/cann/cannbot-skills
CSV 格式输入用例
transformer 仓在现有 UT 框架的基础上，封装了一层 CSV 导入工具。
使用 csv 格式输入的优势：
- 数据与代码分离，用例框架编译完成后，仅修改用例表格时无需重复编译
- 避免重复代码，对于多参数算子，可大幅缩减用例长度，缩小模型上下文
对于 transformer 仓的算子，优先使用 CSV 格式创建用例。
触发条件
repo_type为"ops-transformer"- 用户提及"CSV"、"CSV整改"、"CSV格式"、"CSV化"
模板文件
模板文件位于 assets/csv-refactor/：
| 文件 | 用途 |
|---|---|
template_param.h | 参数结构体模板 |
template_tiling.cpp | Tiling 测试模板 |
template_tiling.csv | Tiling CSV 示例 |
template_infershape.cpp | InferShape 测试模板 |
template_infershape.csv | InferShape CSV 示例 |
template_inferdtype.cpp | InferDataType 测试模板 |
template_inferdtype.csv | InferDataType CSV 示例 |
template_cmake.txt | CMakeLists.txt 模板 |
自动化工具
python scripts/generate_csv_template.py {op_name}_def.cpp详细文档
- CSV 重构流程详解
- CSV 格式规范
- 从旧代码提取参数
多架构支持
ops-transformer 仓支持多种芯片架构，需要特别注意：
SoC 与架构对应关系
| SoC 参数 | 架构 | 编译示例 |
|---|---|---|
ascend910b | arch32 | --soc=ascend910b |
ascend950 | arch35 | --soc=ascend950 |
编译注意事项
重要：编译时必须指定--soc参数，否则对应架构的代码不会被编译！
# 编译 arch35 用例 bash build.sh --ophost_test --noexec --ops=<op_name> --soc=ascend950 # 编译 arch32 用例 bash build.sh --ophost_test --noexec --ops=<op_name> --soc=ascend910b运行注意事项
运行测试时需要设置BUILD_PATH环境变量：
export BUILD_PATH=/path/to/ops-transformer/build ./build/tests/ut/framework_normal/op_host/transformer_op_host_ut --gtest_filter="*Arch35*"【免费下载链接】cannbot-skillsCANNBot 是面向 CANN 开发的用于提升开发效率的系列智能体，本仓库为其提供可复用的 Skills 模块。项目地址: https://gitcode.com/cann/cannbot-skills
创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考