ColossalAI/tests/test_zero/test_low_level
LuGY 79cf1b5f33 [zero]support no_sync method for zero1 plugin (#4138)
* support no sync for zero1 plugin

* polish

* polish
2023-07-31 22:13:29 +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 [zero] refactor low level zero for shard evenly (#4030) 2023-07-31 22:13:29 +08:00
test_zero_init.py
test_zero_tp.py