ColossalAI/tests/test_moe
hxwang 05a78d2f41
[chore] solve moe ckpt test failure and some other arg pass failure
2024-07-22 03:53:02 +00:00
..
moe_utils.py [chore] solve moe ckpt test failure and some other arg pass failure 2024-07-22 03:53:02 +00:00
test_deepseek_layer.py [chore] solve moe ckpt test failure and some other arg pass failure 2024-07-22 03:53:02 +00:00
test_kernel.py [moe] full test for deepseek and mixtral (pp + sp to fix) 2024-07-19 07:32:56 +00:00
test_mixtral_layer.py [chore] solve moe ckpt test failure and some other arg pass failure 2024-07-22 03:53:02 +00:00
test_moe_checkpoint.py [chore] solve moe ckpt test failure and some other arg pass failure 2024-07-22 03:53:02 +00:00
test_moe_ep_tp.py [chore] solve moe ckpt test failure and some other arg pass failure 2024-07-22 03:53:02 +00:00
test_moe_ep_zero.py [chore] solve moe ckpt test failure and some other arg pass failure 2024-07-22 03:53:02 +00:00