Unverified Commit 7362b445 authored by Li, Jiang's avatar Li, Jiang Committed by GitHub
Browse files

[Bugfix] Avoid LD_PRELOAD check on MacOS (#37145)


Signed-off-by: default avatarjiang1.li <jiang1.li@intel.com>
parent 57a314d1
...@@ -2,6 +2,7 @@ ...@@ -2,6 +2,7 @@
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project # SPDX-FileCopyrightText: Copyright contributors to the vLLM project
import os import os
import platform import platform
import sys
from collections.abc import Callable from collections.abc import Callable
from typing import Any from typing import Any
...@@ -63,9 +64,10 @@ class CPUWorker(Worker): ...@@ -63,9 +64,10 @@ class CPUWorker(Worker):
"to setup required pre-loaded libraries." "to setup required pre-loaded libraries."
) )
check_preloaded_libs("libtcmalloc") if sys.platform.startswith("linux"):
if current_platform.get_cpu_architecture() == CpuArchEnum.X86: check_preloaded_libs("libtcmalloc")
check_preloaded_libs("libiomp") if current_platform.get_cpu_architecture() == CpuArchEnum.X86:
check_preloaded_libs("libiomp")
# Setup OpenMP threads affinity. # Setup OpenMP threads affinity.
omp_cpuids = envs.VLLM_CPU_OMP_THREADS_BIND omp_cpuids = envs.VLLM_CPU_OMP_THREADS_BIND
......
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