ColossalAI/tests/test_fx
Boyuan Yao f3687e4ee2
[fx] Add nested checkpoint in activation checkpoint codegen (#1585)
* [fx] add nested activation_checkpoint codegen

* undo algorithms commits

* solver

* undo some commits

* [fx] torch11 add nested activation checkpoint codegen

* remove some imports

* [fx] add some comments in activation codegen

* [fx] codegen instance error fix
2022-09-12 20:00:48 +08:00
..
test_ckpt_solvers [fx] provide a stable but not accurate enough version of profiler. (#1547) 2022-09-07 11:21:04 +08:00
test_codegen [fx] Add nested checkpoint in activation checkpoint codegen (#1585) 2022-09-12 20:00:48 +08:00
test_meta [fx] provide a stable but not accurate enough version of profiler. (#1547) 2022-09-07 11:21:04 +08:00
test_pipeline [fx] refactor tracer to trace complete graph (#1342) 2022-07-20 11:20:38 +08:00
test_tracer [fx]patch nn.functional convolution (#1528) 2022-09-01 19:05:07 +08:00
test_coloproxy.py [fx] refactor tracer to trace complete graph (#1342) 2022-07-20 11:20:38 +08:00
test_comm_size_compute.py [fx] provide a stable but not accurate enough version of profiler. (#1547) 2022-09-07 11:21:04 +08:00
test_complete_workflow.py [fx] tested the complete workflow for auto-parallel (#1336) 2022-07-20 10:45:17 +08:00
test_graph_manipulation.py
test_meta_info_prop.py [fx] provide a stable but not accurate enough version of profiler. (#1547) 2022-09-07 11:21:04 +08:00
test_parallel_1d.py
test_pipeline_passes.py