ColossalAI/colossalai/context/process_group_initializer
Hongxin Liu ac178ca5c1 [legacy] move builder and registry to legacy (#4603) 2023-09-05 21:53:10 +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 [legacy] move builder and registry to legacy (#4603) 2023-09-05 21:53:10 +08:00
initializer_2d.py [legacy] move builder and registry to legacy (#4603) 2023-09-05 21:53:10 +08:00
initializer_2p5d.py [legacy] move builder and registry to legacy (#4603) 2023-09-05 21:53:10 +08:00
initializer_3d.py [legacy] move builder and registry to legacy (#4603) 2023-09-05 21:53:10 +08:00
initializer_data.py [legacy] move builder and registry to legacy (#4603) 2023-09-05 21:53:10 +08:00
initializer_model.py [legacy] move builder and registry to legacy (#4603) 2023-09-05 21:53:10 +08:00
initializer_pipeline.py [legacy] move builder and registry to legacy (#4603) 2023-09-05 21:53:10 +08:00
initializer_sequence.py [legacy] move builder and registry to legacy (#4603) 2023-09-05 21:53:10 +08:00
initializer_tensor.py [legacy] move builder and registry to legacy (#4603) 2023-09-05 21:53:10 +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