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