From 747a739a812d91ed599d78709e5a27b8f01d294a Mon Sep 17 00:00:00 2001 From: Will Lin Date: Tue, 10 Sep 2024 16:55:58 -0700 Subject: [PATCH] add flashinfer to tests --- tests/multi_step/test_correctness_async_llm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/multi_step/test_correctness_async_llm.py b/tests/multi_step/test_correctness_async_llm.py index d4308e8c1207c..754636f2744e2 100644 --- a/tests/multi_step/test_correctness_async_llm.py +++ b/tests/multi_step/test_correctness_async_llm.py @@ -35,7 +35,7 @@ @pytest.mark.parametrize("num_scheduler_steps", NUM_SCHEDULER_STEPS) @pytest.mark.parametrize("num_prompts", NUM_PROMPTS) @pytest.mark.parametrize("num_logprobs", [None]) -@pytest.mark.parametrize("is_async", [False, True]) +@pytest.mark.parametrize("is_async", [True]) @pytest.mark.parametrize("attention_backend", ["FLASHINFER", "FLASH_ATTN"]) @pytest.mark.asyncio async def test_multi_step(