Dump指定算子

可通过配置config.json文件指定需要dump的算子:

  • “dump_mode”设置为1
  • “kernels”设置目标算子,有三种设置方式:算子名称匹配,算子类型的匹配,算子名称的正则表达式匹配。

以算子"Default/Conv-op12"为例:

  1. 算子名称匹配
    “kernels”: [“Default/Conv-op12”]
    注:可以在Dump保存的ms_output_trace_code_graph_{graph_id}.ir文件中获取算子名称。

  2. 算子类型的匹配
    “kernels”: [“conv”]
    注:当字符串中不带算子scope信息和算子id信息时,后台则认为其为算子类型,不区分大小写。

  3. 算子名称的正则表达式匹配

    匹配开头:”kernels”: [“name-regax(Default.*)”]
    匹配结尾:”kernels”: [“name-regax(.*op12)”]
    匹配两头:”kernels”: [“name-regax(Default.*op12)”]
    匹配中间一处:”kernels”: [“name-regax(.Conv. )”]
    匹配中间任意两处:”kernels”: [“name-regax(.*Conv.op. )”]