ColossalAI/colossalai/fx/passes
YuliangLiu0306 189946c5c4
[fx]add uniform policy (#1208)
* [CLI] add CLI launcher

* Revert "[CLI] add CLI launcher"

This reverts commit df7e6506d4.

* [fx]add uniform policy
2022-07-06 13:48:11 +08:00
..
__init__.py [fx]add autoparallel passes (#1121) 2022-06-15 16:36:46 +08:00
adding_split_node_pass.py [fx]add uniform policy (#1208) 2022-07-06 13:48:11 +08:00
shard_1d_pass.py [fx]add autoparallel passes (#1121) 2022-06-15 16:36:46 +08:00