675 Commits (0653c63eaacd1504f5d66f2e11f80defdb155832)

 

All Branches   

Search

Author	SHA1	Message	Date
Ziyue Jiang	0653c63eaa	[Tensor] 1d row embedding (#1075) ... * Add CPU 1d row embedding * polish	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) ... * 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