mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-12 21:25:42 +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.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(
|
|||||||
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
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user