ASCEND_GE_PY_PASS_PATH【免费下载链接】geGEGraph Engine是面向昇腾的图编译器和执行器提供了计算图优化、多流并行、内存复用和模型下沉等技术手段加速模型执行效率减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge产品支持情况产品是否支持Atlas A3 训练系列产品/Atlas A3 推理系列产品√Atlas A2 训练系列产品/Atlas A2 推理系列产品√功能说明ASCEND_GE_PY_PASS_PATH是 GEGraph EnginePython Pass 系统的路径发现环境变量用于告诉 GE 引擎去哪些路径加载用户编写的 Python 融合 Pass 插件。该环境变量控制的是一个完整的发现-注册-执行链路GE 编译阶段启动 Pass 加载流程时C 侧检查ASCEND_GE_PY_PASS_PATH是否已设置且非空。若已设置GE 通过 C/Python bridge 将该环境变量同步到 Python 侧。Python 侧按路径扫描并加载用户编写的 Pass 模块。模块加载过程中用户通过register_fusion_pass或register_decompose_pass等装饰器将 Pass 注册到 Python 注册表。注册信息回传到 C 侧的 PassRegistry在后续编译阶段由 GE Pass 调度框架统一执行。当前阶段ASCEND_GE_PY_PASS_PATH是 Python Pass 发现的主路径而非兜底机制后续版本将补充entry_points(groupge.passes.plugins)自动发现机制。取值格式ASCEND_GE_PY_PASS_PATHpath1[:path2[:...]]多个路径之间以冒号:分隔。每个路径可以是.py文件或目录。路径建议使用绝对路径。路径类型说明路径类型扫描行为单个.py文件直接作为 Python 模块加载目录遍历其中不以_开头的.py文件作为模块加载包含__init__.py的子目录作为 Python 包导入设置环境变量# 指向单个 Python 文件 export ASCEND_GE_PY_PASS_PATH/path/to/my_pass.py # 指向目录目录下所有 .py 文件和 Python 包都会被扫描 export ASCEND_GE_PY_PASS_PATH/path/to/pass_dir/ # 支持多个路径以冒号分隔 export ASCEND_GE_PY_PASS_PATH/path/to/pass1.py:/path/to/pass_dir2/使用示例Python Pass 开发步骤请参考 Python 融合 Pass 开发指南。 完整样例请参考 examples/fusion_pass 目录涵盖FusionBasePass、PatternFusionPass、DecomposePass等 Pass 的开发示例。扫描规则详解目录扫描规则当路径指向目录时bootstrap模块按以下规则扫描目录下以_开头的文件和子目录将被跳过。非下划线开头的.py文件作为独立模块加载。包含__init__.py的子目录作为 Python 包通过importlib.import_module()导入。目录下的子目录按名称排序后依次扫描。模块命名每个通过文件路径加载的模块会被赋予一个内部名称格式为_ge_py_pass_{stem}_{hash}不会与用户模块名称冲突。同一个文件路径不会被重复加载。约束说明必须在 GE 初始化之前设置该环境变量。GE 在编译阶段首次加载 Pass 时读取该变量运行过程中动态修改环境变量不会生效。路径指向的.py文件必须包含合法的 Python 代码且应使用register_fusion_pass或register_decompose_pass装饰器注册至少一个 Pass。未注册任何 Pass 的模块会被加载但不会产生效果。路径指向的目录必须存在否则会在 Pass 加载阶段抛出FileNotFoundError。文件路径必须以.py为后缀否则会抛出ValueError。环境变量值为空或不设置时GE 将跳过 Python Pass 加载流程不影响 C 自定义 Pass 的正常加载。同一个文件路径只会被加载一次重复出现在多个路径段中不会导致重复注册。【免费下载链接】geGEGraph Engine是面向昇腾的图编译器和执行器提供了计算图优化、多流并行、内存复用和模型下沉等技术手段加速模型执行效率减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考