Commit Graph

682 Commits (bad5d4c0a1982b6d6bf7ff645d40cead992d6bc6)
 

Author SHA1 Message Date
Frank Lee bad5d4c0a1
[context] support lazy init of module (#1088)
3 years ago
ver217 be01db37c8
[tensor] refactor chunk mgr and impl MemStatsCollectorV2 (#1077)
3 years ago
Ziyue Jiang b3a03e4bfd
[Tensor] fix equal assert (#1091)
3 years ago
Frank Lee 50ec3a7e06
[test] skip tests when not enough GPUs are detected (#1090)
3 years ago
github-actions[bot] 3a7571b1d7
Automated submodule synchronization (#1081)
3 years ago
Frank Lee 1bd8a72fc9
[workflow] disable p2p via shared memory on non-nvlink machine (#1086)
3 years ago
Frank Lee 65ee6dcc20
[test] ignore 8 gpu test (#1080)
3 years ago
Ziyue Jiang 0653c63eaa
[Tensor] 1d row embedding (#1075)
3 years ago
junxu d66ffb4df4
Remove duplication registry (#1078)
3 years ago
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