mindspore.JitConfig

class mindspore.JitConfig(jit_level='O1', task_sink=True, **kwargs)[source]

Jit config for compile.

Note

This is an experimental function that is subject to change or deletion.

Parameters

  • jit_level (str) –

    Option for argument level for Optimization of lift graph. Supports [“O0”, “O1”, “O2”]. Default: “O1”.

    • ”O0”: Basic optimization.

    • ”O1”: Manual optimization.

    • ”O2”: Manual optimization and graph computation fusion.

  • task_sink (bool) – Determines whether to pass the data through dataset channel. Default: True.

  • **kwargs (dict) – A dictionary of keyword arguments that the class needs.

Examples

>>> from mindspore import JitConfig
>>>
>>> jitconfig = JitConfig(jit_level="O1")
>>> net = LeNet5()
>>>
>>> net.set_jit_config(jitconfig)