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表示生成的子图知识库名称，以实际生成的文件名为准。

使用约束

无

支持的型号

Atlas 200/300/500 推理产品

Atlas 训练系列产品

Atlas 推理系列产品

Atlas A2 训练系列产品/Atlas 800I A2 推理产品

Atlas 200I/500 A2 推理产品

父主题： AOE调优