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 )
...
* reorgnize colotensor directory
* polish code
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 )
...
* [CLI] add CLI launcher
* Revert "[CLI] add CLI launcher"
This reverts commit df7e6506d4
.
* refactor ppschedule to support tensor list
* polish
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 )
...
* add zero optimizer
* torch ok
* unit test ok
* polish code
* fix bugs
* polish unit test
* polish zero optim
* polish colo ddp v2
* refactor folder structure
* add comment
* polish unit test
* polish zero optim
* polish unit test
3 years ago
github-actions[bot]
e32470b6de
Automated submodule synchronization ( #1049 )
...
Co-authored-by: github-actions <github-actions@github.com>
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 )
...
* impl chunk manager
* impl param op hook
* add reduce_chunk
* add zero hook v2
* add zero dp
* fix TensorInfo
* impl load balancing when using zero without chunk
* fix zero hook
* polish chunk
* fix bugs
* ddp ok
* zero ok
* polish code
* fix bugs about load balancing
* polish code
* polish code
* add ene-to-end test
* polish code
* polish code
* polish code
* fix typo
* add test_chunk
* fix bugs
* fix bugs
* polish code
3 years ago
Frank Lee
cfa6c1b46b
[ci] fixed nightly build workflow ( #1040 )
3 years ago
YuliangLiu0306
9feff0f760
[titans]remove model zoo ( #1042 )
...
* [CLI] add CLI launcher
* Revert "[CLI] add CLI launcher"
This reverts commit df7e6506d4
.
* rm model zoo
3 years ago
binmakeswell
0dac86866b
[NFC] add inference ( #1044 )
3 years ago
Ziyue Jiang
7c530b9de2
[Tensor] add Parameter inheritance for ColoParameter ( #1041 )
...
* add Parameter inheritance for ColoParameter
* remove tricks
* remove tricks
* polish
* polish
3 years ago
github-actions[bot]
4d8a574cd3
Automated submodule synchronization ( #1034 )
...
Co-authored-by: github-actions <github-actions@github.com>
3 years ago
ver217
7cfd6c827e
[zero] add load_state_dict for sharded model ( #894 )
...
* add load_state_dict for sharded model
* fix bug
* fix bug
* fix ckpt dtype and device
* support load state dict in zero init ctx
* fix bugs
3 years ago
Ziyue Jiang
6c5996a56e
[Tensor] add module check and bert test ( #1031 )
...
* add Embedding
* Add bert test
* polish
* add check module test
* polish
* polish
* polish
* polish
3 years ago
YuliangLiu0306
7106bd671d
[p2p]add object list send/recv ( #1024 )
...
* [CLI] add CLI launcher
* Revert "[CLI] add CLI launcher"
This reverts commit df7e6506d4
.
* [p2p]add object list send recv
* refactor for code reusability
* polish
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 )
...
* add module spec for linear
* polish
* polish
* polish
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 )
...
* [ci] fixed nightly build workflow
* [ci] fixed nightly build workflow
* [ci] fixed nightly build workflow
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 )
...
* [CLI] add CLI launcher
* Revert "[CLI] add CLI launcher"
This reverts commit df7e6506d4
.
* [hotfix]fix some bugs caused by size mismatch.
* add warning logs
* polish
3 years ago
binmakeswell
9833d814d5
[NFC] fix paper link
3 years ago
ver217
cefc29ff06
[tensor] impl ColoDDP for ColoTensor ( #1009 )
...
* impl ColoDDP for ColoTensor
* polish code
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 )
...
* refactor parallel action
* polish unit tests
3 years ago
github-actions[bot]
9e3d602dba
Automated submodule synchronization ( #1003 )
...
Co-authored-by: github-actions <github-actions@github.com>
3 years ago
ver217
8e3d0ad8f1
[unit test] refactor test tensor ( #1005 )
...
* polish test_gpt
* update op unit tests
* update test model
3 years ago
ver217
ad536e308e
[tensor] refactor colo-tensor ( #992 )
...
* refactor colo-tensor and update linear op
* polish code
* polish code
* update ops and unit tests
* update unit tests
* polish code
* rename dist_spec module
* polish code
* polish code
* remove unneeded import
* fix pipelinable
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