ColossalAI/colossalai
YuliangLiu0306 e34e850a4c
[autoparallel]add essential CommActions for broadcast oprands (#1793)
2022-11-04 18:36:42 +08:00
..
amp
auto_parallel [autoparallel]add essential CommActions for broadcast oprands (#1793) 2022-11-04 18:36:42 +08:00
builder
cli
communication
context
device
engine
fx [autoparallel]add essential CommActions for broadcast oprands (#1793) 2022-11-04 18:36:42 +08:00
gemini [hotfix] fix zero's incompatibility with checkpoint in torch-1.12 (#1786) 2022-11-02 16:11:34 +08:00
kernel
logging
nn
pipeline
registry
tensor
testing [unittest] added doc for the pytest wrapper (#1704) 2022-10-14 10:56:17 +08:00
trainer [NFC] polish _checkpoint_hook.py code style (#1722) 2022-10-19 12:20:51 +08:00
utils
zero
__init__.py
constants.py
core.py
global_variables.py
initialize.py