[Misc] Remove flashinfer warning, add flashinfer tests to CI (#6351)

This commit is contained in:
Lily Liu
2024-07-11 18:32:06 -07:00
committed by GitHub
parent 7ed6a4f0e1
commit d6ab528997
3 changed files with 10 additions and 6 deletions

View File

@@ -2,11 +2,13 @@
Run `pytest tests/basic_correctness/test_basic_correctness.py`.
"""
import os
import weakref
import pytest
from vllm import LLM
from vllm.utils import is_hip
from ..models.utils import check_outputs_equal
@@ -26,6 +28,9 @@ def test_vllm_gc_ed():
assert weak_llm() is None
@pytest.mark.skipif(is_hip()
and os.getenv("VLLM_ATTENTION_BACKEND") == "FLASHINFER",
reason="Flashinfer does not support ROCm/HIP.")
@pytest.mark.parametrize("model", MODELS)
@pytest.mark.parametrize("dtype", ["half"])
@pytest.mark.parametrize("max_tokens", [5])