--mdl_bank_path
Applicability
Product |
Supported (Yes/No) |
|---|---|
Atlas 350 Accelerator Card |
x |
√ |
|
√ |
|
√ |
|
√ |
|
√ |
Description
Atlas 350 Accelerator Card does not support subgraph tuning.
Sets the path of the custom repository generated after subgraph tuning.
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.
- 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.
- 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.
- 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.