mirror of https://github.com/hpcaitech/ColossalAI
a7cda6f57d
* [fx] add input activation offload to codegen * [fx] modify unit test * [fx] remove two skips in torch11 * [fx] use all_input_nodes instead of _input_nodes |
||
---|---|---|
.. | ||
test_ckpt_solvers | ||
test_codegen | ||
test_meta | ||
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 |