github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-19 00:55:09 +00:00
Code Issues Packages Projects Releases Wiki Activity
2,512 Commits 60 Branches 51 Tags
4021b9a8a2dd3a9155bba04c0ed2cd7362fa437f
Commit Graph

2 Commits

Author SHA1 Message Date
Frank Lee
45d9384346 [shardformer] removed inplace tensor sharding (#4018) 2023-07-04 16:05:01 +08:00
Frank Lee
015af592f8 [shardformer] integrated linear 1D with dtensor (#3996) 
* [shardformer] integrated linear 1D with dtensor

* polish code
 2023-07-04 16:05:01 +08:00