mindspore.device_context.ascend.op_tuning.aoe_job_type

查看源文件
mindspore.device_context.ascend.op_tuning.aoe_job_type(config)[源代码]

设置AOE工具专用的参数,需要和mindspore.device_context.op_tuning.aoe_tune_mode(tune_mode)配合使用,框架默认设置为2。

参数:
  • config (str) - 选择调优类型。

    • "1": 设置为子图调优。

    • "2": 设置为算子调优。

样例:

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