Enable BF16 on metal. (#2380)

2025-06-19 03:54:56 +00:00 · 2024-08-01 10:05:07 +01:00
parent ce90287f45
commit 957d604a78
2 changed files with 3 additions and 4 deletions
--- a/candle-core/src/metal_backend/mod.rs
+++ b/candle-core/src/metal_backend/mod.rs
@ -1409,6 +1409,7 @@ impl BackendStorage for MetalStorage {
        let name = match self.dtype {
            DType::F32 => "sgemm",
            DType::F16 => "hgemm",
+            DType::BF16 => "bgemm",
            dtype => {
                return Err(MetalError::Message(format!("matmul doesn't support {dtype:?}")).into())
            }