tests/test_tensor/test_zero_optim.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.py 5.8 KB