Commits · f7361ee1bd31e57004d28418133e3714b08a53b2 · OpenDAS / ColossalAI

06 Apr, 2023 1 commit

[test] refactor tests with spawn (#3452) · 80eba05b

Frank Lee authored Apr 06, 2023

* [test] added spawn decorator

* polish code

* polish code

* polish code

* polish code

* polish code

* polish code

80eba05b

16 Aug, 2022 1 commit

[utils] Add use_reetrant=False in utils.activation_checkpoint (#1460) · 47fd8e4a

Boyuan Yao authored Aug 16, 2022

* [utils] Add use_reetrant=False into colossalai checkpoint

* [utils] add some annotation in utils.activaion_checkpoint

* [test] add reset_seed at the beginning of tests in test_actiavion_checkpointing.py

* [test] modify test_activation_checkpoint.py

* [test] modify test for reentrant=False

47fd8e4a

08 Jul, 2022 1 commit
- [tensor] fix some unittests (#1234) · 3b500984
  Jiarui Fang authored Jul 08, 2022
  
  3b500984
02 Apr, 2022 1 commit

[hotfix] fix bugs in testing (#659) · e5d615ae

HELSON authored Apr 02, 2022

* remove hybrid adam in test_moe_zero_optim

* fix activation checkpointing and its unitest

e5d615ae

14 Mar, 2022 1 commit
- [zero] memtracer to record cuda memory usage of model data and overall system (#395) · 21dc54e0
  Jiarui Fang authored Mar 14, 2022
  
  21dc54e0
11 Mar, 2022 2 commits
- fixed bug in activation checkpointing test (#387) · 1e4bf85c
  Frank Lee authored Mar 11, 2022
  
  1e4bf85c
- Added activation offload (#331) · de464504
  LuGY authored Mar 11, 2022
```
* Added activation offload

* Fixed the import bug, used the pytest
```
  de464504
16 Dec, 2021 1 commit
- added CI for unit testing (#69) · cd9c28e0
  Frank Lee authored Dec 16, 2021
  
  cd9c28e0
28 Oct, 2021 1 commit
- Migrated project · 404ecbdc
  zbian authored Oct 28, 2021
  
  404ecbdc