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
Tongping Liu ab54fed292
[hotfix] add kwargs for colo_addmm (#2171)
2 years ago
..
_C
amp
auto_parallel [autoparallel] memory estimation for shape consistency (#2144) 2 years ago
builder
cli
communication
context
device
engine
fx register meta func for rnn (#2159) 2 years ago
gemini [hotfix] fix auto policy of test_sharded_optim_v2 (#2157) 2 years ago
kernel
logging
nn [hotfix] add kwargs for colo_addmm (#2171) 2 years ago
pipeline
registry
tensor [autoparallel] memory estimation for shape consistency (#2144) 2 years ago
testing
trainer
utils
zero [example] add zero1, zero2 example in GPT examples (#2146) 2 years ago
__init__.py
constants.py
core.py
global_variables.py
initialize.py