--mdl_bank_path

Description

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

For details about subgraph tuning, see AOE Instructions.

See Also

This option must be used in pair with --buffer_optimize and takes effect only when data buffer optimization is enabled to improve performance by temporarily storing data in the buffer.

Argument

Argument: Directory of the custom repository generated after subgraph tuning.

Format: The directory can contain letters, digits, underscores (_), hyphens (-), and periods (.).

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

Suggestions and Benefits

None

Example

Assume that the custom repository generated after subgraph tuning is stored in $HOME/custom_module_bank:

--mdl_bank_path=$HOME/custom_module_path   --buffer_optimize=l2_optimize

Dependencies and Restrictions

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

  1. If the custom repository path is specified by the TUNE_BANK_PATH environment variable before model conversion and --mdl_bank_path is set during model conversion, the custom repository directory specified by --mdl_bank_path does not take effect.
  2. The default directory takes effect if both the directory specified by --mdl_bank_path and that specified by TUNE_BANK_PATH are invalid or contain no custom repository.
  3. If the preceding directories do not have a custom repository, ATC looks up in the built-in repository of the subgraph tuning in ${INSTALL_DIR}/compiler/data/fusion_strategy/built-in.

Applicability

Atlas 200/300/500 Inference Product

Atlas Training Series Product