From afb6575835599938248c027f50a8100c289a1a96 Mon Sep 17 00:00:00 2001
From: Laurent Mazare <laurent.mazare@gmail.com>
Date: Wed, 11 Sep 2024 16:34:05 +0100
Subject: Use the new MLX kernels to handle the BF16 matmul. (#2470)

---
 candle-core/src/device.rs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'candle-core/src/device.rs')

diff --git a/candle-core/src/device.rs b/candle-core/src/device.rs
index 45f6554b..91e56937 100644
--- a/candle-core/src/device.rs
+++ b/candle-core/src/device.rs
@@ -173,8 +173,8 @@ impl Device {
 
     pub fn supports_bf16(&self) -> bool {
         match self {
-            Self::Cuda(_) => true,
-            Self::Metal(_) | Self::Cpu => false,
+            Self::Cuda(_) | Self::Metal(_) => true,
+            Self::Cpu => false,
         }
     }
 
-- 
cgit v1.2.3