dump指定算子的方法分享

作者：慕容梦

转载地址：dump指定算子的方法分享_MindSpore_昇腾论坛_华为云论坛

1. 同步dump， 指定算子dump， 分享

mindspore有个功能叫dump数据， 分同步dump和异步dump, 其中同步dump可以dump所有的数据，也能dump指定的数据

在示例中目前只看到dump所有的数据，所以来分享一下dump指定算子的方法

2. 执行代码：

异步dump需要配置环境变量MINDSPORE_DUMP_CONFIG， 且值为指定的json配置文件，我直接写在代码里面，也可以通过设置环境变量来控制

export MINDSPORE_DUMP_CONFIG=***/e2e_dump.json

在这个代码中还写了对比函数，来对比最后的正确性

3. json文件配置

e2e_dump.json

其中"dump_mode": 1 中设置1 是指定算子dump

kernels可以等执行一遍dump后去查看path路径下面的rank_0/graphs/ms_output_trace_code_graph_0.ir， 里面有所有算子的kernels, 选择你想要的，放进json即可

我指定的是第一个add算子

4.执行结果

查看生成的数据

数据在/tmp/dumpdata/rank_0/net033/0/0/目录下面

邮箱：497776945@qq.com