Commit Graph

773 Commits (11973d892d0273cc4719e30997cfaeafe4bc506c)
 

Author SHA1 Message Date
Jiarui Fang bcab249565
fix issue #1080 (#1071)
3 years ago
ver217 1b17859328
[tensor] chunk manager monitor mem usage (#1076)
3 years ago
ver217 98cdbf49c6
[hotfix] fix chunk comm src rank (#1072)
3 years ago
Frank Lee bfdc5ccb7b
[context] maintain the context object in with statement (#1073)
3 years ago
ver217 c5cd3b0f35
[zero] zero optim copy chunk rather than copy tensor (#1070)
3 years ago
Ziyue Jiang 4fc748f69b
[Tensor] fix optimizer for CPU parallel (#1069)
3 years ago
Jiarui Fang 49832b2344
[refactory] add nn.parallel module (#1068)
3 years ago
Ziyue Jiang 6754f1b77f
fix module utils bug (#1066)
3 years ago
Jiarui Fang a00644079e
reorgnize colotensor directory (#1062)
3 years ago
Frank Lee 3d10be33bd
[cudnn] set False to cudnn benchmark by default (#1063)
3 years ago
Ziyue Jiang df9dcbbff6
[Tensor] add hybrid device demo and fix bugs (#1059)
3 years ago
YuliangLiu0306 b167258b6a
[pipeline]refactor ppschedule to support tensor list (#1050)
3 years ago
ver217 e3fde4ee6b
fix import error in sharded model v2 (#1053)
3 years ago
ver217 e1922ea4f6
[zero] add chunk size search for chunk manager (#1052)
3 years ago
アマデウス 2c42b230f3
updated collective ops api (#1054)
3 years ago
ver217 51b9a49655
[zero] add zero optimizer for ColoTensor (#1046)
3 years ago
github-actions[bot] e32470b6de
Automated submodule synchronization (#1049)
3 years ago
Frank Lee 0ccb8c6141
[release] update version.txt (#1048)
3 years ago
binmakeswell 626dd187e4
add inference submodule (#1047)
3 years ago
ver217 7faef93326
fix dist spec mgr (#1045)
3 years ago
ver217 9492a561c3
[tensor] ColoTensor supports ZeRo (#1015)
3 years ago
Frank Lee cfa6c1b46b
[ci] fixed nightly build workflow (#1040)
3 years ago
YuliangLiu0306 9feff0f760
[titans]remove model zoo (#1042)
3 years ago
binmakeswell 0dac86866b
[NFC] add inference (#1044)
3 years ago
Ziyue Jiang 7c530b9de2
[Tensor] add Parameter inheritance for ColoParameter (#1041)
3 years ago
github-actions[bot] 4d8a574cd3
Automated submodule synchronization (#1034)
3 years ago
ver217 7cfd6c827e
[zero] add load_state_dict for sharded model (#894)
3 years ago
Ziyue Jiang 6c5996a56e
[Tensor] add module check and bert test (#1031)
3 years ago
YuliangLiu0306 7106bd671d
[p2p]add object list send/recv (#1024)
3 years ago
Frank Lee e4685832f8
[engine] fixed bug in gradient accumulation dataloader to keep the last step (#1030)
3 years ago
Ziyue Jiang 32291dd73f
[Tensor] add module handler for linear (#1021)
3 years ago
Frank Lee ee50497db2
[ci] fixed nightly build workflow (#1029)
3 years ago
Ryan Russell 9b0c037027
fix typo in constants (#1027)
3 years ago
ver217 007ca0df92
fix colo init context (#1026)
3 years ago
Frank Lee 58a7dd2ede
[ci] fixed nightly build workflow (#1022)
3 years ago
Frank Lee 1a76c88aba
[ci] added nightly build (#1018) (#1019)
3 years ago
Frank Lee 8d06186ff9
[doc] update docker instruction (#1020)
3 years ago
Frank Lee e17a43184b
[ci] update the docker image name (#1017)
3 years ago
YuliangLiu0306 d182b0bd47
[hotfix] fix some bugs caused by size mismatch. (#1011)
3 years ago
binmakeswell 9833d814d5 [NFC] fix paper link
3 years ago
ver217 cefc29ff06
[tensor] impl ColoDDP for ColoTensor (#1009)
3 years ago
zhengzangw ae7c338105 [NFC] polish colossalai/kernel/cuda_native/csrc/colossal_C_frontend.cpp code style
3 years ago
ver217 a3b66f6def
[tensor] refactor parallel action (#1007)
3 years ago
github-actions[bot] 9e3d602dba
Automated submodule synchronization (#1003)
3 years ago
ver217 8e3d0ad8f1
[unit test] refactor test tensor (#1005)
3 years ago
ver217 ad536e308e
[tensor] refactor colo-tensor (#992)
3 years ago
Frank Lee 1467d83edf
[cli] remove unused imports (#1001)
3 years ago
Frank Lee 533d0c46d8
[kernel] fixed the include bug in dropout kernel (#999)
3 years ago
binmakeswell c27ea0d980
fix download link (#998)
3 years ago
Jiarui Fang 802ac297cc
[Tensor] remove useless import in tensor dir (#997)
3 years ago