ColossalAI/colossalai/_analyzer/fx/__init__.py

4 lines
129 B
Python

from .node_util import MetaInfo
from .symbolic_profile import symbolic_profile
from .tracer.symbolic_trace import symbolic_trace