megengine.jit#
>>> from megengine.jit import trace
Wraps a callable and provide: |
|
Trace Config#
Configuration for sublinear memory optimization. |
|
Configuration for DTR memory optimization. |
|
Configuration for graph optimization: False for OFF, True for ON. |