ColossalAI/colossalai/fx/passes
Frank Lee 75abc75c15
[fx] fixed compatiblity issue with torch 1.10 (#1331)
2022-07-18 11:41:27 +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] Add unit test and fix bugs for transform_mlp_pass (#1299) 2022-07-15 14:37:58 +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