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
..
__init__.py	[fx] Add pofo solver (#1608)	2022-09-20 11:20:48 +08:00
build_c_ext.py	[autoparallel] add rotor C version (#1658)	2022-10-03 17:13:30 +08:00
ckpt_solver_chen.py	[fx] refactor memory utils and extend shard utils. (#1754)	2022-10-26 14:24:41 +08:00
ckpt_solver_pofo.py	[fx/meta/rpc] move _meta_registration.py to fx folder / register fx functions with compatibility checks / remove color debug (#1710)	2022-10-18 10:44:23 +08:00
ckpt_solver_rotor.py	[fx] refactor memory utils and extend shard utils. (#1754)	2022-10-26 14:24:41 +08:00
dynamic_programs.c	[autoparallel] fix C version rotor inconsistency (#1691)	2022-10-12 15:21:58 +08:00
linearize.py	[fx/tuning] tune performance on rotor with meta info. (#1599)	2022-09-15 14:46:36 +08:00
operation.py	[autoparallel] Add pofo sequence annotation (#1637)	2022-09-24 01:52:57 +08:00