diff --git a/tests/unit_tests/transformer/test_cuda_graphs.py b/tests/unit_tests/transformer/test_cuda_graphs.py index d602346c370..a1e9dab269f 100644 --- a/tests/unit_tests/transformer/test_cuda_graphs.py +++ b/tests/unit_tests/transformer/test_cuda_graphs.py @@ -1252,10 +1252,6 @@ def test_moe_partial_cudagraph(self, ep_size, moe_dropless_dispatcher, moe_dispa extra_kwargs["moe_token_dispatcher_type"] = "flex" extra_kwargs["moe_flex_dispatcher_backend"] = "deepep" elif moe_dispatcher_type == "hybridep": - pytest.skip( - "Currently, the Hybrid EP is broken. " - "Temporarily skip the test and wait for the fix." - ) if not is_hybrid_ep_available(): pytest.skip("Hybrid EP is not available") extra_kwargs["moe_token_dispatcher_type"] = "flex" @@ -1265,8 +1261,6 @@ def test_moe_partial_cudagraph(self, ep_size, moe_dropless_dispatcher, moe_dispa if not moe_dropless_dispatcher: if moe_dispatcher_type == "deepep": pytest.skip("Deep EP doesn't support drop&pad MoE") - if moe_dispatcher_type == "hybridep" and ep_size == 1: - pytest.skip("Hybrid EP doesn't support drop&pad MoE with ep_size == 1") extra_kwargs["moe_expert_capacity_factor"] = 1.0 extra_kwargs["moe_pad_expert_input_to_capacity"] = True