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
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
shard_1d_pass.py
split_module.py
utils.py [fx] methods to get fx graph property. (#1246) 2022-07-12 14:10:37 +08:00