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.

ver217 cefc29ff06 [tensor] impl ColoDDP for ColoTensor (#1009) ... * impl ColoDDP for ColoTensor * polish code		3 years ago
..
amp	[doc] improved docstring in the amp module (#857)	3 years ago
builder	[NFC] polish colossalai/builder/pipeline.py code style (#951)	3 years ago
cli	[cli] remove unused imports (#1001)	3 years ago
communication	[doc] improved docstring in the communication module (#863)	3 years ago
context	[usability] improved error messages in the context module (#856)	3 years ago
engine	[pipelinable]use pipelinable to support GPT model. (#903)	3 years ago
gemini	[gemini] accelerate adjust_layout() (#878)	3 years ago
kernel	[NFC] polish colossalai/kernel/cuda_native/csrc/colossal_C_frontend.cpp code style	3 years ago
logging	[doc] improved docstring in the logging module (#861)	3 years ago
nn	[tensor] impl ColoDDP for ColoTensor (#1009)	3 years ago
registry	[usability] added assertion message in registry (#864)	3 years ago
tensor	[tensor] refactor parallel action (#1007)	3 years ago
testing	[test] added a decorator for address already in use error with backward compatibility (#760)	3 years ago
trainer	[doc] improved assertion messages in trainer (#873)	3 years ago
utils	[tensor] refactor colo-tensor (#992)	3 years ago
zero	[gemini] accelerate adjust_layout() (#878)	3 years ago
__init__.py	[NFC] polish __init__.py code style (#965)	3 years ago
constants.py	fix format constants.py (#358)	3 years ago
core.py	[polish] polish singleton and global context (#500)	3 years ago
global_variables.py	[MOE] add unitest for MOE experts layout, gradient handler and kernel (#469)	3 years ago
initialize.py	[doc] improved error messages in initialize (#872)	3 years ago