You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/tests/test_fx
Jiarui Fang 32c1b843a9
skip torchrec unittests if not installed (#1790)
2 years ago
..
test_ckpt_solvers [autoparallel] move ckpt solvers to autoparallel folder / refactor code (#1764) 2 years ago
test_codegen [autoparallel] move ckpt solvers to autoparallel folder / refactor code (#1764) 2 years ago
test_meta
test_pipeline [fx] support module with bias addition (#1780) 2 years ago
test_profiler
test_tracer skip torchrec unittests if not installed (#1790) 2 years ago
test_coloproxy.py
test_comm_size_compute.py
test_complete_workflow.py
test_graph_manipulation.py
test_meta_info_prop.py
test_parallel_1d.py
test_pipeline_passes.py