ColossalAI/tests/test_fx
Super Daniel e7383f578b
[fx] add rules to linearize computation graphs for searching. (#1461)
* [fx] polish ckpt_test.

* [fx] add rules to linearize computation graphs for searching.

* [fx] remove chen_sqrt for sake of simplicity

* [fx] fix inconsistencies.
2022-08-17 14:47:12 +08:00
..
test_ckpt_solvers [fx] add rules to linearize computation graphs for searching. (#1461) 2022-08-17 14:47:12 +08:00
test_codegen [fx] Add use_reentrant=False to checkpoint in codegen (#1463) 2022-08-17 10:34:50 +08:00
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