ColossalAI/colossalai/fx/tracer/__init__.py

3 lines
69 B
Python

from .tracer import ColoTracer
from ._meta_trace import meta_trace