ColossalAI/colossalai/context
Zirui Zhu c9e3ee389e
[NFC] polish colossalai/context/process_group_initializer/initializer_2d.py code style (#2726)
2023-02-15 22:27:13 +08:00
..
process_group_initializer [NFC] polish colossalai/context/process_group_initializer/initializer_2d.py code style (#2726) 2023-02-15 22:27:13 +08:00
random
__init__.py
config.py
moe_context.py [NFC] polish colossalai/context/moe_context.py code style (#2693) 2023-02-14 18:02:45 +08:00
parallel_context.py Revert "Update parallel_context.py (#2408)" 2023-01-19 12:27:48 +08:00
parallel_mode.py updated tp layers 2022-11-02 12:19:38 +08:00
singleton_meta.py [autoparallel] refactored shape consistency to remove redundancy (#1591) 2022-09-13 18:30:18 +08:00