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/context
kurisusnowdeng 0b8161fab8
updated tp layers
2 years ago
..
process_group_initializer updated tp layers 2 years ago
random [doc] update rst and docstring (#1351) 2 years ago
__init__.py
config.py
moe_context.py [zero] add constant placement policy (#1705) 2 years ago
parallel_context.py [usability] improved error messages in the context module (#856) 3 years ago
parallel_mode.py updated tp layers 2 years ago
singleton_meta.py [autoparallel] refactored shape consistency to remove redundancy (#1591) 2 years ago