AOE

ge.mdl_bank_path

For Ascend 950PR/Ascend 950DT, this parameter is not supported.

Sets the directory of the custom repository generated after subgraph tuning.

This parameter must be used together with ge.bufferOptimize and takes effect only when buffer optimization is enabled to improve performance by temporarily storing data in the buffer.

Parameter values: directory of the custom repository generated after model tuning.

Format: Only letters, digits, underscores (_), hyphens (-), and periods (.) are allowed.

Default value: $HOME/Ascend/latest/data/aoe/custom/graph/<soc_version>

Restrictions:

Priority ranked from high to low: directory specified by ge.mdl_bank_path > directory specified by TUNE_BANK_PATH > default directory.

  1. The custom repository directory specified by ge.mdl_bank_path takes effect and the directory specified by TUNE_BANK_PATH does not when TUNE_BANK_PATH is used to specify the directory before model compilation, and then ge.mdl_bank_path is used to specify the directory during model build.
  2. The default directory takes effect if both the directories specified by ge.mdl_bank_path and the environment variable are invalid or contain no custom repository.
  3. If no custom repository is available in the preceding directories, the built-in repository for subgraph tuning is searched in the ${INSTALL_DIR}/<arch>-linux/data/fusion_strategy/built-in path. Replace <arch> with the actual OS architecture.

Mandatory/Optional: optional

Effective level: global, session, and graph

ge.op_bank_path

For Ascend 950PR/Ascend 950DT, this parameter is not supported.

Sets the directory of the custom repository generated after operator tuning.

Format: Only letters, digits, underscores (_), hyphens (-), and periods (.) are allowed.

Default: ${HOME}/Ascend/latest/data/aoe/custom/op

Restrictions:

Path (path of the custom repository generated after operator tuning) priority ranked from high to low: path specified by the TUNE_BANK_PATH environment variable > path specified by ge.op_bank_path > default path of the custom repository generated after operator tuning.

  1. If the TUNE_BANK_PATH environment variable is used to specify the custom repository path before model conversion and ge.op_bank_path is used to specify the custom repository path during model build, then the path specified by the TUNE_BANK_PATH environment variable takes effect and the path specified by ge.op_bank_path does not take effect.
  2. The default directory takes effect if both the directories specified by ge.op_bank_path and the environment variable are invalid.
  3. If none of the preceding directories contains the custom repository, the system searches the built-in directory of the custom repository generated after operator tuning.

Mandatory/Optional: optional

Effective level: global, session, and graph