tests/test_zero/test_zero_optim_state_dict.py · 0c51ff2c135d911b0b1b250c6db9d367a18bc98d · OpenDAS / ColossalAI

ver217 authored Jul 18, 2022

* process group supports getting ranks in group

* chunk mgr receives a process group

* update unit test

* fix unit tests

0c51ff2c

test_zero_optim_state_dict.py 3.82 KB