From 08d81f1014d174d4dd96518914c7ed9767c67a3f Mon Sep 17 00:00:00 2001 From: Varun Sundar Rabindranath Date: Tue, 1 Jul 2025 03:29:08 -0400 Subject: [PATCH] [Bugfix] Fix deepep tests (#20288) Signed-off-by: Varun Sundar Rabindranath Co-authored-by: Varun Sundar Rabindranath --- tests/kernels/moe/test_deepep_deepgemm_moe.py | 2 +- tests/kernels/moe/test_deepep_moe.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/kernels/moe/test_deepep_deepgemm_moe.py b/tests/kernels/moe/test_deepep_deepgemm_moe.py index 475427f439289..008406c3f1593 100644 --- a/tests/kernels/moe/test_deepep_deepgemm_moe.py +++ b/tests/kernels/moe/test_deepep_deepgemm_moe.py @@ -30,7 +30,7 @@ if has_deep_ep(): from vllm.model_executor.layers.fused_moe.deepep_ll_prepare_finalize import ( # noqa: E501 DeepEPLLPrepareAndFinalize) - from .deepep_utils import DeepEPHTArgs, DeepEPLLArgs, make_deepep_a2a + from .utils import DeepEPHTArgs, DeepEPLLArgs, make_deepep_a2a if has_deep_gemm(): import deep_gemm diff --git a/tests/kernels/moe/test_deepep_moe.py b/tests/kernels/moe/test_deepep_moe.py index 80a36dc39712a..94947c809e3a3 100644 --- a/tests/kernels/moe/test_deepep_moe.py +++ b/tests/kernels/moe/test_deepep_moe.py @@ -31,7 +31,7 @@ if has_deep_ep(): from vllm.model_executor.layers.fused_moe.deepep_ll_prepare_finalize import ( # noqa: E501 DeepEPLLPrepareAndFinalize) - from .deepep_utils import DeepEPHTArgs, DeepEPLLArgs, make_deepep_a2a + from .utils import DeepEPHTArgs, DeepEPLLArgs, make_deepep_a2a requires_deep_ep = pytest.mark.skipif( not has_deep_ep(),