tests/test_tensor/test_dist_spec_mgr.py · ffa025e120cea28f4bf8d091e5f90f3388eed95f · OpenDAS / ColossalAI

[tensor] dist spec s2s uses all-to-all (#1136) · ffa025e1

ver217 authored Jun 22, 2022



* dist spec s2s uses all-to-all

* update unit test

* add sanity check

* polish unitest test with titans

* add sanity check for DistMgr

* add sanity check
Co-authored-by: jiaruifang <fangjiarui123@gmail.com>

ffa025e1

test_dist_spec_mgr.py 2.49 KB

Replace test_dist_spec_mgr.py