hxwang cb01c0d5ce [moe] refactor mesh assignment

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

..

moe_utils.py

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

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

test_deepseek_layer.py

[moe] refactor mesh assignment

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

test_kernel.py

[moe] full test for deepseek and mixtral (pp + sp to fix)

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

test_mixtral_layer.py

[moe] refactor mesh assignment

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

test_moe_checkpoint.py

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

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

test_moe_ep_tp.py

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

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

test_moe_ep_zero.py

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

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