文档
注册

执行固定Shape算子

接口调用流程,请参见接口调用流程

示例代码

执行固定Shape算子(此处以Add算子为例)分为以下几步:

  1. 准备Add算子的模型文件。
    1. 先构造该算子的描述文件(*.json文件,描述输入输出Tensor描述、算子属性等)。
      Add算子的描述文件示例如下:
      [
        {
          "op": "Add",
          "input_desc": [
            {
              "format": "ND",
              "shape": [8, 16],
              "type": "int32"
            },
            {
              "format": "ND",
              "shape": [8, 16],
              "type": "int32"
            }
          ],
          "output_desc": [
            {
              "format": "ND",
              "shape": [8, 16],
              "type": "int32"
            }
          ]
        }
      ]
    2. 借助ATC工具,将该算子描述文件编译成单算子模型文件(*.om文件),再分别调用AscendCL接口加载om模型文件、执行算子。

      ATC工具的命令示例如下:

      atc --singleop=$HOME/singleop/add.json --output=$HOME/singleop/out/op_model --soc_version=<soc_version>

      关键参数解释如下(详细参数说明,请参见ATC工具使用指南。):

      • --singleop:单算子描述文件(json格式)的路径。
      • --output:存放单算子模型文件的目录。
      • --soc_version:昇腾AI处理器的版本。

        进入“CANN软件安装目录/compiler/data/platform_config”目录,".ini"文件的文件名即为昇腾AI处理器的版本,请根据实际情况选择。

  2. 编写单算子加载、执行的代码逻辑。

    以下是关键步骤的代码示例,不可以直接拷贝编译运行,仅供参考,调用接口后,需增加异常处理的分支,并记录报错日志、提示日志,此处不一一列举。完整代码,您可以从acl_execute_add样例中查看。

    // 1.AscendCL初始化
    aclRet = aclInit(nullptr);
    
    // 2.运行管理资源申请(使用默认Context、默认Stream,默认Stream在作为其它接口入参时,可传空指针)
    aclRet = aclrtSetDevice(0);
    获取软件栈的运行模式,不同运行模式影响后续的接口调用流程(例如是否进行数据传输等)
    aclrtRunMode runMode;
    bool g_isDevice = false;
    aclError aclRet = aclrtGetRunMode(&runMode);
    g_isDevice = (runMode == ACL_DEVICE);
    
    // 3.加载单算子模型文件(*.om文件)
    // 该目录相对可执行文件所在的目录,例如,编译出来的可执行文件存放在out目录下,此处就表示out/op_models目录
    aclRet = aclopSetModelDir("op_models");
    
    // 4.执行算子
    // opType表示算子类型名称,例如Add
    // numInputs表示算子输入个数,例如Add算子是2个输入
    // inputDesc表示算子输入tensor描述的数组,描述每个输入的format、shape、数据类型
    // inputs表示算子输入tensor数据
    // numOutputs表示算子输出个数,例如Add算子是1个输出
    // outputDesc表示算子输出tensor描述的数组,描述每个输出的format、shape、数据类型
    // outputs表示算子输出tensor数据
    // attr表示算子属性,如果算子没有属性,也需要调用aclopCreateAttr接口创建aclopAttr类型的数据
    // stream用于维护一些异步操作的执行顺序
    
    aclopExecuteV2(opType, numInputs, inputDesc, inputs,                 
                   numOutputs, outputDesc, outputs, attr, nullptr);
    
    // 处理执行算子后的输出数据,例如在屏幕上显示、写入文件等,由用户根据实际情况自行实现
    // ......
    
    // 阻塞应用运行,直到指定Stream中的所有任务都完成
    aclrtSynchronizeStream(nullptr);
    
    // 5. 释放运行管理资源(默认Context、Stream无需用户释放,调用aclrtResetDevice接口后自动释放)
    aclRet = aclrtResetDevice(0);
    
    // 6.AscendCL去初始化
    aclRet = aclFinalize();
    
    // ....

相关资源

通过在线视频课程学习该功能,请参见CANN应用开发初级

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词