可通过配置config.json文件指定需要dump的算子:
- “dump_mode”设置为1
- “kernels”设置目标算子,有三种设置方式:算子名称匹配,算子类型的匹配,算子名称的正则表达式匹配。
以算子"Default/Conv-op12"为例:
-
算子名称匹配
“kernels”: [“Default/Conv-op12”]
注:可以在Dump保存的ms_output_trace_code_graph_{graph_id}.ir文件中获取算子名称。 -
算子类型的匹配
“kernels”: [“conv”]
注:当字符串中不带算子scope信息和算子id信息时,后台则认为其为算子类型,不区分大小写。 -
算子名称的正则表达式匹配
匹配开头:”kernels”: [“name-regax(Default.*)”]
匹配结尾:”kernels”: [“name-regax(.*op12)”]
匹配两头:”kernels”: [“name-regax(Default.*op12)”]
匹配中间一处:”kernels”: [“name-regax(.Conv. )”]
匹配中间任意两处:”kernels”: [“name-regax(.*Conv.op. )”]
