github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-02 16:43:24 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
cde7b8a5b8e0803c8342e7ecfb3d5ad2de237e86
ColossalAI/colossalai/fx/__init__.py
Frank Lee 6d86f1bc91 [fx] supported data-dependent control flow in model tracing (#1185) 
* [fx] supported data-dependent control flow in model tracing

* polish code
2022-06-29 15:05:25 +08:00

2 lines
32 B
Python
Raw Blame History

from .tracer import ColoTracer
Reference in New Issue View Git Blame Copy Permalink