ColossalAI

Author	SHA1	Message	Date
zbian	e94c79f15b	improved allgather & reducescatter for 3d	2023-01-03 17:46:08 +08:00
kurisusnowdeng	0b8161fab8	updated tp layers	2022-11-02 12:19:38 +08:00
lucasliunju	912eb58ea0	[NFC] polish tests/test_layers/test_3d/checks_3d/common.py code style (#1733 )	2022-10-19 12:20:51 +08:00
Xue Fuzhao	754aa7c81f	[NFC] polish tests/test_layers/test_3d/checks_3d/check_layer_3d.py code style (#1731 )	2022-10-19 12:20:51 +08:00
zbian	3dba070580	fixed padding index issue for vocab parallel embedding layers; updated 3D linear to be compatible with examples in the tutorial	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
アマデウス	0fedef4f3c	Layer integration (#83 ) * integrated parallel layers for ease of building models * integrated 2.5d layers * cleaned codes and unit tests * added log metric by step hook; updated imagenet benchmark; fixed some bugs * reworked initialization; cleaned codes Co-authored-by: BoxiangW <45734921+BoxiangW@users.noreply.github.com>	2021-12-27 15:04:32 +08:00
Frank Lee	cd9c28e055	added CI for unit testing (#69 )	2021-12-16 10:32:08 +08:00