Unverified Commit fcd6ac97 authored by rasmith's avatar rasmith Committed by GitHub
Browse files

[CI][AMD][BugFix] Skip tests in test_unquantized_backend_selection that should...


[CI][AMD][BugFix] Skip tests in test_unquantized_backend_selection that should not run on ROCm (#34655)
Signed-off-by: default avatarRandall Smith <Randall.Smith@amd.com>
parent 95be2a7f
...@@ -9,6 +9,7 @@ from vllm.model_executor.layers.fused_moe.oracle.unquantized import ( ...@@ -9,6 +9,7 @@ from vllm.model_executor.layers.fused_moe.oracle.unquantized import (
UnquantizedMoeBackend, UnquantizedMoeBackend,
select_unquantized_moe_backend, select_unquantized_moe_backend,
) )
from vllm.platforms import current_platform
@pytest.mark.parametrize( @pytest.mark.parametrize(
...@@ -65,6 +66,9 @@ def test_select_default_backend_by_platform( ...@@ -65,6 +66,9 @@ def test_select_default_backend_by_platform(
"vllm.model_executor.layers.fused_moe.oracle.unquantized.is_supported_config_trtllm_bf16", "vllm.model_executor.layers.fused_moe.oracle.unquantized.is_supported_config_trtllm_bf16",
return_value=(True, None), return_value=(True, None),
) )
@pytest.mark.skipif(
not current_platform.is_cuda(), reason="Only supported on NVIDIA platforms."
)
def test_select_cuda_flashinfer_trtllm_backend( def test_select_cuda_flashinfer_trtllm_backend(
mock_has_flashinfer, mock_is_supported_trtllm, monkeypatch mock_has_flashinfer, mock_is_supported_trtllm, monkeypatch
): ):
...@@ -101,6 +105,9 @@ def test_select_cuda_flashinfer_trtllm_backend( ...@@ -101,6 +105,9 @@ def test_select_cuda_flashinfer_trtllm_backend(
"vllm.model_executor.layers.fused_moe.oracle.unquantized.is_supported_config_trtllm_bf16", "vllm.model_executor.layers.fused_moe.oracle.unquantized.is_supported_config_trtllm_bf16",
return_value=(False, None), return_value=(False, None),
) )
@pytest.mark.skipif(
not current_platform.is_cuda(), reason="Only supported on NVIDIA platforms."
)
def test_select_cuda_flashinfer_cutlass_backend( def test_select_cuda_flashinfer_cutlass_backend(
mock_has_flashinfer, mock_is_supported_trtllm, monkeypatch mock_has_flashinfer, mock_is_supported_trtllm, monkeypatch
): ):
......
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