Make device_count() return 0 when there is no GPU (#3486)

zcbenz · web-flow · commit c9aa560577d4 · 2026-05-08T08:33:55.000+09:00
diff --git 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>>&

Original file line number	Diff line number	Diff line change
`@@ -13,7 +13,12 @@ bool is_available() {`
`13`	`13`	`}`
`14`	`14`
`15`	`15`	`int device_count() {`
`16`		`- return 1;`
	`16`	`+ try {`
	`17`	`+ metal::device(Device::gpu);`
	`18`	`+ return 1;`
	`19`	`+ } catch (...) {`
	`20`	`+ return 0;`
	`21`	`+ }`
`17`	`22`	`}`
`18`	`23`
`19`	`24`	`const std::unordered_map<std::string, std::variant<std::string, size_t>>&`