图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) |
参数名 |
参数说明 |
是否必选 |
---|---|---|
graph_dump.type |
开启图结构信息dump,同时通过graph_dump.type指定导出文件格式。当前仅支持如下格式:
graph_dump.type的缺省值为“None”,不导出图结构信息。
说明:
|
否 |
产物说明
指定图dump导出格式后,生成的图结构文件缺省路径为当前脚本所在路径。当模型执行异常(如模型精度不准确)时,可通过该文件进行问题分析和排查。
- dynamo_original_graph_${timestamp}.${graph_dump_type}:模型原始图结构文件。
- dynamo_optimized_graph_${timestamp}.${graph_dump_type}:优化后的模型图结构文件。
其中${timestamp}表示时间戳,${graph_dump_type}表示导图文件格式。
父主题: DFX功能