Skip to content

GitLab

Unverified Commit 6247bae6 authored by Michael Goin's avatar Michael Goin Committed by GitHub
Browse files

[Bugfix] Restrict MacOS CPU detection (#14210) 


Signed-off-by: default avatarmgoin <mgoin64@gmail.com>
parent 3610fb49
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
vllm/platforms/__init__.py
View file @ 6247bae6
...@@ -160,11 +160,11 @@ def cpu_platform_plugin() -> Optional[str]: ...@@ -160,11 +160,11 @@ def cpu_platform_plugin() -> Optional[str]:
logger.debug("Confirmed CPU platform is available because" logger.debug("Confirmed CPU platform is available because"
" vLLM is built with CPU.") " vLLM is built with CPU.")
if not is_cpu: if not is_cpu:
import platform import sys
is_cpu = platform.machine().lower().startswith("arm") is_cpu = sys.platform.startswith("darwin")
if is_cpu: if is_cpu:
logger.debug("Confirmed CPU platform is available" logger.debug("Confirmed CPU platform is available"
" because the machine is ARM.") " because the machine is MacOS.")
except Exception as e: except Exception as e:
logger.debug("CPU platform is not available because: %s", str(e)) logger.debug("CPU platform is not available because: %s", str(e))
......
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