算子调用【免费下载链接】ops-tensorops-tensor 是 CANN Compute Architecture for Neural Networks算子库中提供张量类计算的基础算子库采用模块化设计支持灵活的算子开发和管理。项目地址: https://gitcode.com/cann/ops-tensor前提条件环境部署调用项目算子之前请先参考环境部署完成基础环境搭建。调用算子列表项目可调用的算子参见算子列表。编译执行基于社区版CANN包对算子源码修改时可采用如下方式进行源码编译ops-tensor包选择整个项目编译生成的包称为ops-tensor包可完整替换CANN包对应部分。ops-tensor包编译ops-tensor包进入项目根目录执行如下编译命令# 编译所有算子并生成安装包 bash build.sh --pkg若提示如下信息说明编译成功。Self-extractable archive cann-950-ops-tensor_9.0.0_linux-*.run successfully created. Build package success: build_out/cann-950-ops-tensor_9.0.0_linux-*.run编译成功后run包存放于项目根目录的build目录下。安装ops-tensor包# 安装命令 ./build_out/cann-*-ops-tensor-*linux*.run --fullops-tensor安装在${ASCEND_HOME_PATH}/cann路径中${ASCEND_HOME_PATH}表示CANN软件安装目录。配置环境变量source ${ASCEND_HOME_PATH}/cann/set_env.bash可选卸载ops-tensor包# 卸载命令 ./${install_path}/cann/share/info/ops_tensor/scripts/uninstall.sh本地验证通过项目根目录build.sh脚本可快速调用算子和UT用例验证项目功能是否正常build参数介绍参见build参数说明。运行测试# 编译并运行测试 bash build.sh --run # 编译指定算子并运行测试 bash build.sh --opsadd --run执行测试后会打印执行结果以add算子为例结果如下all_ops_test .......... Passed * sec编译选项说明选项说明示例--opsNAME编译指定算子--opsadd--build-typeTYPE构建类型Release/Debug--build-typeDebug--run编译后运行测试--run--pkg生成安装包--pkg-j[N]并行编译线程数-j8-v详细输出-v常用命令示例# 基本编译 bash build.sh # 编译并运行测试 bash build.sh --run # 编译指定算子并运行测试 bash build.sh --opsadd --run # 编译生成安装包 bash build.sh --pkg # 多线程编译 bash build.sh -j16 # 调试模式编译 bash build.sh --build-typeDebug --run # 详细输出 bash build.sh -v --run更多帮助CANN 开发文档Ascend C API 参考【免费下载链接】ops-tensorops-tensor 是 CANN Compute Architecture for Neural Networks算子库中提供张量类计算的基础算子库采用模块化设计支持灵活的算子开发和管理。项目地址: https://gitcode.com/cann/ops-tensor创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考