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

2024-07-19 07:32:56 +00:00

..

moe_utils.py

[moe] clean legacy code

2024-07-19 07:32:01 +00:00

test_deepseek_layer.py

[shardformer] DeepseekMoE support (#5871)

2024-07-05 16:13:58 +08:00

test_kernel.py

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

2024-07-19 07:32:56 +00:00

test_moe_checkpoint.py

[moe] init moe plugin comm setting with sp

2024-07-19 07:32:54 +00:00

test_moe_ep_tp.py

[moe] test deepseek

2024-07-19 07:32:00 +00:00

test_moe_ep_zero.py

[moe] test deepseek

2024-07-19 07:32:00 +00:00