ColossalAI/tests/test_fx at ae7d3f49270c08fe3e37c82821f7d1de10de2837 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-16 07:28:45 +00:00

Files

History

Jiarui Fang ae7d3f4927 [refactor] move process group from _DistSpec to ColoTensor. (#1203 )

2022-07-06 16:15:16 +08:00

..

[fx] added testing for all albert variants (#1211 )

2022-07-06 15:11:08 +08:00

test_coloproxy.py

[refactor] move process group from _DistSpec to ColoTensor. (#1203 )

2022-07-06 16:15:16 +08:00

test_parallel_1d.py

[fx]add autoparallel passes (#1121 )

2022-06-15 16:36:46 +08:00

test_pipeline_passes.py

[fx]add uniform policy (#1208 )

2022-07-06 13:48:11 +08:00