Unverified Commit bab438ff authored by Ryan Rock's avatar Ryan Rock Committed by GitHub
Browse files

[CI/Build] Skip ray tests on ROCm (#29556)


Signed-off-by: default avatarRyan Rock <ryan.rock@amd.com>
parent 882851dc
...@@ -12,6 +12,7 @@ from vllm import SamplingParams ...@@ -12,6 +12,7 @@ from vllm import SamplingParams
from vllm.config import VllmConfig from vllm.config import VllmConfig
from vllm.engine.arg_utils import AsyncEngineArgs from vllm.engine.arg_utils import AsyncEngineArgs
from vllm.inputs import PromptType from vllm.inputs import PromptType
from vllm.platforms import current_platform
from vllm.sampling_params import RequestOutputKind from vllm.sampling_params import RequestOutputKind
from vllm.v1.engine.async_llm import AsyncLLM from vllm.v1.engine.async_llm import AsyncLLM
from vllm.v1.engine.core_client import DPAsyncMPClient from vllm.v1.engine.core_client import DPAsyncMPClient
...@@ -84,6 +85,10 @@ async def test_load( ...@@ -84,6 +85,10 @@ async def test_load(
if async_scheduling and data_parallel_backend == "ray": if async_scheduling and data_parallel_backend == "ray":
# TODO(NickLucche) Re-enable when async scheduling is supported # TODO(NickLucche) Re-enable when async scheduling is supported
pytest.skip("Async scheduling is not supported with ray") pytest.skip("Async scheduling is not supported with ray")
elif data_parallel_backend == "ray" and current_platform.is_rocm():
pytest.skip(
"Ray as the distributed executor backend is not supported with ROCm."
)
stats_loggers = {} stats_loggers = {}
@dataclass @dataclass
......
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