ColossalAI/colossalai/context/process_group_initializer
Liang Bowen ec5086c49c Refactored docstring to google style 2022-03-29 17:17:47 +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 Refactored docstring to google style 2022-03-29 17:17:47 +08:00
initializer_2d.py Refactored docstring to google style 2022-03-29 17:17:47 +08:00
initializer_2p5d.py Refactored docstring to google style 2022-03-29 17:17:47 +08:00
initializer_3d.py Refactored docstring to google style 2022-03-29 17:17:47 +08:00
initializer_data.py Refactored docstring to google style 2022-03-29 17:17:47 +08:00
initializer_model.py Refactored docstring to google style 2022-03-29 17:17:47 +08:00
initializer_pipeline.py Refactored docstring to google style 2022-03-29 17:17:47 +08:00
initializer_sequence.py Refactored docstring to google style 2022-03-29 17:17:47 +08:00
initializer_tensor.py Refactored docstring to google style 2022-03-29 17:17:47 +08:00
process_group_initializer.py Refactored docstring to google style 2022-03-29 17:17:47 +08:00