--mdl_bank_path

Applicability

Product

Supported (Yes/No)

Atlas 350 Accelerator Card

x

Atlas A3 training product/Atlas A3 inference product

Atlas A2 training product/Atlas A2 inference product

Atlas 200I/500 A2 inference product

Atlas inference product

Atlas training product

Description

Atlas 350 Accelerator Card does not support subgraph tuning.

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

For details about subgraph tuning, see AOE Tuning Tool.

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 no custom repository is available in the preceding directories, ATC searches for the built-in repository in ${INSTALL_DIR}/x86_64-linux/data/fusion_strategy/built-in. Replace x86_64-linux based on the actual OS.