ColossalAI/tests/test_fx
Jiarui Fang ae7d3f4927
[refactor] move process group from _DistSpec to ColoTensor. (#1203)
2022-07-06 16:15:16 +08:00
..
test_tracer [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