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. |
||
---|---|---|
.. | ||
profiler_function | ||
profiler_module | ||
__init__.py | ||
meta_tensor.py | ||
profiler.py | ||
registry.py |