ColossalAI/colossalai/fx/passes
Frank Lee 2cc1175c76
[fx] tested the complete workflow for auto-parallel (#1336)
* [fx] tested the complete workflow for auto-parallel

* polish code

* polish code

* polish code
2022-07-20 10:45:17 +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] 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