Super Daniel 0584654c79 [fx] refactor memory utils and extend shard utils. (#1754) ... * [fx] change memory.py to memory_utils.py. * [fx] add shard utils. * [fx] fix import. * [fx] check code style. * [fx] add comment. * [autoparallel] first move. * [fx] add time computations.		2022-10-26 14:24:41 +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] refactor memory utils and extend shard utils. (#1754)	2022-10-26 14:24:41 +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