TUNE_BANK_PATH
功能描述
可通过此环境变量指定调优后自定义知识库的存储路径。
设置的存储路径必须为绝对路径或相对于执行AOE调优引擎所在路径的相对路径,配置的路径需要为已存在的目录,且执行用户具有读、写、可执行权限。若配置的TUNE_BANK_PATH路径不存在或用户无权限,则调优进程会报错并退出。
- 子图自定义知识库
- 若不配置此环境变量,请使用env命令查询ASCEND_CACHE_PATH是否存在,若存在,自定义知识库存储在:${ASCEND_CACHE_PATH}/aoe_data;若不存在,自定义知识库默认存储在:${HOME}/Ascend/latest/data/aoe/custom/graph/${soc_version}。
- 若配置了此环境变量,自定义知识库存储在该环境变量路径下。
- 算子自定义知识库
- 若不配置此环境变量,请使用env命令查询ASCEND_CACHE_PATH是否存在,若存在,自定义知识库存储在:${ASCEND_CACHE_PATH}/aoe_data/${soc_version};若不存在,自定义知识库默认存储在:${HOME}/Ascend/latest/data/aoe/custom/op/${soc_version}。
- 若配置此环境变量,则调优后的最优策略存储在配置路径的${soc_version}。

在多用户共享知识库场景下,共享知识库的用户需要设置TUNE_BANK_PATH为同一路径,并且对配置的路径具有读、写权限。
若调优时自定义了知识库路径,后续进行模型转换时,若想直接使用自定义知识库,也需要配置此环境变量。
配置示例
export TUNE_BANK_PATH=/home/HwHiAiUser/custom_tune_bank
调优前设置该环境变量,作用是指定自定义的知识库存储路径,设置后,会在TUNE_BANK_PATH指定的目录下,生成如下结构的目录和文件。
调优后设置该环境变量,作用是使用已有的自定义知识库,请确保TUNE_BANK_PATH指定的目录结构如下。
├──Ascendxxx //表示soc_version,以实际生成的名字为准。该文件夹下存放的是算子知识库。 │ ├── cube │ │ ├── op1.bin //op1,op2,op3表示生成的算子知识库名称,以实际生成的文件名为准。 │ │ ├── op1.json │ │ ├── op2.bin │ │ └── op2.json │ └── unified_bank │ └── op3.json └── subgraph.json //subgraph表示生成的子图知识库名称,以实际生成的文件名为准。
使用约束
无
支持的型号
父主题: AOE调优