Ziyue Jiang 632753abbc [fx]Split partition with DAG information (#2025) ...

* add DAG to split_module

* add comment

* add test case for DAG

* remove print

Co-authored-by: Ziyue Jiang <ziyue.jiang@gmail.com>

2022-11-25 17:42:48 +08:00

..

codegen

[autoparallel] move ckpt solvers to autoparallel folder / refactor code (#1764)

2022-11-01 10:43:15 +08:00

passes

[fx]Split partition with DAG information (#2025)

2022-11-25 17:42:48 +08:00

profiler

[fx] add more meta_registry for MetaTensor execution. (#2000)

2022-11-23 10:55:46 +08:00

tracer

[fx] add more meta_registry for MetaTensor execution. (#2000)

2022-11-23 10:55:46 +08:00

__init__.py

[fx] metainfo_trace as an API. (#1873)

2022-11-10 20:58:37 +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] add more meta_registry for MetaTensor execution. (#2000)

2022-11-23 10:55:46 +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