ColossalAI/tests/test_auto_parallel
Zihao 18dbe76cae
[auto-parallel] add auto-offload feature (#3154)
* add auto-offload feature

* polish code

* fix syn offload runtime pass bug

* add offload example

* fix offload testing bug

* fix example testing bug
2023-03-21 14:17:41 +08:00
..
test_ckpt_solvers [hotfix] skip auto checkpointing tests (#3029) 2023-03-07 15:50:00 +08:00
test_offload [auto-parallel] add auto-offload feature (#3154) 2023-03-21 14:17:41 +08:00
test_pass
test_tensor_shard [autoparallel] apply repeat block to reduce solving time (#2912) 2023-02-28 11:03:30 +08:00
__init__.py