megengine.jit

>>> from megengine.jit import trace

trace

包装一个可调用对象,并提供以下功能:

Trace Config

SublinearMemoryConfig

亚线性内存优化的配置。

DTRConfig

DTR 内存优化的配置。

GraphOptimizationConfig

图优化的配置:False 关闭,True 开启。