ColossalAI/colossalai/fx/passes
Jiarui Fang db1bef9032
[hotfix] fx shard 1d pass bug fixing (#1220)
2022-07-07 13:37:31 +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 [hotfix] fx shard 1d pass bug fixing (#1220) 2022-07-07 13:37:31 +08:00