下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

图dump功能

功能简介

用于配置是否导出图结构信息以及导出的格式,方便后续进行问题定位。

使用方法

该功能通过torchair.get_npu_backend中compiler_config参数配置,配置示例如下,参数说明参见表1

1
2
3
4
5
import torch_npu, torchair
config = torchair.CompilerConfig()
# 设置导出图结构文件格式
config.debug.graph_dump.type = "pbtxt"
npu_backend = torchair.get_npu_backend(compiler_config=config)
表1 参数说明

参数名

参数说明

是否必选

graph_dump.type

开启图结构信息dump,同时通过graph_dump.type指定导出文件格式。当前仅支持如下格式:

  • txt:是CANN最终接收到的TorchAir构图结果,为protobuf格式,可通过Vscode等工具查看。
  • pbtxt:是可以被TensorBoard读取的构图结果,可通过TensorBoard等工具查看。
  • py:是torch代码经由converter转化后的GE构图信息,可通过Vscode等工具查看。

graph_dump.type的缺省值为“None”,不导出图结构信息。

说明:
  • 多次定义导出格式时,以最后一次定义的格式为准。
  • 不支持txt、pbtxt、py三种格式同时导出。

产物说明

指定图dump导出格式后,生成的图结构文件缺省路径为当前脚本所在路径。当模型执行异常(如模型精度不准确)时,可通过该文件进行问题分析和排查。

  • dynamo_original_graph_${timestamp}.${graph_dump_type}:模型原始图结构文件。
  • dynamo_optimized_graph_${timestamp}.${graph_dump_type}:优化后的模型图结构文件。

其中${timestamp}表示时间戳,${graph_dump_type}表示导图文件格式。

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词