github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/fx/tracer/__init__.py

3 lines
69 B
Python
Raw Normal View History

[fx] supported data-dependent control flow in model tracing (#1185) * [fx] supported data-dependent control flow in model tracing * polish code
2022-06-29 07:05:25 +00:00
from .tracer import ColoTracer
[fx] support meta tracing for aten level computation graphs like functorch. (#1536) * [fx] support meta tracing for aten level computation graphs like functorch. * [fx] support meta tracing for aten level computation graphs like functorch. * [fx] remove redundant import. * [fx] add docstring.
2022-09-05 04:10:09 +00:00
from ._meta_trace import meta_trace