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 8e14344ec9
[hotfix] fix a type in ColoInitContext (#2106)
2 years ago
..
_C
amp
auto_parallel [autoparallel] add sum handler (#2101) 2 years ago
builder
cli
communication
context
device
engine
fx [autoparallel] support linear function bias addition (#2104) 2 years ago
gemini [Gemini] gemini use the runtime memory tracer (RMT) (#2099) 2 years ago
kernel
logging
nn
pipeline [Pipeline Middleware] fix data race in Pipeline Scheduler for DAG (#2087) 2 years ago
registry
tensor
testing
trainer
utils [hotfix] fix a type in ColoInitContext (#2106) 2 years ago
zero [Gemini] use MemStats to store the tracing data. Seperate it from Collector. (#2084) 2 years ago
__init__.py
constants.py
core.py
global_variables.py
initialize.py