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>&