作者:慕容梦
转载地址: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