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
Jiarui Fang cb5a4778e1
Revert "[WIP] Applying ColoTensor on TP-1D-row Linear. (#831)" (#835)
3 years ago
..
amp
builder
cli [cli] added check installation cli 3 years ago
communication
context
engine
gemini [tensor] reorganize files (#820) 3 years ago
kernel
logging
nn
registry
tensor Revert "[WIP] Applying ColoTensor on TP-1D-row Linear. (#831)" (#835) 3 years ago
testing
trainer
utils [refactor] moving InsertPostInitMethodToModuleSubClasses to utils. (#824) 3 years ago
zero revert zero tensors back (#829) 3 years ago
__init__.py
constants.py
core.py
global_variables.py
initialize.py