github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-02-21 14:32:09 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
c668801d363eb0142d8cb0fc789b1cf7d55f8077
ColossalAI/tests/test_zero/test_low_level
History
LuGY c668801d36 [zero] allow passing process group to zero12 (#4153) 
* allow passing process group to zero12

* union tp-zero and normal-zero

* polish code
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
[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