github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-20 09:34:48 +00:00
Code Issues Packages Projects Releases Wiki Activity
542 Commits 60 Branches 51 Tags
4df6471f5d044150edfedfe1cf8370a604e096ca
Commit Graph

4 Commits

Author SHA1 Message Date
Jiarui Fang
909211453b [Tensor] Add some attributes to ColoTensor (#877) 
* [Tensor] add some function to ColoTensor

* torch.allclose

* rm torch.add
 2022-04-26 15:10:47 +08:00
Jiarui Fang
96211c2cc8 [tensor] customized op returns ColoTensor (#875) 
* [tensor] customized op returns ColoTensor

* polish

* polish code
 2022-04-26 13:23:59 +08:00
Jiarui Fang
126ba573a8 [Tensor] add layer norm Op (#852) 2022-04-25 11:49:20 +08:00
Jiarui Fang
0ce8924ceb [tensor] reorganize files (#820) 2022-04-21 14:15:48 +08:00