Skip to content

GitLab

Unverified Commit 3a4d4177 authored by Richard Zou's avatar Richard Zou Committed by GitHub
Browse files

[Misc] Cleanup compilation tests (#19343) 


Signed-off-by: default avatarrzou <zou3519@gmail.com>
parent 8335667c
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 9 deletions
tests/compile/test_config.py
View file @ 3a4d4177
......@@ -11,15 +11,6 @@ from vllm.config import (CompilationConfig, CompilationLevel, VllmConfig,
from .piecewise.test_simple import SillyModel
@pytest.fixture(scope="function", autouse=True)
def use_v1(monkeypatch):
"""
TODO(rzou): The rest of tests/compile runs VLLM_USE_V1=0 right now,
I'll switch them over later.
"""
monkeypatch.setenv('VLLM_USE_V1', '1')
@pytest.mark.parametrize("enabled", [True, False])
def test_use_cudagraphs(enabled):
assert vllm.envs.VLLM_USE_V1
......
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