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
アマデウス 297b8baae2
[model checkpoint] add gloo groups for cpu tensor communication (#589)
3 years ago
..
process_group_initializer [model checkpoint] add gloo groups for cpu tensor communication (#589) 3 years ago
random
__init__.py
config.py
moe_context.py
parallel_context.py [model checkpoint] add gloo groups for cpu tensor communication (#589) 3 years ago
parallel_mode.py
singleton_meta.py