Commit Graph

15 Commits (be229217ce963ace870795ee281c98e438b5130b)

Author SHA1 Message Date
YuliangLiu0306 5542816690
[fx]add gpt2 passes for pipeline performance test (#1366)
2 years ago
YuliangLiu0306 051592c64e
[fx] update MetaInforProp pass to process more complex node.meta (#1344)
2 years ago
Frank Lee 2cc1175c76
[fx] tested the complete workflow for auto-parallel (#1336)
2 years ago
Frank Lee 75abc75c15
[fx] fixed compatiblity issue with torch 1.10 (#1331)
2 years ago
YuliangLiu0306 e8acf55e8b
[fx] add balanced policy v2 (#1251)
2 years ago
XYE ca2d3f284f
[fx] Add unit test and fix bugs for transform_mlp_pass (#1299)
2 years ago
YuliangLiu0306 97d713855a
[fx] methods to get fx graph property. (#1246)
2 years ago
YuliangLiu0306 30b4fc0eb0
[fx]add split module pass and unit test from pipeline passes (#1242)
2 years ago
Jiarui Fang 9bcd2fd4af
[tensor] a shorter shard and replicate spec (#1245)
2 years ago
Jiarui Fang 0e199d71e8
[hotfix] fx get comm size bugs (#1233)
2 years ago
YuliangLiu0306 2b7dca44b5
[fx]get communication size between partitions (#1224)
2 years ago
Jiarui Fang db1bef9032
[hotfix] fx shard 1d pass bug fixing (#1220)
2 years ago
XYE 291e22aac6
[fx] temporarily used (#1215)
2 years ago
YuliangLiu0306 189946c5c4
[fx]add uniform policy (#1208)
2 years ago
YuliangLiu0306 fcf55777dd
[fx]add autoparallel passes (#1121)
2 years ago