mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-10 04:54:56 +08:00
[CI/Build] Skip ray tests on ROCm (#29556)
Signed-off-by: Ryan Rock <ryan.rock@amd.com>
This commit is contained in:
parent
882851dc81
commit
bab438ff3e
@ -12,6 +12,7 @@ from vllm import SamplingParams
|
||||
from vllm.config import VllmConfig
|
||||
from vllm.engine.arg_utils import AsyncEngineArgs
|
||||
from vllm.inputs import PromptType
|
||||
from vllm.platforms import current_platform
|
||||
from vllm.sampling_params import RequestOutputKind
|
||||
from vllm.v1.engine.async_llm import AsyncLLM
|
||||
from vllm.v1.engine.core_client import DPAsyncMPClient
|
||||
@ -84,6 +85,10 @@ async def test_load(
|
||||
if async_scheduling and data_parallel_backend == "ray":
|
||||
# TODO(NickLucche) Re-enable when async scheduling is supported
|
||||
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 = {}
|
||||
|
||||
@dataclass
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user