add where_cond f32 for metal (#2236)

2025-06-18 11:37:11 +00:00 · 2024-06-02 14:30:06 +02:00
parent f7773d498a
commit 1ec3b2cc18
3 changed files with 23 additions and 1 deletions
--- a/candle-core/src/metal_backend/mod.rs
+++ b/candle-core/src/metal_backend/mod.rs
@ -718,6 +718,7 @@ impl BackendStorage for MetalStorage {
        }
        let name = match (self.dtype, t.dtype()) {
            (DType::U8, DType::F32) => "where_u8_f32",
+            (DType::U32, DType::F32) => "where_u32_f32",
            (DType::U8, DType::BF16) => "where_u8_bf16",
            (DType::U8, DType::F16) => "where_u8_f16",
            (DType::U8, DType::I64) => "where_u8_i64",