--op_bank_path

Applicable Products

• Ascend 950PR/Ascend 950DT: not supported
• Atlas A3 training products/Atlas A3 inference products: supported
• Atlas A2 training products/Atlas A2 inference products: supported
• Atlas 200I/500 A2 inference products: supported
• Atlas inference products: supported
• Atlas training products: supported

Description

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

For details about operator tuning, see AOE Tuning Tool.

Arguments

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 the preceding directories do not have a custom repository, ATC searches the built-in directory of the custom repository generated after operator tuning.