ColossalAI/colossalai/fx/passes
YuliangLiu0306 97d713855a
[fx] methods to get fx graph property. (#1246)
* [CLI] add CLI launcher

* Revert "[CLI] add CLI launcher"

This reverts commit df7e6506d4.

* manipulation

* [fx]add graph manipulation methods.

* [fx]methods to get fx graph property.

* add unit test

* add docstring to explain top node and leaf node in this context
2022-07-12 14:10:37 +08:00
..
__init__.py [fx]add autoparallel passes (#1121) 2022-06-15 16:36:46 +08:00
adding_split_node_pass.py [fx] methods to get fx graph property. (#1246) 2022-07-12 14:10:37 +08:00
meta_info_prop.py [fx]get communication size between partitions (#1224) 2022-07-07 16:22:00 +08:00
shard_1d_pass.py [tensor] a shorter shard and replicate spec (#1245) 2022-07-11 15:51:48 +08:00
split_module.py [fx]add split module pass and unit test from pipeline passes (#1242) 2022-07-12 13:45:01 +08:00
utils.py [fx] methods to get fx graph property. (#1246) 2022-07-12 14:10:37 +08:00