[CI/Build] Use file rendezvous for UniProc loader fixtures (#52367)
Signed-off-by: yu-xin-c <2182712990@qq.com>
Stellar鱼 committed
Aug 29, 2026 at 11:53 UTC
5182f2705be914bd986c3f2ba68737be78b3b8da
2 files changed
+4
-4
tests/model_executor/model_loader/runai_streamer_loader/conftest.py
+2
-2
@@ -1,7 +1,7 @@
1
# SPDX-License-Identifier: Apache-2.0
2
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
3
4
-from vllm.utils.network_utils import get_distributed_init_method, get_ip, get_open_port
4
+from vllm.utils.network_utils import get_file_store_init_method
5
from vllm.v1.executor import UniProcExecutor
6
from vllm.v1.worker.worker_base import WorkerWrapperBase
7
@@ -11,7 +11,7 @@ from vllm.v1.worker.worker_base import WorkerWrapperBase
11
# The worker process reimports the patched entities, and the patch is not applied.
12
class RunaiDummyExecutor(UniProcExecutor):
13
def _init_executor(self) -> None:
14
- distributed_init_method = get_distributed_init_method(get_ip(), get_open_port())
14
+ distributed_init_method = get_file_store_init_method()
15
16
local_rank = 0
17
rank = 0
tests/model_executor/model_loader/tensorizer_loader/conftest.py
+2
-2
@@ -8,7 +8,7 @@ from vllm import LLM, EngineArgs
8
from vllm.distributed import cleanup_dist_env_and_memory
9
from vllm.model_executor.model_loader import tensorizer as tensorizer_mod
10
from vllm.model_executor.model_loader.tensorizer import TensorizerConfig
11
-from vllm.utils.network_utils import get_distributed_init_method, get_ip, get_open_port
11
+from vllm.utils.network_utils import get_file_store_init_method
12
from vllm.v1.executor import UniProcExecutor
13
from vllm.v1.worker.worker_base import WorkerWrapperBase
14
@@ -68,7 +68,7 @@ class DummyExecutor(UniProcExecutor):
68
def _init_executor(self) -> None:
69
"""Initialize the worker and load the model."""
70
self.driver_worker = WorkerWrapperBase(rpc_rank=0)
71
- distributed_init_method = get_distributed_init_method(get_ip(), get_open_port())
71
+ distributed_init_method = get_file_store_init_method()
72
local_rank = 0
73
# set local rank as the device index if specified
74
device_info = self.vllm_config.device_config.device.__str__().split(":")