ColossalAI/tests/test_auto_parallel/test_offload
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
..
model_utils.py [auto-parallel] add auto-offload feature (#3154) 2023-03-21 14:17:41 +08:00
test_perf.py [auto-parallel] add auto-offload feature (#3154) 2023-03-21 14:17:41 +08:00
test_solver.py [auto-parallel] add auto-offload feature (#3154) 2023-03-21 14:17:41 +08:00