CANN/ge GE 图 Dump 格式说明

GE 图 Dump 格式说明

【免费下载链接】geGE（Graph Engine）是面向昇腾的图编译器和执行器，提供了计算图优化、多流并行、内存复用和模型下沉等技术手段，加速模型执行效率，减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力，并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge

概述

GE（Graph Engine）支持将计算图导出为多种格式，便于开发者查看、调试和分析图结构。本文档介绍三种 dump 格式：ge_proto、onnx和readable，以及它们的特点和使用方法。

Dump 格式概览

Dump 使用方式

通过环境变量自动 Dump

通过设置环境变量，可以在图执行时自动生成 dump 文件：

# 设置图 dump 级别 export DUMP_GE_GRAPH=1 # 设置 dump 路径 export DUMP_GRAPH_PATH=/path/to/dump/directory # 设置 dump 格式 export DUMP_GRAPH_FORMAT="ge_proto|onnx|readable"

环境变量说明：

通过 Graph API 导出

C++

#include "ge/graph.h" // 创建图 ge::Graph graph("my_graph"); // ... 构建图结构 ... // 导出为不同格式 graph.DumpToFile(ge::Graph::DumpFormat::kTxt, "suffix"); // ge_proto graph.DumpToFile(ge::Graph::DumpFormat::kOnnx, "suffix"); // onnx graph.DumpToFile(ge::Graph::DumpFormat::kReadable, "suffix"); // readable

Python

from ge.graph import Graph, DumpFormat # 创建图 graph = Graph("my_graph") # ... 构建图结构 ... # 方式1: 导出为文件 graph.dump_to_file(format=DumpFormat.kTxt, suffix="suffix") # ge_proto graph.dump_to_file(format=DumpFormat.kOnnx, suffix="suffix") # onnx graph.dump_to_file(format=DumpFormat.kReadable, suffix="suffix") # readable # 方式2: 直接打印（仅 readable 格式支持） print(graph) # 直接打印 readable 格式到控制台查看图结构 readable_str = str(graph) # 获取 readable 格式字符串，可用于保存或进一步处理

关于ge.graph的详细说明，请参考 graph模块

附录

Netron 可视化说明

在 Netron 中打开ge_onnx*.pbtxt文件时：

• 节点表示：图中的每个节点表示为一个算子

• 连边关系：连边关系用带箭头的实线表示，箭头方向表示数据流向（从源节点指向目标节点）

• 节点信息查看：点击算子节点可查看算子的详细信息，重点信息包括：

  属性名	说明
  type	算子类型
  name	算子名
  input_desc_dtype:x	第x个输入的数据类型
  input_desc_layout:x	第x个输入的数据格式
  input_desc_shape:x	第x个输入的shape
  output_desc_dtype:x	第x个输出的数据类型
  output_desc_layout:x	第x个输出的数据格式
  output_desc_shape:x	第x个输出的shape

【免费下载链接】geGE（Graph Engine）是面向昇腾的图编译器和执行器，提供了计算图优化、多流并行、内存复用和模型下沉等技术手段，加速模型执行效率，减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力，并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge