Skip to content

GitLab

"tests/entrypoints/pooling/embed/test_online_vision.py" did not exist on "2f652e6cdf09da407e78a60327832f913b32e26e"
Unverified Commit 08d81f10 authored by Varun Sundar Rabindranath's avatar Varun Sundar Rabindranath Committed by GitHub
Browse files

[Bugfix] Fix deepep tests (#20288) 


Signed-off-by: default avatarVarun Sundar Rabindranath <vsundarr@redhat.com>
Co-authored-by: default avatarVarun Sundar Rabindranath <vsundarr@redhat.com>
parent 6cc1e7d9
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/kernels/moe/test_deepep_deepgemm_moe.py
View file @ 08d81f10
......@@ -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
......
tests/kernels/moe/test_deepep_moe.py
View file @ 08d81f10
......@@ -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(),
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment