Generalize gpu backend (#2138)

* generalize gpu backend * fix no_gpu build * fix no_gpu build * generalize gpu backend
2025-12-16 01:49:05 +08:00 · 2025-04-30 09:08:17 -07:00
parent 87720a8908
commit f1606486d2
33 changed files with 275 additions and 200 deletions
--- a/mlx/backend/cpu/available.h
+++ b/mlx/backend/cpu/available.h
@@ -0,0 +1,9 @@
+// Copyright © 2025 Apple Inc.
+
+#pragma once
+
+namespace mlx::core::cpu {
+
+bool is_available();
+
+} // namespace mlx::core::cpu