Skip to content

Commit 0e42b7d

Browse files
isikhiIsotr0py
authored andcommitted
fix: update platform detection for M-series arm based MacBook processors (vllm-project#12227)
Signed-off-by: isikhi <[email protected]> Signed-off-by: Isotr0py <[email protected]>
1 parent b31df2f commit 0e42b7d

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

vllm/platforms/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -101,6 +101,10 @@ def cpu_platform_plugin() -> Optional[str]:
101101
try:
102102
from importlib.metadata import version
103103
is_cpu = "cpu" in version("vllm")
104+
if is_cpu == False:
105+
import platform
106+
is_cpu = platform.machine().lower().startswith("arm")
107+
104108
except Exception:
105109
pass
106110

0 commit comments

Comments
 (0)