xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-05-20 14:57:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Misc] Remove ignore_reinit_error for ray.init() (#15373)

Browse Source
This commit is contained in:
Rui Qiao 2025-03-24 00:41:53 -07:00 committed by GitHub
parent 3892e58ad7
commit 5797fb97e9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
vllm/executor/ray_utils.py
View File

@ -289,16 +289,14 @@ def initialize_ray_cluster(
elif current_platform.is_rocm() or current_platform.is_xpu(): elif current_platform.is_rocm() or current_platform.is_xpu():
# Try to connect existing ray instance and create a new one if not found # Try to connect existing ray instance and create a new one if not found
try: try:
ray.init("auto", ignore_reinit_error=True) ray.init("auto")
except ConnectionError: except ConnectionError:
logger.warning( logger.warning(
"No existing RAY instance detected. " "No existing RAY instance detected. "
"A new instance will be launched with current node resources.") "A new instance will be launched with current node resources.")
ray.init(address=ray_address, ray.init(address=ray_address, num_gpus=parallel_config.world_size)
ignore_reinit_error=True,
num_gpus=parallel_config.world_size)
else: else:
ray.init(address=ray_address, ignore_reinit_error=True) ray.init(address=ray_address)
device_str = current_platform.ray_device_key device_str = current_platform.ray_device_key
if not device_str: if not device_str: