mindspore.device_context.ascend.op_tuning.aoe_tune_mode

View Source On Gitee
mindspore.device_context.ascend.op_tuning.aoe_tune_mode(tune_mode)[source]

AOE tuning mode setting, which is not set by default. For detailed information, please refer to Ascend Optimization Enging .

Parameters

tune_mode (str) –

AOE tuning mode setting.

  • "online": the online tuning function is turned on.

  • "offline": ge graph will be saved for offline tuning.

Examples

>>> import mindspore as ms
>>> ms.device_context.ascend.op_tuning.aoe_tune_mode("online")