mindspore.device_context.ascend.op_tuning.aoe_job_type

View Source On Gitee
mindspore.device_context.ascend.op_tuning.aoe_job_type(config)[source]

Set the parameters specific to Ascend Optimization Engine.It needs to be used in conjunction with mindspore.device_context.op_tuning.aoe_tune_mode(tune_mode). The framework set to "2" by default.

Parameters

config (str) –

Choose the tuning type.

  • "1": Set to subgraph tuning.

  • "2": Set to operator tuning.

Examples

>>> import mindspore as ms
>>> ms.device_context.ascend.op_tuning.aoe_job_type("1")