ColossalAI/colossalai
Frank Lee ea74a3b9cc
[cli] updated installation cheheck with more inforamtion (#2050)
* [cli] updated installation cheheck with more inforamtion

* polish code

* polish code
2022-11-30 17:53:55 +08:00
..
_C
amp
auto_parallel [autoparallel] add split handler (#2032) 2022-11-29 11:03:51 +08:00
builder
cli [cli] updated installation cheheck with more inforamtion (#2050) 2022-11-30 17:53:55 +08:00
communication
context
device [autoparallel] mix gather (#1977) 2022-11-23 21:49:17 +08:00
engine
fx [rpc] split with dag (#2028) 2022-11-29 11:36:28 +08:00
gemini [Gemini] free and allocate cuda memory by tensor.storage, add grad hook (#2040) 2022-11-30 15:57:45 +08:00
kernel
logging
nn [gemini] add arguments (#2046) 2022-11-30 16:40:13 +08:00
pipeline [rpc] split with dag (#2028) 2022-11-29 11:36:28 +08:00
registry
tensor [autoparallel] add experimental permute handler (#2029) 2022-11-27 20:26:52 +08:00
testing [zero] test gradient accumulation (#1964) 2022-11-29 13:00:30 +08:00
trainer
utils [gemini] fix init bugs for modules (#2047) 2022-11-30 17:06:10 +08:00
zero [zero] test gradient accumulation (#1964) 2022-11-29 13:00:30 +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