HELSON 7544347145 [MOE] add unitest for MOE experts layout, gradient handler and kernel (#469)

2022-03-21 13:35:04 +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

fix format ColossalAI\colossalai\context\process_group_initializer

2022-03-11 15:50:28 +08:00

initializer_2d.py

fix format ColossalAI\colossalai\context\process_group_initializer

2022-03-11 15:50:28 +08:00

initializer_2p5d.py

fix format ColossalAI\colossalai\context\process_group_initializer

2022-03-11 15:50:28 +08:00

initializer_3d.py

fix format ColossalAI\colossalai\context\process_group_initializer

2022-03-11 15:50:28 +08:00

initializer_data.py

Fixed docstring in colossalai (#171)

2022-01-21 10:44:30 +08:00

initializer_model.py

fix format ColossalAI\colossalai\context\process_group_initializer

2022-03-11 15:50:28 +08:00

initializer_pipeline.py

Fixed docstring in colossalai (#171)

2022-01-21 10:44:30 +08:00

initializer_sequence.py

Fixed docstring in colossalai (#171)

2022-01-21 10:44:30 +08:00

initializer_tensor.py

Fixed docstring in colossalai (#171)

2022-01-21 10:44:30 +08:00

process_group_initializer.py

Fixed docstring in colossalai (#171)

2022-01-21 10:44:30 +08:00