github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-11 18:14:16 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
198a74b9fdfb4fccd33ef621acaceed45f75445c
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