ColossalAI/colossalai/fx/profiler
Super Daniel 70129603aa
[fx] support meta tracing for aten level computation graphs like functorch. (#1536)
* [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.
2022-09-05 12:10:09 +08:00
..
profiler_function [fx] add more op patches for profiler and error message for unsupported ops. (#1495) 2022-08-25 23:11:13 +08:00
profiler_module [fx] add more op patches for profiler and error message for unsupported ops. (#1495) 2022-08-25 23:11:13 +08:00
__init__.py [fx] support meta tracing for aten level computation graphs like functorch. (#1536) 2022-09-05 12:10:09 +08:00
meta_tensor.py [fx] hack __torch_dispatch__ for meta tensor and autograd. (#1515) 2022-08-31 16:30:16 +08:00
profiler.py [fx] hack __torch_dispatch__ for meta tensor and autograd. (#1515) 2022-08-31 16:30:16 +08:00
registry.py [fx] add profiler for fx nodes. (#1480) 2022-08-24 16:22:44 +08:00