ColossalAI/tests/test_fx
Frank Lee 58df720570 [shardformer] adapted T5 and LLaMa test to use kit (#4049)
* [shardformer] adapted T5 and LLaMa test to use kit

* polish code
2023-07-04 16:05:01 +08:00
..
test_codegen [CI] fix typo with tests/ etc. (#3727) 2023-05-11 16:30:58 +08:00
test_meta [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_pipeline [autoparallel] adapt autoparallel with new analyzer (#3261) 2023-03-30 17:47:24 +08:00
test_profiler [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_tracer [shardformer] adapted T5 and LLaMa test to use kit (#4049) 2023-07-04 16:05:01 +08:00
test_coloproxy.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_comm_size_compute.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_graph_manipulation.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_meta_info_prop.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_parallel_1d.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_pipeline_passes.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00