xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-05-27 07:34:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

[AMD][CI] fix v1/engine test_preprocess_error_handling (#31192)

Browse Source 
Signed-off-by: Divakar Verma <divakar.verma@amd.com>
This commit is contained in:
Divakar Verma 2025-12-22 19:28:19 -06:00 committed by GitHub
parent b57b967386
commit 78e5e62bbf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/v1/engine/test_preprocess_error_handling.py
View File

@ -5,6 +5,7 @@ import pytest
import torch.cuda import torch.cuda
from vllm import LLM, SamplingParams from vllm import LLM, SamplingParams
from vllm.platforms import current_platform
from vllm.v1.engine import EngineCoreRequest from vllm.v1.engine import EngineCoreRequest
from vllm.v1.engine.core import EngineCore from vllm.v1.engine.core import EngineCore
@ -14,6 +15,11 @@ MODEL_NAME = "hmellor/tiny-random-LlamaForCausalLM"
def test_preprocess_error_handling(monkeypatch: pytest.MonkeyPatch): def test_preprocess_error_handling(monkeypatch: pytest.MonkeyPatch):
"""Test that preprocessing errors are handled gracefully.""" """Test that preprocessing errors are handled gracefully."""
if current_platform.is_rocm():
pytest.skip(
"Skipped on ROCm: this test only works with 'fork', but ROCm uses 'spawn'."
)
assert not torch.cuda.is_initialized(), ( assert not torch.cuda.is_initialized(), (
"fork needs to be used for the engine " "fork needs to be used for the engine "
"core process and this isn't possible if cuda is already initialized" "core process and this isn't possible if cuda is already initialized"