ColossalAI/tests/test_legacy/test_moe at grpo-zero-bubble-rebase - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-15 15:29:48 +00:00

Files

History

hxwang 70c9924d0d [chore] solve moe ckpt test failure and some other arg pass failure

2024-08-01 10:06:59 +08:00

..

moe_utils.py

[moe] clean legacy code

2024-08-01 10:06:59 +08:00

test_grad_handler.py

[moe] clean legacy code

2024-08-01 10:06:59 +08:00

test_moe_group.py

[moe] clean legacy code

2024-08-01 10:06:59 +08:00

test_moe_hybrid_zero.py

[moe] clean legacy code

2024-08-01 10:06:59 +08:00

test_moe_load_balance.py

[moe] clean legacy code

2024-08-01 10:06:59 +08:00