mindspore.device_context.ascend.op_tuning.aoe_tune_mode

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

AOE tuning mode setting, which is not set by default.

Parameters

tune_mode (str) –

The AOE tuning mode to set.

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

  • "offline": The GE graph will be saved for offline tuning.

Examples

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