ColossalAI/colossalai
YuliangLiu0306 4b03c25f85
[tensor]add 1D device mesh (#1492)
2022-08-25 16:48:12 +08:00
..
amp
auto_parallel [autoparallel] added namespace constraints (#1490) 2022-08-24 15:44:07 +08:00
builder
cli
communication
context
device [tensor]add 1D device mesh (#1492) 2022-08-25 16:48:12 +08:00
engine
fx [fx] fixed adapative pooling size concatenation error (#1489) 2022-08-25 09:05:07 +08:00
gemini
kernel
logging
nn [FAW] LFU cache for the FAW 2022-08-25 13:08:46 +08:00
pipeline [pipeline/rpc] implement distributed optimizer | test with assert_close (#1486) 2022-08-25 10:49:01 +08:00
registry
tensor [tensor]add 1D device mesh (#1492) 2022-08-25 16:48:12 +08:00
testing
trainer
utils
zero
__init__.py
constants.py
core.py
global_variables.py
initialize.py