ColossalAI/colossalai/context/process_group_initializer
Zirui Zhu c9e3ee389e
[NFC] polish colossalai/context/process_group_initializer/initializer_2d.py code style (#2726)
2023-02-15 22:27:13 +08:00
..
__init__.py [MOE] add unitest for MOE experts layout, gradient handler and kernel (#469) 2022-03-21 13:35:04 +08:00
initializer_1d.py [model checkpoint] add gloo groups for cpu tensor communication (#589) 2022-04-01 10:15:52 +08:00
initializer_2d.py [NFC] polish colossalai/context/process_group_initializer/initializer_2d.py code style (#2726) 2023-02-15 22:27:13 +08:00
initializer_2p5d.py [usability] improved error messages in the context module (#856) 2022-04-25 13:42:31 +08:00
initializer_3d.py updated tp layers 2022-11-02 12:19:38 +08:00
initializer_data.py [NFC] polish <colossalai/context/process_group_initializer/initializer_data.py> code stype (#626) 2022-04-06 11:40:59 +08:00
initializer_model.py [model checkpoint] add gloo groups for cpu tensor communication (#589) 2022-04-01 10:15:52 +08:00
initializer_pipeline.py [NFC] polish pipeline process group code style (#2694) 2023-02-14 18:12:01 +08:00
initializer_sequence.py [NFC] posh colossalai/context/process_group_initializer/initializer_sequence.py code style (#2712) 2023-02-15 10:53:38 +08:00
initializer_tensor.py [NFC] polish colossalai/context/process_group_initializer/initializer_sequence.py colossalai/context/process_group_initializer initializer_tensor.py code style (#639) 2022-04-06 11:40:59 +08:00
process_group_initializer.py [NFC] polish colossalai/context/process_group_initializer/process_group_initializer.py code stype (#617) 2022-04-06 11:40:59 +08:00