ColossalAI/colossalai/fx/tracer
Frank Lee 262652c8bc
[autoparallel] added addbmm handler (#1751)
2022-10-21 18:55:48 +08:00
..
meta_patch [autoparallel] added addbmm handler (#1751) 2022-10-21 18:55:48 +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_trace.py [fx/profiler] tuned the calculation of memory estimation (#1619) 2022-09-23 10:59:47 +08:00
_tracer_utils.py [fx] refactor tracer to trace complete graph (#1342) 2022-07-20 11:20:38 +08:00
tracer.py [doc] update rst and docstring (#1351) 2022-07-21 15:54:53 +08:00