昇腾社区首页
中文
注册

操作指南

  1. 进入算子和模型速查工具所在路径。
    cd $HOME/Ascend/ascend-toolkit/latest/tools/ms_fast_query
  2. 查询算子和模型。命令中的参数说明请参见表1
    • 执行如下命令查询算子:
      python ms_fast_query.py -t op --opp_path /home/xxx/Ascend/ascend-toolkit/latest/opp -o /home/opp/xxx.json
    • 执行如下命令查询模型:
      python ms_fast_query.py -t model -o /home/model/xxx.json
      表1 参数说明

      参数

      说明

      -t

      --type

      查询类型,必选。
      • op:查询算子
      • model:查询模型

      --opp_path

      CANN toolkit中的算子包的安装路径,查询算子时必选。

      /home/xxx/Ascend/ascend-toolkit/latest/opp

      -o

      --output

      存放查询后的结果文件(json格式)路径,必选。

      • op.json:算子信息查询结果文件,路径保存示例为/home/opp/xxx/op.json。
      • model.json:模型信息查询结果文件,路径保存示例为/home/model/xxx/model.json。

      -h

      --help

      显示帮助信息,可选。

  3. 查看结果文件中result对应的字段,如果为“success”代表查询成功,如果为“fail”代表查询失败,回显如下所示。
    • 成功查询算子:
      1
      {"result": "success", "ops": xxx}
      
    • 成功查询模型:
      1
      {"result": "success", "count": xxx, "models": xxx}
      

      xxx表示查询的算子或者模型的具体信息,以查询结果为准。

    • 查询失败:
      1
      {"result": "fail"}