Jiarui Fang 49832b2344 [refactory] add nn.parallel module (#1068)		2022-06-06 15:34:41 +08:00
..
amp	…
builder	[NFC] polish colossalai/builder/pipeline.py code style (#951)	2022-05-17 10:25:06 +08:00
cli	[hotfix] fix some bugs caused by size mismatch. (#1011)	2022-05-23 14:02:28 +08:00
communication	[pipeline]refactor ppschedule to support tensor list (#1050)	2022-06-02 13:48:59 +08:00
context	[usability] improved error messages in the context module (#856)	2022-04-25 13:42:31 +08:00
engine	[pipeline]refactor ppschedule to support tensor list (#1050)	2022-06-02 13:48:59 +08:00
gemini	[gemini] accelerate adjust_layout() (#878)	2022-04-26 18:08:31 +08:00
kernel	[NFC] polish colossalai/kernel/cuda_native/csrc/colossal_C_frontend.cpp code style	2022-05-20 23:57:38 +08:00
logging	…
nn	[refactory] add nn.parallel module (#1068)	2022-06-06 15:34:41 +08:00
registry	…
tensor	reorgnize colotensor directory (#1062)	2022-06-03 18:04:22 +08:00
testing	…
trainer	[doc] improved assertion messages in trainer (#873)	2022-04-26 10:00:12 +08:00
utils	[refactory] add nn.parallel module (#1068)	2022-06-06 15:34:41 +08:00
zero	[refactory] add nn.parallel module (#1068)	2022-06-06 15:34:41 +08:00
__init__.py	[NFC] polish __init__.py code style (#965)	2022-05-17 10:25:06 +08:00
constants.py	fix typo in constants (#1027)	2022-05-26 08:45:08 +08:00
core.py	…
global_variables.py	…
initialize.py	[cudnn] set False to cudnn benchmark by default (#1063)	2022-06-03 17:58:06 +08:00