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
HELSON abba4d84e1
[hotfix] fix bert model test in unitests (#1272)
2 years ago
..
amp [hotfix]different overflow status lead to communication stuck. (#1175) 2 years ago
builder
cli
communication [hotfix]fixed p2p process send stuck (#1181) 2 years ago
context
engine [hotfix] fix an assertion bug in base schedule. (#1250) 2 years ago
fx [fx] added ndim property to proxy (#1253) 2 years ago
gemini make AutoPlacementPolicy configurable (#1191) 2 years ago
kernel
logging
nn [hotfix] fix bert model test in unitests (#1272) 2 years ago
pipeline
registry
tensor hotfix colotensor _scan_for_pg_from_args (#1276) 2 years ago
testing
trainer
utils [tensor] distributed checkpointing for parameters (#1240) 2 years ago
zero [hotfix] fix sharded optim step and clip_grad_norm (#1226) 2 years ago
__init__.py
constants.py
core.py [Tensor] distributed view supports inter-process hybrid parallel (#1169) 2 years ago
global_variables.py
initialize.py