Skip to content

Commit 6f42e61

Browse files
committed
use eager mode by default in ut
Signed-off-by: MengqingCao <cmq0113@163.com>
1 parent 3c16d52 commit 6f42e61

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

tests/compile/test_aclgraph.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -96,5 +96,7 @@ def test_deepseek_raises_error(monkeypatch: pytest.MonkeyPatch) -> None:
9696
m.setenv("VLLM_USE_MODELSCOPE", "True")
9797
m.setenv("VLLM_USE_V1", "1")
9898
with pytest.raises(NotImplementedError) as excinfo:
99-
VllmRunner("deepseek-ai/DeepSeek-V2-Lite-Chat", max_model_len=1024)
99+
VllmRunner("deepseek-ai/DeepSeek-V2-Lite-Chat",
100+
max_model_len=1024,
101+
enforce_eager=False)
100102
assert "ACL Graph does not support deepseek" in str(excinfo.value)

tests/conftest.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,7 @@ def __init__(
7777
block_size: int = 16,
7878
enable_chunked_prefill: bool = False,
7979
swap_space: int = 4,
80-
enforce_eager: Optional[bool] = False,
80+
enforce_eager: Optional[bool] = True,
8181
**kwargs,
8282
) -> None:
8383
self.model = LLM(

0 commit comments

Comments
 (0)