ColossalAI/tests/test_moe at 0160a62a3c8f2b4bef5edd9997037fba69bf0da7 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-13 20:39:59 +00:00

Files

History

HELSON 95c35f73bd [moe] initialize MoE groups by ProcessGroup (#1640 )

2022-09-23 17:20:41 +08:00

..

test_grad_handler.py

…

test_kernel.py

…

test_moe_colo_init.py

[moe] initialize MoE groups by ProcessGroup (#1640 )

2022-09-23 17:20:41 +08:00

test_moe_group.py

…

test_moe_zero_init.py

…

test_moe_zero_model.py

…

test_moe_zero_optim.py

…