ColossalAI/colossalai/fx/__init__.py

3 lines
75 B
Python

from .tracer import ColoTracer
from .graph_module import ColoGraphModule