获取GPU/NPU的映射表

说明

本节涉及的.json文件、目录等名称均为举例，请根据实际环境替换。其中，--out指定的结果存放路径，需确保操作用户具有读写权限。

操作步骤

  1. 以HwHiAiUser用户登录开发环境。
  2. 生成json文件。

    Ascend-cann-toolkit安装目录/ascend-toolkit/latest/bin/atc --mode=1 --om=/home/HwHiAiUser/data/resnet50.om --json=/home/HwHiAiUser/data/resnet50.json

  3. 进入/home/HwHiAiUser/Ascend/ascend-toolkit/latest/tools/operator_cmp/compare目录。
  4. 执行获取GPU/NPU的映射表命令。

    由于dump和npy比对数据文件是由多个文件组成，故下文操作步骤中-m和-g参数须指定数据文件所在的父目录。如：/home/HwHiAiUser/MyApp_mind/resnet50，其中resnet50文件夹下直接保存比对数据文件。

    目录结构示例如下：

    root@xxx:/home/HwHiAiUser/MyApp_mind/resnet50# tree
.
├── BatchMatMul.bert_encoder_layer_0_attention_self_MatMul_1.24.1614717261785536
├── BatchMatMul.bert_encoder_layer_0_attention_self_MatMul.21.1614717261768864
├── BatchMatMul.bert_encoder_layer_10_attention_self_MatMul_1.235.1614717263664916
#仅为示例，此处省略剩余文件名。

    python3 msaccucmp.py compare -m /home/HwHiAiUser/MyApp_mind/resnet50 -g /home/HwHiAiUser/Standard_caffe/resnet50 -f /home/HwHiAiUser/data/resnet50.json -out /home/HwHiAiUser/result -map

    输出结果为mapping_*.csv文件内容如图1所示。

    图1 GPU/NPU的映射表
    表1 输出参数说明

    参数

    说明

    Index

    算子的ID。

    OpType

    算子类型。指定-f，-cf或-q参数时获取算子类型。

    NPUDump

    表示基于昇腾AI处理器运行生成的dump数据的算子名。

    GroundTruth

    表示基于GPU/CPU运行生成的npy或dump数据的算子名。

    TensorIndex

    表示基于昇腾AI处理器运行生成的dump数据的算子的input ID和output ID。

    NPUDumpPath

    表示基于昇腾AI处理器运行生成的dump文件路径。

    GroundTruthPath

    表示基于GPU/CPU运行生成的npy或dump文件路径。

父主题： 扩展功能