ColossalAI/tests/test_auto_parallel/test_offload
Baizhou Zhang 0bb0b481b4 [gemini] fix argument naming during chunk configuration searching 2023-06-25 13:34:15 +08:00
..
model_utils.py [auto-parallel] add auto-offload feature (#3154) 2023-03-21 14:17:41 +08:00
test_perf.py [gemini] fix argument naming during chunk configuration searching 2023-06-25 13:34:15 +08:00
test_solver.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00