AI开发平台MODELARTS-使用Notebook开发Ascend算子:在VS Code中调试Add算子
在VS Code中调试Add算子
- 在Terminal中执行如下命令进入Add算子所在目录。
cd samples/cplusplus/level1_single_api/4_op_dev/6_ascendc_custom_op/kernel_invocation/Add
- 执行如下命令编译和运行脚本。
- CPU模式下执行如下命令。
bash run.sh add_custom ascend910B1 VectorCore cpu
其中,add_custom表示需要运行的算子,ascend910B1表示算子运行的AI处理器型号,VectorCore表示在VectorCore上运行,cpu表示算子以cpu模式运行。
运行结果如下,当前使用md5sum对比了所有输出bin文件,md5值一致表示实际的输出数据和真值数据相符合。
图3 CPU模式运行结果
- NPU模式下执行如下命令
bash run.sh add_custom ascend910B1 VectorCore npu
运行结果如下,当前使用md5sum对比了所有输出bin文件,md5值一致表示实际的输出数据和真值数据相符合。
图4 NPU模式运行结果
- CPU模式下执行如下命令。