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_zero/test_low_level
LuGY cbac782254
[zero]fix zero ckptIO with offload (#4529)
1 year ago
..
test_grad_acc.py [zero]support zero2 with gradient accumulation (#4511) 1 year ago
test_zero1_2.py [hotfix] fix unsafe async comm in zero (#4404) 1 year ago
test_zero_ckpt.py [zero]fix zero ckptIO with offload (#4529) 1 year ago
test_zero_tp.py [gemini] improve compatibility and add static placement policy (#4479) 1 year ago