Enable the qwen3 test (#21195)
Co-authored-by: Shunkang <182541032+Shunkangz@users.noreply.github.co>
This commit is contained in:
@@ -33,6 +33,7 @@ from sglang.srt.distributed import (
|
|||||||
get_moe_tensor_parallel_world_size,
|
get_moe_tensor_parallel_world_size,
|
||||||
get_pp_group,
|
get_pp_group,
|
||||||
get_tensor_model_parallel_rank,
|
get_tensor_model_parallel_rank,
|
||||||
|
moe_expert_parallel_all_reduce,
|
||||||
moe_tensor_model_parallel_all_reduce,
|
moe_tensor_model_parallel_all_reduce,
|
||||||
)
|
)
|
||||||
from sglang.srt.eplb.expert_distribution import get_global_expert_distribution_recorder
|
from sglang.srt.eplb.expert_distribution import get_global_expert_distribution_recorder
|
||||||
@@ -321,6 +322,9 @@ class Qwen3MoeSparseMoeBlock(nn.Module):
|
|||||||
topk_output = self.topk(hidden_states, router_logits)
|
topk_output = self.topk(hidden_states, router_logits)
|
||||||
final_hidden_states = self.experts(hidden_states, topk_output)
|
final_hidden_states = self.experts(hidden_states, topk_output)
|
||||||
|
|
||||||
|
if self.ep_size > 1 and not should_allreduce_fusion:
|
||||||
|
final_hidden_states = moe_expert_parallel_all_reduce(final_hidden_states)
|
||||||
|
|
||||||
if (
|
if (
|
||||||
self.tp_size > 1
|
self.tp_size > 1
|
||||||
and not should_allreduce_fusion
|
and not should_allreduce_fusion
|
||||||
|
|||||||
@@ -11,11 +11,8 @@ from sglang.test.test_utils import (
|
|||||||
popen_launch_server,
|
popen_launch_server,
|
||||||
)
|
)
|
||||||
|
|
||||||
register_cuda_ci(
|
register_cuda_ci(est_time=300, suite="stage-c-test-4-gpu-h100")
|
||||||
est_time=300,
|
|
||||||
suite="stage-c-test-4-gpu-h100",
|
|
||||||
disabled="Temporarily disable the flaky test.",
|
|
||||||
)
|
|
||||||
QWEN3_30B_MODEL_PATH = "Qwen/Qwen3-30B-A3B-FP8"
|
QWEN3_30B_MODEL_PATH = "Qwen/Qwen3-30B-A3B-FP8"
|
||||||
|
|
||||||
GSM8K_BASELINE_ACCURACY = 0.85
|
GSM8K_BASELINE_ACCURACY = 0.85
|
||||||
|
|||||||
Reference in New Issue
Block a user