github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-14 22:28:18 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
efdc240f1fba00efd670cea357b7f04ef976ed0c
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