Skip to content

Commit 08d81f1

Browse files
varun-sundar-rabindranathVarun Sundar Rabindranath
andauthored
[Bugfix] Fix deepep tests (vllm-project#20288)
Signed-off-by: Varun Sundar Rabindranath <vsundarr@redhat.com> Co-authored-by: Varun Sundar Rabindranath <vsundarr@redhat.com>
1 parent 6cc1e7d commit 08d81f1

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

tests/kernels/moe/test_deepep_deepgemm_moe.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@
3030
from vllm.model_executor.layers.fused_moe.deepep_ll_prepare_finalize import ( # noqa: E501
3131
DeepEPLLPrepareAndFinalize)
3232

33-
from .deepep_utils import DeepEPHTArgs, DeepEPLLArgs, make_deepep_a2a
33+
from .utils import DeepEPHTArgs, DeepEPLLArgs, make_deepep_a2a
3434

3535
if has_deep_gemm():
3636
import deep_gemm

tests/kernels/moe/test_deepep_moe.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@
3131
from vllm.model_executor.layers.fused_moe.deepep_ll_prepare_finalize import ( # noqa: E501
3232
DeepEPLLPrepareAndFinalize)
3333

34-
from .deepep_utils import DeepEPHTArgs, DeepEPLLArgs, make_deepep_a2a
34+
from .utils import DeepEPHTArgs, DeepEPLLArgs, make_deepep_a2a
3535

3636
requires_deep_ep = pytest.mark.skipif(
3737
not has_deep_ep(),

0 commit comments

Comments
 (0)