mirror of https://github.com/hpcaitech/ColossalAI
e7383f578b
* [fx] polish ckpt_test. * [fx] add rules to linearize computation graphs for searching. * [fx] remove chen_sqrt for sake of simplicity * [fx] fix inconsistencies. |
||
---|---|---|
.. | ||
test_ckpt_solvers | ||
test_codegen | ||
test_pipeline | ||
test_tracer | ||
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 |