Commits · e532679c95e60492ddc6ce0931a73cc0d677cfaf · OpenDAS / ColossalAI

06 Jan, 2023 1 commit

[setup] support pre-build and jit-build of cuda kernels (#2374) · 40d376c5

Frank Lee authored Jan 06, 2023

* [setup] support pre-build and jit-build of cuda kernels

* polish code

* polish code

* polish code

* polish code

* polish code

* polish code

40d376c5

04 Jan, 2023 1 commit
- [builder] reconfig op_builder for pypi install (#2314) · db6eea35
  Jiarui Fang authored Jan 04, 2023
  
  db6eea35
23 Dec, 2022 2 commits
- [buider] use builder() for cpu adam and fused optim in setup.py (#2187) · bc0e271e
  Jiarui Fang authored Dec 23, 2022
  
  bc0e271e
- [builder] runtime adam and fused_optim builder (#2184) · d42afd30
  Jiarui Fang authored Dec 23, 2022
  
  d42afd30
17 Nov, 2022 1 commit
- [kernel] move all symlinks of kernel to `colossalai._C` (#1971) · f8a7148d
  ver217 authored Nov 17, 2022
  
  f8a7148d
05 Aug, 2022 1 commit
- [hotfix] fix CPUAdam kernel nullptr (#1410) · 12b48870
  ver217 authored Aug 05, 2022
  
  12b48870
25 Mar, 2022 2 commits
- [zero]added hybrid adam, removed loss scale in adam (#527) · 105c5301
  LuGY authored Mar 25, 2022
```
* [zero]added hybrid adam, removed loss scale of adam

* remove useless code
```
  105c5301
- [cuda] modify the fused adam, support hybrid of fp16 and fp32 (#497) · 6a3f9fda
  LuGY authored Mar 25, 2022
  
  6a3f9fda
16 Mar, 2022 1 commit
- [hotfix] run cpu adam unittest in pytest (#424) · 5d7dc352
  Jiarui Fang authored Mar 16, 2022
  
  5d7dc352
11 Mar, 2022 1 commit

[zero] cpu adam kernel (#288) · a3269de5

LuGY authored Mar 04, 2022



* Added CPU Adam

* finished the cpu adam

* updated the license

* delete useless parameters, removed resnet

* modified the method off cpu adam unittest

* deleted some useless codes

* removed useless codes
Co-authored-by: ver217 <lhx0217@gmail.com>
Co-authored-by: Frank Lee <somerlee.9@gmail.com>
Co-authored-by: jiaruifang <fangjiarui123@gmail.com>

a3269de5