Enable the affine kernel for u8/u32. (#2376)

2025-06-18 03:28:50 +00:00 · 2024-08-01 09:03:11 +01:00
parent 4a52aeb437
commit 8696cf6494
2 changed files with 4 additions and 0 deletions
--- a/candle-core/src/metal_backend/mod.rs
+++ b/candle-core/src/metal_backend/mod.rs
@ -119,6 +119,8 @@ impl BackendStorage for MetalStorage {
                DType::F32 => "affine_f32",
                DType::F16 => "affine_f16",
                DType::BF16 => "affine_bf16",
+                DType::U8 => "affine_u8",
+                DType::U32 => "affine_u32",
                dtype => crate::bail!("Metal contiguous affine {dtype:?} not implemented"),
            };
            candle_metal_kernels::call_affine(