From b60faebea4b39cdeceb949ac5db464e05983b153 Mon Sep 17 00:00:00 2001 From: Laurent Mazare Date: Thu, 12 Sep 2024 12:58:50 +0100 Subject: [PATCH] Missing metal kernels. (#2474) --- candle-core/src/metal_backend/mod.rs | 2 ++ 1 file changed, 2 insertions(+) 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",