DeepSpeed
4fc181b0 - [CCLBackend] update API (#4378)

Commit
2 years ago
[CCLBackend] update API (#4378) * unify ccl.py for cpu and gpu' * handle all coll in on func * add coresponding func in cclbackend * add coresponding pybind func * change device setting --------- Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>
Author
Parents
Loading