diff --git a/candle-core/src/metal_backend/mod.rs b/candle-core/src/metal_backend/mod.rs index 9845a42fd1..a60c475bbc 100644 --- a/candle-core/src/metal_backend/mod.rs +++ b/candle-core/src/metal_backend/mod.rs @@ -412,6 +412,8 @@ impl BackendStorage for MetalStorage { .map_err(MetalError::from)?; } else { let kernel_name = match (self.dtype, dtype) { + (DType::U32, DType::F16) => "cast_u32_f16_strided", + (DType::U32, DType::BF16) => "cast_u32_bf16_strided", (DType::U32, DType::F32) => "cast_u32_f32_strided", (DType::U32, DType::U8) => "cast_u32_u8_strided", (DType::U32, DType::I64) => "cast_u32_i64_strided",