ColossalAI/colossalai/fx/passes
YuliangLiu0306 051592c64e
[fx] update MetaInforProp pass to process more complex node.meta (#1344)
* [CLI] add CLI launcher

* Revert "[CLI] add CLI launcher"

This reverts commit df7e6506d4.

* [fx] update MetaInforProp pass to process more complex node.meta
2022-07-21 10:57:52 +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] update MetaInforProp pass to process more complex node.meta (#1344) 2022-07-21 10:57:52 +08:00
shard_1d_pass.py [fx] tested the complete workflow for auto-parallel (#1336) 2022-07-20 10:45:17 +08:00
split_module.py [fx] fixed compatiblity issue with torch 1.10 (#1331) 2022-07-18 11:41:27 +08:00
utils.py [fx] methods to get fx graph property. (#1246) 2022-07-12 14:10:37 +08:00