github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/tests/test_auto_parallel/test_offload
History
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