mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-05-03 17:37:23 +00:00
70129603aa
* [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.
3 lines
87 B
Python
3 lines
87 B
Python
from .tracer import ColoTracer, meta_trace
|
|
from .graph_module import ColoGraphModule
|