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

2 lines
32 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