7 Commits

Author	SHA1	Message	Date
Zirui Zhu	c9e3ee389e	[NFC] polish colossalai/context/process_group_initializer/initializer_2d.py code style (#2726)	2023-02-15 22:27:13 +08:00
アマデウス	297b8baae2	[model checkpoint] add gloo groups for cpu tensor communication (#589)	2022-04-01 10:15:52 +08:00
Liang Bowen	ec5086c49c	Refactored docstring to google style	2022-03-29 17:17:47 +08:00
Maruyama_Aya	e83970e3dc	fix format ColossalAI\colossalai\context\process_group_initializer	2022-03-11 15:50:28 +08:00
アマデウス	9ee197d0e9	moved env variables to global variables; (#215) ... added branch context; added vocab parallel layers; moved split_batch from load_batch to tensor parallel embedding layers; updated gpt model; updated unit test cases; fixed few collective communicator bugs	2022-02-15 11:31:13 +08:00
HELSON	0f8c7f9804	Fixed docstring in colossalai (#171)	2022-01-21 10:44:30 +08:00
zbian	404ecbdcc6	Migrated project	2021-10-28 18:21:23 +02:00