ColossalAI/tests/test_zero/test_low_level
LuGY d86ddd9b29
[hotfix] fix unsafe async comm in zero (#4404)
* improve stablility of zero

* fix wrong index

* add record stream
2023-08-11 15:09:24 +08:00
..
test_grad_acc.py [zero]support no_sync method for zero1 plugin (#4138) 2023-07-31 22:13:29 +08:00
test_zero1_2.py [hotfix] fix unsafe async comm in zero (#4404) 2023-08-11 15:09:24 +08:00
test_zero_ckpt.py [zero] add state dict for low level zero (#4179) 2023-07-31 22:13:29 +08:00
test_zero_init.py [zero] allow passing process group to zero12 (#4153) 2023-07-31 22:13:29 +08:00
test_zero_tp.py [zero] allow passing process group to zero12 (#4153) 2023-07-31 22:13:29 +08:00