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