--mdl_bank_path

功能说明

加载子图调优后自定义知识库的路径。

关联参数

无。

参数取值

参数值：子图调优后自定义知识库路径。

参数值格式：支持大小写字母（a-z，A-Z）、数字（0-9）、下划线（_）、中划线（-）、句点（.）。

参数默认值：${HOME}/ascend/latest/data/aoe/custom/graph/<soc_version>

推荐配置及收益

无。

示例

例如子图调优后自定义知识库的路径为$HOME/custom_module_bank，则使用示例为：

--mdl_bank_path=$HOME/custom_module_path

依赖约束

加载子图调优后自定义知识库路径优先级：--mdl_bank_path参数加载路径>TUNE_BANK_PATH环境变量设置路径>默认子图调优后自定义知识库路径。

1. 如果模型转换前，通过TUNE_BANK_PATH环境变量指定了子图调优自定义知识库路径，模型转换时又通过--mdl_bank_path参数加载了自定义知识库路径，该场景下以--mdl_bank_path参数加载的路径为准，TUNE_BANK_PATH环境变量设置的路径不生效。
2. --mdl_bank_path参数和环境变量指定路径都不生效或无可用自定义知识库，则使用默认自定义知识库路径。
3. 如果默认自定义知识库路径下也无可用知识库，则atc工具会查找${INSTALL_DIR}/compiler/data/fusion_strategy/custom路径下是否有可用知识库，其中，${INSTALL_DIR}请替换为CANN软件安装后文件存储路径。例如，若安装的Ascend-cann-toolkit软件包，则安装后文件存储路径为：$HOME/Ascend/ascend-toolkit/latest。
4. 如果上述路径下都无可用的自定义知识库，则atc工具会查找子图调优内置知识库，该路径为：${INSTALL_DIR}/compiler/data/fusion_strategy/built-in

支持的芯片型号

Atlas 200/300/500 推理产品

Atlas 训练系列产品