Skip to content

GitLab

Unverified Commit b6a136b5 authored by Isotr0py's avatar Isotr0py Committed by GitHub
Browse files

[CI/Build] Fix disabled v1 attention backend selection test (#25471) 


Signed-off-by: default avatarIsotr0py <mozf@mail2.sysu.edu.cn>
parent 0d9fe260
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
tests/kernels/attention/test_attention_selector.py
View file @ b6a136b5
...@@ -67,7 +67,6 @@ def generate_params(): ...@@ -67,7 +67,6 @@ def generate_params():
return params return params
@pytest.mark.skip(reason="Skipped for now. Should be revisited.")
@pytest.mark.parametrize("device, name, use_mla, block_size", @pytest.mark.parametrize("device, name, use_mla, block_size",
generate_params()) generate_params())
def test_env( def test_env(
...@@ -189,7 +188,7 @@ def test_env( ...@@ -189,7 +188,7 @@ def test_env(
# FlashMLA only supports block_size == 64 # FlashMLA only supports block_size == 64
pytest.skip("FlashMLA only supports block_size 64") pytest.skip("FlashMLA only supports block_size 64")
else: else:
from vllm.attention.backends.flashmla import ( from vllm.v1.attention.backends.mla.flashmla import ( # noqa: E501
is_flashmla_supported) is_flashmla_supported)
is_supported, _ = is_flashmla_supported() is_supported, _ = is_flashmla_supported()
if not is_supported: if not is_supported:
......
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