Files
ColossalAI/colossalai/fx/__init__.py
Super Daniel 441d584e4a [fx] add a symbolic_trace api. (#1812)
* [fx] add a symbolic_trace api.

* [fx] fix import errors.
2022-11-08 13:59:20 +08:00

5 lines
201 B
Python

from ._compatibility import compatibility, is_compatible_with_meta
from .graph_module import ColoGraphModule
from .passes import MetaInfoProp
from .tracer import ColoTracer, meta_trace, symbolic_trace