megengine.jit

>>> from megengine.jit import trace

trace

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

exclude_from_trace

xla_trace

Wraps a callable, and provides accelerated evaluation compiled by xla.

Trace Config

SublinearMemoryConfig

亚线性内存优化的配置。

DTRConfig

DTR 内存优化的配置。

GraphOptimizationConfig

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