ColossalAI/tests/test_fx
Frank Lee 5581170890
[fx] fixed huggingface OPT and T5 results misalignment (#1227)
2022-07-07 16:29:58 +08:00
..
test_tracer [fx] fixed huggingface OPT and T5 results misalignment (#1227) 2022-07-07 16:29:58 +08:00
test_coloproxy.py [fx] added timm model tracing testing (#1221) 2022-07-07 14:02:17 +08:00
test_comm_size_compute.py [fx]get communication size between partitions (#1224) 2022-07-07 16:22:00 +08:00
test_meta_info_prop.py [fx]get communication size between partitions (#1224) 2022-07-07 16:22:00 +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