--op_bank_path

Description

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

For details about operator tuning, see AOE Instructions.

Argument

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

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

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

Suggestions and Benefits

None

Example

Assume that the custom repository generated after operator tuning is stored in $HOME/custom_tune_bank:

--op_bank_path=$HOME/custom_tune_bank

Restrictions

Priority ranked from high to low: the directory specified by TUNE_BANK_PATH > the directory specified by --op_bank_path > the default directory of the custom repository generated after operator tuning.

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

Applicability

Atlas 200/300/500 Inference Product

Atlas Training Series Product