Files
ColossalAI/colossalai/fx/passes/adding_split_node_pass.py
YuliangLiu0306 52bc2dc271 [fx] update split module pass and add customized policy (#1373)
* [CLI] add CLI launcher

* Revert "[CLI] add CLI launcher"

This reverts commit df7e6506d4.

* [fx]update split module pass and add customized policy
2022-07-27 13:40:54 +08:00

4.8 KiB