GE-PY Python 模块文档【免费下载链接】geGEGraph Engine是面向昇腾的图编译器和执行器提供了计算图优化、多流并行、内存复用和模型下沉等技术手段加速模型执行效率减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge概述GE-PY 是 GraphEngine 的 Python 接口模块提供了 Pythonic 的图相关接口。为用户提供了便捷的图构建和操作编译执行等功能。该模块对外头文件位于api/python/ge/ge/目录下。GE-PY 模块包含以下核心组件graph 模块- 图基础操作模块提供 Graph、Node、Tensor、TensorDesc 等核心类passes 模块- 自定义 Fusion Pass 开发模块提供 Python 级别的图融合 Pass 开发能力ge_global 模块- GE 全局初始化和析构接口error 模块- GE Python API 失败时抛出的错误类型携带 GE 内部错误信息和接口上下文offline_compile 模块- 离线图编译接口session 模块- 图编译执行接口allocator 模块- 内存分配器抽象供异步执行场景注册外置 allocatorutils 模块- GE 公共工具接口提供 Shape 推导、节点 AICore 支持性校验等能力es 模块- Eager-Style 图构建接口提供函数式风格的图构建方式pyatc 模块-atc命令行等价入口便于指定 ATC 进程内的 python 解释器文档导航设计文档GE-PY 模块类关系文档- Graph、Node、Tensor、TensorDesc、Session 等基础模块的详细说明offline_compile 模块离线图编译接口Graph 类图操作的主要接口Node 类图节点操作接口Tensor 类张量数据类Shape/TensorDesc 类张量形状和元信息描述GeApi 类GE 初始化和析构GeError 类GE Python API 错误类型Session 类图编译执行接口Allocator 类异步执行场景下的外置内存分配器接口GeUtils 类Shape 推导与节点 AICore 支持性校验工具接口ES-PY 模块文档- Eager-Style 图构建模块的详细说明GraphBuilder 类Eager-Style 图构建器TensorHolder 类张量持有者Python Pass 设计文档- Python 自定义 Fusion Pass 开发能力的详细设计说明FusionBasePass 类基础融合 Pass 基类PatternFusionPass 类模式匹配融合 Pass 基类DecomposePass 类算子分解 Pass 基类Pass 注册与发现机制Bridge 架构、native helper、多版本产物装载与 runtime fallback codegen 设计pyatc CLI 设计文档- pyatc 命令行的详细设计说明API 参考API 参考文档- 各模块的接口参考文档Graph、Node、Tensor、TensorDesc、 Shape、DataTypeSession、Allocator、GeApi、ErrorGraphBuilder、TensorHolderOfflineCompile、GeUtilsPasses、pyatc环境变量ASCEND_GE_PY_PASS_PATH- Python Pass 插件路径发现环境变量模块关系graph 模块- 提供图的基础操作能力是其他模块的基础passes 模块- 提供自定义 Fusion Pass 开发能力通过装饰器注册 Pass在编译阶段由 GE 自动发现并执行es 模块- 提供函数式图构建方式最终构建出 graph 模块的 Graph 对象allocator 模块- 为 session 异步执行提供按 stream 维度注册的外置内存分配能力utils 模块- 面向 graph 模块对象提供公共工具能力供 Python pass 等场景对 replacement graph 执行 Shape 推导和节点支持性校验session 模块- 使用 graph 模块构建的图进行编译和执行编译过程中会加载并执行 passes 模块注册的 Passge_global 模块- 提供全局初始化和资源管理error 模块- 提供 GE Python API 统一错误类型失败异常中包含 GE 内部错误信息和接口上下文offline_compile 模块- 提供离线模型构建、导出能力pyatc 模块- 提供与atc等价的命令行入口便于指定 ATC 进程内的 python 解释器使用示例基础图操作示例参考 使用es的python api构图sample的方式执行用例 特别需要说明的是: 需要先安装包并设置对应的环境变量离线图编译执行示例参考 使用offline_compile的python api离线图编译执行sample 的方式执行用例特别需要说明的是 需要先安装包并设置对应的环境变量更多示例更多 Python 用例请参考 examples/es 目录下的各个子目录开发路线图我们在2025年首次推出了ge-python的模块目标是提供 Python 语言的构图、编译图、执行图的能力。 2026Q1 我们主要工作是重点完成 es api 集成让用户安装好 ops 包后使用 Python 的 es api 构图能力核心架构[December 2025]ge-python模块已经完成设计和落地具备了基本的使用es api 构图、 编译图、 执行图的能力。基础API 集成[December 2025]基础接口已经完成设计和落地。[February 2026] es 的 python 算子 api 支持详见es api集成路标。[April 2026] 图异步执行的python接口提供[April 2026] 离线图编译执行的python接口提供[April 2026] pyatc接口提供自定义pass[April 2026] 开发态主链已完成FusionBasePassPatternFusionPassDecomposePass[May 2026] 预制版本、多版本 native artifact 补齐[June 2026] fallback codegen 能力补齐sample和相关文档[December 2025]已提供对应的sample涵盖常见使用场景。[December 2025]已提供细化的文档即本目录。后向兼容[December 2025]Python api 后向兼容完成设计并落地。others[] [后续阶段] 自定义算子入图python化支持【免费下载链接】geGEGraph Engine是面向昇腾的图编译器和执行器提供了计算图优化、多流并行、内存复用和模型下沉等技术手段加速模型执行效率减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考