github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-18 08:35:08 +00:00
Code Issues Packages Projects Releases Wiki Activity
1,217 Commits 60 Branches 51 Tags
f8aeecef46461ff574f51982d03310fa8c57888e
Commit Graph

3 Commits

Author SHA1 Message Date
YuliangLiu0306
4b03c25f85 [tensor]add 1D device mesh (#1492) 2022-08-25 16:48:12 +08:00
YuliangLiu0306
b73fb7a077 [tensor] support runtime ShardingSpec apply (#1453) 
* [tensor] support runtime ShardingSpec apply

* polish code

* polish code
 2022-08-19 13:39:51 +08:00
YuliangLiu0306
0442f940f0 [device] add DeviceMesh class to support logical device layout (#1394) 
* [device] add DeviceMesh class to support logical device layout

* polish code

* add doc string
 2022-08-02 19:23:48 +08:00