From 6247bae6c69de73e2c8a9964d23357bef724ee16 Mon Sep 17 00:00:00 2001 From: Michael Goin Date: Tue, 4 Mar 2025 09:25:27 -0500 Subject: [PATCH] [Bugfix] Restrict MacOS CPU detection (#14210) Signed-off-by: mgoin --- vllm/platforms/__init__.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/vllm/platforms/__init__.py b/vllm/platforms/__init__.py index 89e69c7f5780d..74ef8bd1cff1a 100644 --- a/vllm/platforms/__init__.py +++ b/vllm/platforms/__init__.py @@ -160,11 +160,11 @@ def cpu_platform_plugin() -> Optional[str]: logger.debug("Confirmed CPU platform is available because" " vLLM is built with CPU.") if not is_cpu: - import platform - is_cpu = platform.machine().lower().startswith("arm") + import sys + is_cpu = sys.platform.startswith("darwin") if is_cpu: logger.debug("Confirmed CPU platform is available" - " because the machine is ARM.") + " because the machine is MacOS.") except Exception as e: logger.debug("CPU platform is not available because: %s", str(e))