ColossalAI/colossalai/fx/passes
YuliangLiu0306 e8acf55e8b
[fx] add balanced policy v2 (#1251)
* [CLI] add CLI launcher

* Revert "[CLI] add CLI launcher"

This reverts commit df7e6506d4.

* [fx] add balanced policy v2

* add unittest
2022-07-15 14:54:26 +08:00
..
__init__.py [fx]add autoparallel passes (#1121) 2022-06-15 16:36:46 +08:00
adding_split_node_pass.py [fx] add balanced policy v2 (#1251) 2022-07-15 14:54:26 +08:00
meta_info_prop.py [fx] add balanced policy v2 (#1251) 2022-07-15 14:54:26 +08:00
shard_1d_pass.py [fx] Add unit test and fix bugs for transform_mlp_pass (#1299) 2022-07-15 14:37:58 +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