From 764441b7bd2a39b0fb51191b0906bdc05d1618f9 Mon Sep 17 00:00:00 2001
From: Cheng <git@zcbenz.com>
Date: Wed, 6 May 2026 09:46:42 +0900
Subject: [PATCH] Make device_count() return 0 when there is no GPU

---
 mlx/backend/metal/device_info.cpp | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/mlx/backend/metal/device_info.cpp b/mlx/backend/metal/device_info.cpp
index b8f5f0e752..3902e6b5e0 100644
--- a/mlx/backend/metal/device_info.cpp
+++ b/mlx/backend/metal/device_info.cpp
@@ -13,7 +13,12 @@ bool is_available() {
 }
 
 int device_count() {
-  return 1;
+  try {
+    metal::device(Device::gpu);
+    return 1;
+  } catch (...) {
+    return 0;
+  }
 }
 
 const std::unordered_map<std::string, std::variant<std::string, size_t>>&