Commit Graph

3873 Commits

Author SHA1 Message Date
binmakeswell
6582aedc94 fix format (#583) 2022-04-06 11:40:59 +08:00
DouJS
f08fc17f2b block_reduce.h fix format (#581) 2022-04-06 11:40:59 +08:00
Maruyama_Aya
d2dc6049b5 fix format (#580) 2022-04-06 11:40:59 +08:00
wky
174b9c1d85 fix format (#574) 2022-04-06 11:40:59 +08:00
BoxiangW
dfe423ae42 fix format (#572) 2022-04-06 11:40:59 +08:00
yuxuan-lou
cfb41297ff 'fix/format' (#573) 2022-04-06 11:40:59 +08:00
Kai Wang (Victor Kai)
b0f708dfc1 fix format (#570) 2022-04-06 11:40:59 +08:00
Xu Kai
2a915a8b62 fix format (#568) 2022-04-06 11:40:59 +08:00
YuliangLiu0306
9420d3ae31 fix format (#567) 2022-04-06 11:40:59 +08:00
Jie Zhu
0f1da44e5e [format]colossalai/kernel/cuda_native/csrc/layer_norm_cuda.cpp (#566) 2022-04-06 11:40:59 +08:00
coder-chin
5835631218 fix format (#564) 2022-04-06 11:40:59 +08:00
Luxios22
e014144c44 fix format (#565) 2022-04-06 11:40:59 +08:00
Ziyue Jiang
1762ba14ab fix format (#563) 2022-04-06 11:40:59 +08:00
Sze-qq
ce8a3eae5b
update GPT-2 experiment result (#666) 2022-04-04 13:47:43 +08:00
HELSON
17e73e62cc
[hotfix] fix bugs for unsharded parameters when restore data (#664) 2022-04-03 22:02:11 +08:00
Jiarui Fang
0aab52301e
[hotfix] fix a bug in model data stats tracing (#655) 2022-04-03 21:48:06 +08:00
YuliangLiu0306
ade05a5d83
[refactor] pipeline, put runtime schedule into engine. (#627) 2022-04-03 20:46:45 +08:00
HELSON
e5d615aeee
[hotfix] fix bugs in testing (#659)
* remove hybrid adam in test_moe_zero_optim

* fix activation checkpointing and its unitest
2022-04-02 21:58:47 +08:00
Jiarui Fang
036404ca8a
Revert "[zero] polish init context (#645)" (#657) 2022-04-02 18:30:06 +08:00
HELSON
b31daed4cf
fix bugs in CPU adam (#633)
* add cpu adam counter for all cpu adam

* fixed updating error in adam kernel
2022-04-02 17:04:05 +08:00
LuGY
1e2557e801
[zero] fixed the activation offload (#647)
* fixed the activation offload

* polish
2022-04-02 16:21:32 +08:00
Liang Bowen
828e465622
[hotfix] Raise messages for indivisible batch sizes with tensor parallelism (#622) 2022-04-02 16:12:04 +08:00
binmakeswell
e0f875a8e2
[GitHub] Add prefix and label in issue template (#652) 2022-04-02 16:09:25 +08:00
Jiarui Fang
67b4928244
[zero] polish init context (#645) 2022-04-02 15:52:04 +08:00
ver217
f5d3a9c2b0
polish checkpoint docstring (#637) 2022-04-02 13:34:33 +08:00
HELSON
055fbf5be6
[zero] adapt zero for unsharded paramters (Optimizer part) (#601) 2022-04-01 20:10:47 +08:00
KAIYUAN GAN
229382c844
[NFC] polish colossalai/kernel/cuda_native/csrc/kernels/cuda_util.cu code stype (#625) 2022-04-01 17:45:53 +08:00
アマデウス
354b7954d1
[model checkpoint] added unit tests for checkpoint save/load (#599) 2022-04-01 16:53:32 +08:00
アマデウス
28b515d610
[model checkpoint] updated checkpoint hook (#598) 2022-04-01 16:53:03 +08:00
アマデウス
77ad24bf94
[model checkpoint] updated saving/loading for 3d layers (#597) 2022-04-01 16:52:47 +08:00
アマデウス
93089ed708
[model checkpoint] updated saving/loading for 2.5d layers (#596) 2022-04-01 16:52:33 +08:00
アマデウス
6302069c0e
[model checkpoint] updated communication ops for cpu tensors (#590) 2022-04-01 16:52:20 +08:00
アマデウス
c50bfb807b
[model checkpoint] updated saving/loading for 1d layers (#594) 2022-04-01 16:51:52 +08:00
アマデウス
7636d518e1
[model checkpoint] updated saving/loading for 2d layers (#595) 2022-04-01 16:50:34 +08:00
アマデウス
cd13b63832
[model checkpoint] reworked unified layers for ease of save/load states (#593) 2022-04-01 16:49:56 +08:00
アマデウス
acae68eb04
[model checkpoint] updated checkpoint save/load utils (#592) 2022-04-01 16:49:21 +08:00
Ziyue Jiang
1c40ee8749
[TP] add assert for tp1d (#621) 2022-04-01 16:44:23 +08:00
ver217
369a288bf3
polish utils docstring (#620) 2022-04-01 16:36:47 +08:00
ver217
e619a651fb
polish optimizer docstring (#619) 2022-04-01 16:27:03 +08:00
ver217
8432dc7080
polish moe docsrting (#618) 2022-04-01 16:15:36 +08:00
ver217
c5b488edf8
polish amp docstring (#616) 2022-04-01 16:09:39 +08:00
ver217
f69507dd22
update rst (#615) 2022-04-01 15:46:38 +08:00
FredHuang99
93f14d2a33
[zero] test zero tensor utils (#609) 2022-04-01 15:16:59 +08:00
ver217
0ef8819c67
polish docstring of zero (#612) 2022-04-01 14:50:56 +08:00
LuGY
02b187c14f
[zero] add sampling time for memstats collector (#610) 2022-04-01 14:03:00 +08:00
ver217
9bee119104
[hotfix] fix sharded optim zero grad (#604)
* fix sharded optim zero grad

* polish comments
2022-04-01 12:41:20 +08:00
アマデウス
297b8baae2
[model checkpoint] add gloo groups for cpu tensor communication (#589) 2022-04-01 10:15:52 +08:00
アマデウス
54e688b623
moved ensure_path_exists to utils.common (#591) 2022-04-01 09:46:33 +08:00
Jiarui Fang
e956d93ac2
[refactor] memory utils (#577) 2022-04-01 09:22:33 +08:00
ver217
104cbbb313
[hotfix] add hybrid adam to __init__ (#584) 2022-03-31 19:08:34 +08:00