--op_tune_kernel_path

Applicability

Description

After static kernel tuning is complete, AOE generates static kernels. This option specifies the path for storing the static kernels.

Currently, this function supports the following models:

Atlas A2 training products/Atlas A2 inference products

Atlas A3 training products/Atlas A3 inference products

See Also

Use this option in conjunction with --op_tune_mode (= static_kernel or fast) and --op_tune_file.

Arguments

If this option is not set, the kernels are stored in the /aoe_workspace/op_tune_kernel folder in the current working directory where the tuning command is executed by default.

Examples

--op_tune_file=/home/lxl/json --op_tune_mode=static_kernel --precision_mode=must_keep_origin_dtype --op_tune_kernel_path=/home/lxl/kernel