BlueRum b3f73ce1c8 [Gemini] Update coloinit_ctx to support meta_tensor (#2147)		2022-12-19 22:37:07 +08:00
..
_C	[optimizer] add div_scale for optimizers (#2117)	2022-12-12 17:58:57 +08:00
amp	[kernel] move all symlinks of kernel to `colossalai._C` (#1971)	2022-11-17 13:42:33 +08:00
auto_parallel	[autoparallel] process size nodes in runtime pass (#2130)	2022-12-14 16:10:50 +08:00
builder	…
cli	[cli] updated installation cheheck with more inforamtion (#2050)	2022-11-30 17:53:55 +08:00
communication	…
context	…
device	[device] update flatten device mesh usage (#2079)	2022-12-05 16:16:07 +08:00
engine	…
fx	register aten._convolution.default (#2137)	2022-12-18 19:27:01 +08:00
gemini	[Gemini] revert ZeROInitCtx related tracer (#2138)	2022-12-16 12:37:06 +08:00
kernel	updated attention kernel (#2133)	2022-12-16 10:54:03 +08:00
logging	…
nn	[Gemini] chunk init using runtime visited param order (#2115)	2022-12-12 18:06:16 +08:00
pipeline	[PP Middleware] Add bwd and step for PP middleware (#2111)	2022-12-12 12:40:03 +08:00
registry	…
tensor	[NFC] polish comments for Chunk class (#2116)	2022-12-12 15:39:31 +08:00
testing	[zero] test gradient accumulation (#1964)	2022-11-29 13:00:30 +08:00
trainer	…
utils	[Gemini] Update coloinit_ctx to support meta_tensor (#2147)	2022-12-19 22:37:07 +08:00
zero	[Gemini] update API of the chunkmemstatscollector. (#2129)	2022-12-14 00:47:06 +08:00
__init__.py	[setup] supported conda-installed torch (#2048)	2022-11-30 16:45:15 +08:00
constants.py	…
core.py	…
global_variables.py	…
initialize.py	…