ColossalAI/tests/test_fx
Frank Lee 5da87ce35d
[fx] added testing for all albert variants (#1211)
2022-07-06 15:11:08 +08:00
..
test_tracer [fx] added testing for all albert variants (#1211) 2022-07-06 15:11:08 +08:00
test_coloproxy.py [fx] added testing for all gpt variants (#1210) 2022-07-06 14:03:13 +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