Commits · d16671da75ccb6aecacb2a15826a6d34367373d1 · OpenDAS / ColossalAI

28 Apr, 2022 3 commits
- [Tensor] initialize the ColoOptimizer (#898) · d16671da
  Jiarui Fang authored Apr 28, 2022
```
* [Tensor] activation is an attr of ColoTensor

* [Tensor] add optimizer

* only detach parameters in context

* polish code
```
  d16671da
- [Tensor] test parameters() as member function (#896) · e76f76c0
  Jiarui Fang authored Apr 28, 2022
  
  e76f76c0
- [tensor] refine linear and add gather for laynorm (#893) · cb182da7
  Ziyue Jiang authored Apr 28, 2022
```
* refine linear and add function to ColoTensor

* add gather for layernorm

* polish

* polish
```
  cb182da7
27 Apr, 2022 2 commits
- [Tensor] overriding paramters() for Module using ColoTensor (#889) · 26c49639
  Jiarui Fang authored Apr 27, 2022
  
  26c49639
- [Tensor] test model check results for a simple net (#887) · a0e59716
  Jiarui Fang authored Apr 27, 2022
  
  a0e59716
26 Apr, 2022 3 commits
- [Tensor] make a simple net works with 1D row TP (#879) · 7f76517a
  Jiarui Fang authored Apr 26, 2022
  
  7f76517a
- [Tensor] get named parameters for model using ColoTensors (#874) · e43f83aa
  Jiarui Fang authored Apr 26, 2022
  
  e43f83aa
- [tensor] customized op returns ColoTensor (#875) · 96211c2c
  Jiarui Fang authored Apr 26, 2022
```
* [tensor] customized op returns ColoTensor

* polish

* polish code
```
  96211c2c
25 Apr, 2022 1 commit
- [tensor] add cross_entrophy_loss (#868) · 1190b2c4
  Jiarui Fang authored Apr 25, 2022
  
  1190b2c4
24 Apr, 2022 2 commits
- [Tensor ] Add 1Drow weight reshard by spec (#854) · bcc86550
  Ziyue Jiang authored Apr 24, 2022
  
  bcc86550
- [Tensor] init a tp network training unittest (#849) · 62f05925
  Jiarui Fang authored Apr 24, 2022
  
  62f05925