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

..

codegen

[fx] fix offload codegen test (#1648)

2022-09-27 10:25:27 +08:00

passes

[fx] refactor memory utils and extend shard utils. (#1754)

2022-10-26 14:24:41 +08:00

profiler

[fx] refactor memory utils and extend shard utils. (#1754)

2022-10-26 14:24:41 +08:00

tracer

[fx] refactor memory utils and extend shard utils. (#1754)

2022-10-26 14:24:41 +08:00

__init__.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

_compatibility.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

_meta_registrations.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

graph_module.py

[fx] Add offload codegen (#1598)

2022-09-14 15:49:06 +08:00

proxy.py

[hotfix] fix coloproxy typos. (#1519)

2022-08-30 11:39:03 +08:00