mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-07-13 12:05:48 +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
69 B
Python
3 lines
69 B
Python
from .tracer import ColoTracer
|
|
from ._meta_trace import meta_trace
|