mirror of https://github.com/hpcaitech/ColossalAI
4f59693207
* [fx] compute memory stat and flop count for MetaInfoProp. * [fx] modify node attribute. * [fx] modify ckpt_chen. * [fx] fix compatibility. * [fx] fix import error. * [fx] skip test for MetaInfoProp. * [fx] skip test for MetaInfoProp. * [fx] skip test for MetaInfoProp. * [fx] skip test for MetaInfoProp. * [fx] skip if torch 1.11.0. * [fx] recover MetaInfoProp support for PyTorch 1.11. * [fx] provide a stable but not accurate enough version of profiler. * [fx] provide a stable but not accurate enough version of profiler. * [fx] fix compatibility in tests. * [fx] fix compatibility in tests. * [fx] fix compatibility in tests. * [fx] fix compatibility in tests. * [fx] fix compatibility in tests. * [fx] fix compatibility in tests. * [fx] fix compatibility in tests. * [fx] fix compatibility in tests. * [fx] fix compatibility in tests. * [fx] fix compatibility in tests. * [fx] fix import error. |
||
---|---|---|
.. | ||
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 |