mirror of https://github.com/hpcaitech/ColossalAI
![]() * [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. |
||
---|---|---|
.. | ||
meta_patch | ||
__init__.py | ||
_meta_trace.py | ||
_tracer_utils.py | ||
tracer.py |