forked from mindspore-Ecosystem/mindspore
26 lines
865 B
ReStructuredText
26 lines
865 B
ReStructuredText
mindspore.JitConfig
|
|
====================
|
|
|
|
.. py:class:: mindspore.JitConfig(jit_level="O1", exc_mode="auto", **kwargs)
|
|
|
|
编译时所使用的JitConfig配置项。
|
|
|
|
.. note::
|
|
- 这是一个实验性接口，后续可能删除或修改。
|
|
|
|
参数：
|
|
- **jit_level** (str) - 设置编译优化的级别，支持["O0", "O1", "O2", "O3"]。默认值："O1"。
|
|
|
|
- "O0": 基础优化。
|
|
- "O1": 手动优化。
|
|
- "O2": 手动优化与图算优化结合。
|
|
- "O3": 性能优化，无法保证泛化性。
|
|
|
|
- **exc_mode** (str) - 设置执行模式，支持["auto", "sink", "no_sink"]。默认值："auto"。
|
|
|
|
- "auto": 自动策略。
|
|
- "sink": 计算图下沉策略。
|
|
- "no_sink": 非计算图下沉策略。
|
|
|
|
- **kwargs** (dict) - 关键字参数字典。
|