You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/tests/test_auto_parallel/test_offload
Frank Lee 638a07a7f9
[test] fixed gemini plugin test (#3411)
2 years ago
..
model_utils.py [auto-parallel] add auto-offload feature (#3154) 2 years ago
test_perf.py [test] fixed gemini plugin test (#3411) 2 years ago
test_solver.py [auto-parallel] add auto-offload feature (#3154) 2 years ago