ColossalAI/colossalai/context
Arsmart1 8af977f223 [NFC] polish colossalai/context/parallel_context.py code style (#3276) 2023-03-29 15:22:21 +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 [doc] update rst and docstring (#1351) 2022-07-21 15:54:53 +08:00
__init__.py [polish] polish singleton and global context (#500) 2022-03-23 18:03:39 +08:00
config.py Refactored docstring to google style 2022-03-29 17:17:47 +08:00
moe_context.py [NFC] polish colossalai/context/moe_context.py code style (#2693) 2023-02-14 18:02:45 +08:00
parallel_context.py [NFC] polish colossalai/context/parallel_context.py code style (#3276) 2023-03-29 15:22:21 +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