Direct Async 测试算子 nda-offline-operator
- 基于AscendC 算子语法实现 NPU Write / Read 带宽与试验测试
- 算子能力持续优化中
Direct Async 测试算子编译指导
编译依赖
- 编译前请确认环境已安装NPU driver包和CANN包
- 编译前请确认环境是否已安装以下Python包,若环境无以下python包,请尝试安装:
pip3 install numpy sympy scipy attrs psutil decorator
编译步骤
- 修改 CMakePresets.json
"ASCEND_CANN_PACKAGE_PATH": {
"type": "PATH",
"value": "/usr/local/Ascend/cann"
},
value字段换成环境上CANN包的路径
- 执行以下命令,修改shell脚本文件的格式
find . -type f -name "*.sh" -exec dos2unix {} \;
- 添加CANN的环境变量(CANN包安装路径可根据/etc/Ascend/ascend_cann_install.info文件查看)
source /usr/local/Ascend/ascend-toolkit/set_env.sh
- 为shell脚本添加可执行权限
find . -name "*.sh" -exec chmod +x {} \;
- 执行build.sh 编译
- 会在当前路径下的build_out目录中生成custom_opp_*_*.run包,执行run包,并通过 “--install-path” 参数添加安装路径,例如:
./build_out/custom_opp_*_*.run --install-path=/home/test
- 算子包安装完成后,会提示导入算子包的环境变量:
source /home/test/vendors/customize/bin/set_env.bash