ColossalAI

Commit Graph

Author	SHA1	Message	Date
github-actions[bot]	3b1f5f07ce	Automated submodule synchronization (#907 ) Co-authored-by: github-actions <github-actions@github.com>	3 years ago
Ziyue Jiang	f593a5637e	[Tensor] add embedding tp1d row (#904 )	3 years ago
ver217	16122d5fac	update release bdist CI (#902 )	3 years ago
Ziyue Jiang	2c0d19d755	[Tensor] add ColoTensor TP1Dcol Embedding (#899 )	3 years ago
ver217	e46e423c00	add CI for releasing bdist wheel (#901 )	3 years ago
Jiarui Fang	e1108caf7d	change version to 0.1.4 (#900 )	3 years ago
Jiarui Fang	d16671da75	[Tensor] initialize the ColoOptimizer (#898 ) * [Tensor] activation is an attr of ColoTensor * [Tensor] add optimizer * only detach parameters in context * polish code	3 years ago
Jiarui Fang	676f191532	[Tensor] activation is an attr of ColoTensor (#897 )	3 years ago
Jiarui Fang	e76f76c08b	[Tensor] test parameters() as member function (#896 )	3 years ago
Ziyue Jiang	cb182da7c5	[tensor] refine linear and add gather for laynorm (#893 ) * refine linear and add function to ColoTensor * add gather for layernorm * polish * polish	3 years ago
Jiarui Fang	26c49639d8	[Tensor] overriding paramters() for Module using ColoTensor (#889 )	3 years ago
ver217	daf59ff72e	[setup] add local version label (#890 )	3 years ago
Ziyue Jiang	1d0aba4153	[tensor] add ColoTensor 1Dcol (#888 )	3 years ago
Jiarui Fang	a0e5971692	[Tensor] test model check results for a simple net (#887 )	3 years ago
Jiarui Fang	72cdc06875	[Tensor] make ColoTensor more robust for getattr (#886 ) * [Tensor] make ColoTensor more robust for getattr * polish * polish	3 years ago
Ziyue Jiang	9bc5a77c31	[tensor] wrap function in the torch_tensor to ColoTensor (#881 )	3 years ago
ver217	4df6471f5d	fix import error (#880 )	3 years ago
Jiarui Fang	7f76517a85	[Tensor] make a simple net works with 1D row TP (#879 )	3 years ago
ver217	c4d903e64a	[gemini] accelerate adjust_layout() (#878 ) * add lru cache * polish code * update unit test * fix sharded optim	3 years ago
Jiarui Fang	909211453b	[Tensor] Add some attributes to ColoTensor (#877 ) * [Tensor] add some function to ColoTensor * torch.allclose * rm torch.add	3 years ago
HELSON	425b4a96b8	[gemini] polish stateful_tensor_mgr (#876 )	3 years ago
Jiarui Fang	e43f83aa5c	[Tensor] get named parameters for model using ColoTensors (#874 )	3 years ago
LuGY	2883040286	[example] change qkv processing (#870 )	3 years ago
Jiarui Fang	96211c2cc8	[tensor] customized op returns ColoTensor (#875 ) * [tensor] customized op returns ColoTensor * polish * polish code	3 years ago
Ziyue Jiang	26d4ab8b03	[Tensor] Add function to spec and update linear 1Drow and unit tests (#869 )	3 years ago
Frank Lee	11f54c7b6b	[doc] improved docstring and assertion messages for the engine module (#871 )	3 years ago
Frank Lee	1c34382678	[doc] improved assertion messages in trainer (#873 )	3 years ago
Frank Lee	7a64fae33a	[doc] improved error messages in initialize (#872 )	3 years ago
Jiarui Fang	1190b2c4a4	[tensor] add cross_entrophy_loss (#868 )	3 years ago
HELSON	3107817172	[gemini] add stateful tensor container (#867 )	3 years ago
Jiarui Fang	d01d3b8cb0	colo init context add device attr. (#866 )	3 years ago
Frank Lee	2238758c2e	[usability] improved error messages in the context module (#856 )	3 years ago
Frank Lee	9fdebadd69	[doc] improved docstring in the amp module (#857 )	3 years ago
Frank Lee	b862d89d00	[doc] improved docstring in the logging module (#861 )	3 years ago
Frank Lee	8004c8e938	[doc] improved docstring in the communication module (#863 )	3 years ago
Jiarui Fang	8af5f7423d	[tensor] an initial dea of tensor spec (#865 ) * a initial dea of tensor spec * polish * polish	3 years ago
Jiarui Fang	126ba573a8	[Tensor] add layer norm Op (#852 )	3 years ago
Frank Lee	a82da26f7e	[cli] refactored micro-benchmarking cli and added more metrics (#858 )	3 years ago
Frank Lee	ee222dfbf3	[usability] added assertion message in registry (#864 )	3 years ago
HELSON	f0e654558f	[gemini] polish code (#855 )	3 years ago
Jiarui Fang	29159d9b5b	hotfix tensor unittest bugs (#862 )	3 years ago
Frank Lee	1258af71cc	[ci] cache cuda extension (#860 )	3 years ago
YuliangLiu0306	c6930d8ddf	[pipelinable]use ColoTensor to replace dummy tensor. (#853 )	3 years ago
Ziyue Jiang	bcc8655021	[Tensor ] Add 1Drow weight reshard by spec (#854 )	3 years ago
ver217	d7e0303d1e	[zero] use GeminiMemoryManager when sampling model data (#850 )	3 years ago
ver217	232142f402	[utils] refactor profiler (#837 ) * add model data profiler * add a subclass of torch.profiler.profile * refactor folder structure * remove redundant codes * polish code * use GeminiMemoryManager * fix import path * fix stm profiler ext * polish comments * remove useless file	3 years ago
Jiarui Fang	62f059251b	[Tensor] init a tp network training unittest (#849 )	3 years ago
ver217	0dea140760	[hotfix] add deconstructor for stateful tensor (#848 ) * add deconstructor for stateful tensor * fix colo init context	3 years ago
ver217	0f7ed8c192	fix _post_init_method of zero init ctx (#847 )	3 years ago
Ziyue Jiang	2a0a427e04	[tensor]add assert for colo_tensor 1Drow (#846 )	3 years ago

1 2 3 4 5 ...

558 Commits (3b1f5f07ce37668716d7034ca3549bd835beefb5) All Branches Search

558 Commits (3b1f5f07ce37668716d7034ca3549bd835beefb5)

All Branches