ColossalAI/colossalai/context/process_group_initializer at a30e2b4c24ec0de82206b6f2943df96cc80d7cad - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-15 23:23:11 +00:00

Files

History

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