You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai
YuliangLiu0306 942c8cd1fb
[fx] refactor tracer to trace complete graph (#1342)
2 years ago
..
amp [hotfix]different overflow status lead to communication stuck. (#1175) 2 years ago
builder [NFC] polish colossalai/builder/builder.py code style (#1265) 2 years ago
cli
communication [NFC] polish colossalai/communication/collective.py (#1262) 2 years ago
context
engine [hotfix] fix PipelineSharedModuleGradientHandler (#1314) 2 years ago
fx [fx] refactor tracer to trace complete graph (#1342) 2 years ago
gemini [hotfix] ZeroDDP use new process group (#1333) 2 years ago
kernel Recover kernal files 2 years ago
logging
nn [hotfix] ZeroDDP use new process group (#1333) 2 years ago
pipeline
registry
tensor [checkpoint] use gather_tensor in checkpoint and update its unit test (#1339) 2 years ago
testing
trainer
utils [fx] tested the complete workflow for auto-parallel (#1336) 2 years ago
zero [hotfix] shared model returns cpu state_dict (#1328) 2 years ago
__init__.py [NFC] polish colossalai/__init__.py code style (#1285) 2 years ago
constants.py
core.py [Tensor] distributed view supports inter-process hybrid parallel (#1169) 2 years ago
global_variables.py
initialize.py [hotfix] remove potiential circle import (#1307) 2 years ago