ColossalAI/colossalai/fx/tracer/__init__.py

5 lines
157 B
Python

from colossalai.fx.tracer.meta_patch.patched_function.python_ops import operator_getitem
from ._meta_trace import meta_trace
from .tracer import ColoTracer