ColossalAI/colossalai/fx/passes
YuliangLiu0306 fcf55777dd
[fx]add autoparallel passes (#1121)
* [CLI] add CLI launcher

* Revert "[CLI] add CLI launcher"

This reverts commit df7e6506d4.

* feature/add autoparallel passes
2022-06-15 16:36:46 +08:00
..
__init__.py [fx]add autoparallel passes (#1121) 2022-06-15 16:36:46 +08:00
adding_split_node_pass.py [fx]add autoparallel passes (#1121) 2022-06-15 16:36:46 +08:00
shard_1d_pass.py [fx]add autoparallel passes (#1121) 2022-06-15 16:36:46 +08:00