AI开发平台MODELARTS-使用Notebook开发Ascend算子:在VS Code中调试Add算子

时间:2024-04-30 17:04:02

在VS Code中调试Add算子

  1. 在Terminal中执行如下命令进入Add算子所在目录。
    cd samples/cplusplus/level1_single_api/4_op_dev/6_ascendc_custom_op/kernel_invocation/Add
  2. 执行如下命令编译和运行脚本。
    1. CPU模式下执行如下命令。
      bash run.sh add_custom ascend910B1 VectorCore cpu

      其中,add_custom表示需要运行的算子,ascend910B1表示算子运行的AI处理器型号,VectorCore表示在VectorCore上运行,cpu表示算子以cpu模式运行。

      运行结果如下,当前使用md5sum对比了所有输出bin文件,md5值一致表示实际的输出数据和真值数据相符合。

      图3 CPU模式运行结果
    2. NPU模式下执行如下命令
      bash run.sh add_custom ascend910B1 VectorCore npu

      运行结果如下,当前使用md5sum对比了所有输出bin文件,md5值一致表示实际的输出数据和真值数据相符合。

      图4 NPU模式运行结果
support.huaweicloud.com/devtool-modelarts/devtool-modelarts_0142.html