--op_bank_path

Applicability

Product	Supported
Atlas A3 training products/Atlas A3 inference products	√
Atlas A2 training products/Atlas A2 inference products	√
Atlas 200I/500 A2 inference products	√
Atlas inference products	√
Atlas training products	√

Description

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

For details about operator tuning, see AOE Tuning Tool User Guide.

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.

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.
The default directory takes effect if both the directory specified by --op_bank_path and that specified by TUNE_BANK_PATH are invalid.
If the preceding directories do not have a custom repository, ATC searches the built-in directory of the custom repository generated after operator tuning.

Parent topic: Operator Tuning Options