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