diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2024-09-11 16:34:05 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-09-11 17:34:05 +0200 |
| commit | afb6575835599938248c027f50a8100c289a1a96 (patch) | |
| tree | 32c5f34269cb2a45684f3131a275a3bc8ad50237 /candle-core/src/device.rs | |
| parent | 5635650d386ff7cfeb0dee84d02e8d19574c6faf (diff) | |
| download | candle-afb6575835599938248c027f50a8100c289a1a96.tar.gz candle-afb6575835599938248c027f50a8100c289a1a96.tar.bz2 candle-afb6575835599938248c027f50a8100c289a1a96.zip | |
Use the new MLX kernels to handle the BF16 matmul. (#2470)
Diffstat (limited to 'candle-core/src/device.rs')
| -rw-r--r-- | candle-core/src/device.rs | 4 |
1 files changed, 2 insertions, 2 deletions
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, } } |