CANN/ge：Python Pass路径配置

ASCEND_GE_PY_PASS_PATH【免费下载链接】geGEGraph Engine是面向昇腾的图编译器和执行器提供了计算图优化、多流并行、内存复用和模型下沉等技术手段加速模型执行效率减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge功能描述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/扫描规则目录扫描规则当路径指向目录时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），仅供参考