mirror of https://github.com/hpcaitech/ColossalAI
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. |
||
---|---|---|
.. | ||
codegen | ||
passes | ||
profiler | ||
tracer | ||
__init__.py | ||
graph_module.py | ||
proxy.py |