ColossalAI/colossalai/fx/passes
XYE 291e22aac6
[fx] temporarily used (#1215)
2022-07-06 17:19:26 +08:00
..
__init__.py [fx]add autoparallel passes (#1121) 2022-06-15 16:36:46 +08:00
adding_split_node_pass.py [fx]add uniform policy (#1208) 2022-07-06 13:48:11 +08:00
shard_1d_pass.py [fx] temporarily used (#1215) 2022-07-06 17:19:26 +08:00