ColossalAI/tests/test_moe/test_moe_colo_init.py at e99edfcb51df48dec17498c60bbbd06baa293c22 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-12-02 19:56:09 +00:00

Files

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

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

2.1 KiB

Raw Blame History

View Raw