github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-04 01:48:43 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
2dd4d556fb296059a9bed19061fd9e0027bfc3ea
ColossalAI/tests/test_tensor/test_module_spec.py
Jiarui Fang 1aad903c15 [tensor] redistribute among different process groups (#1247) 
* make it faster

* [tensor] rename convert_to_dist -> redistribute

* [tensor] ShardSpec and ReplicaSpec

* [tensor] redistribute among diff pgs

* polish code
2022-07-12 10:24:05 +08:00

8.8 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink