CANN/ops-tensor构建脚本参数说明
build.sh 参数说明【免费下载链接】ops-tensorops-tensor 是 CANN Compute Architecture for Neural Networks算子库中提供张量类计算的基础算子库采用模块化设计支持灵活的算子开发和管理。项目地址: https://gitcode.com/cann/ops-tensor简介build.sh 是 ops-tensor 项目的构建脚本位于项目根目录下。该脚本通过配置不同参数实现多种功能包括编译算子、运行测试、生成安装包等。使用方法1. 配置环境变量在使用 build.sh 之前需要先配置 CANN 环境变量# 默认路径安装 source /usr/local/Ascend/cann/set_env.sh # 验证环境变量 echo $ASCEND_HOME_PATH2. 构建命令格式./build.sh [OPTIONS]参数说明build.sh 支持多种功能可通过--help参数查看所有选项./build.sh --help参数必选/可选说明--opsOP_LIST可选指定要编译的算子列表多个算子用逗号分隔如--opsadd,sub。不指定时编译所有算子。--run可选编译后执行测试。需要配合BUILD_TESTINGON使用。--pkg可选编译并打包成 .run 安装包。--socSOC可选指定目标 SoC 型号支持大小写不敏感输入如--socascend950或--socAscend950。默认为Ascend950。当前版本仅支持 Ascend950。-j[N]可选指定编译线程数默认为 8如-j16。若线程数超过 CPU 核心数会自动调整为 CPU 核心数。--test-timeoutN可选指定测试超时时间单位秒默认为 300。仅在--run模式下有效。-h, --help可选显示帮助信息。支持的 SoC 型号SoC 型号SOC_VERSIONCANN 编译器说明Ascend950ascend950dt_9595默认支持dav-3510说明当前版本仅支持Ascend950其他 SoC 型号暂不支持。后续版本将逐步支持更多型号。使用示例基本编译# 编译所有算子默认 8 线程 ./build.sh # 编译指定算子 ./build.sh --opsadd # 编译多个算子 ./build.sh --opsadd,sub # 使用 16 线程编译 ./build.sh -j16编译并测试# 编译所有算子并运行测试 ./build.sh --run # 编译指定算子并运行测试 ./build.sh --opsadd --run # 指定测试超时时间600 秒 ./build.sh --run --test-timeout600打包# 编译所有算子并打包默认 SoC: Ascend950 ./build.sh --pkg # 编译指定算子并打包 ./build.sh --opsadd --pkg # 大小写不敏感当前仅支持 Ascend950 ./build.sh --socascend950 --pkg组合使用# 编译 add 算子、运行测试、使用 16 线程 ./build.sh --opsadd --run -j16 # 编译所有算子、打包默认 SoC: Ascend950 ./build.sh --pkg -j16行为说明命令行为无参数编译所有算子不执行测试--opsadd只编译 add 算子不执行测试--opsadd,sub编译 add 和 sub 算子不执行测试--run编译所有算子并执行所有算子的测试--opsadd --run编译 add 算子并执行 add 算子的测试--opsadd,sub --run编译 add、sub 算子并执行这些算子的测试--pkg编译所有算子并打包成 .run 文件默认 SoC: Ascend950--opsadd --pkg编译 add 算子并打包成 .run 文件--socascend950 --pkg为 Ascend950 芯片打包支持大小写不敏感输出说明编译输出编译成功后生成的文件位于build/目录build/ ├── libops_tensor.so # 动态库 ├── tests/ │ └── all_ops_test # 测试可执行文件 └── ...打包输出打包成功后会在build/目录生成 .run 文件build_out/cann-{soc}-ops-tensor_{version}_linux-{arch}.run例如cann-950-ops-tensor_9.0.0_linux-x86_64.runcann-950-ops-tensor_9.0.0_linux-aarch64.run说明{soc}会根据指定的 SoC 型号自动替换当前为950。注意事项环境变量要求必须设置ASCEND_HOME_PATH环境变量否则脚本会报错退出。线程数限制如果指定的线程数超过 CPU 核心数脚本会自动调整为 CPU 核心数。测试超时默认测试超时时间为 300 秒可根据实际情况调整。SoC 大小写--soc参数支持大小写不敏感输入脚本会自动标准化为首字母大写、其余小写的格式。算子验证如果使用--ops指定了不存在的算子脚本会列出所有可用的算子并报错退出。相关文档环境部署算子调用算子开发【免费下载链接】ops-tensorops-tensor 是 CANN Compute Architecture for Neural Networks算子库中提供张量类计算的基础算子库采用模块化设计支持灵活的算子开发和管理。项目地址: https://gitcode.com/cann/ops-tensor创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考