From b58b24732383257c73912c57f3e8bfa77f377f7d Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Sat, 11 Nov 2023 17:14:06 +0100 Subject: [PATCH] Putting back f16 index select. --- candle-metal-kernels/src/indexing.metal | 1 + 1 file changed, 1 insertion(+) diff --git a/candle-metal-kernels/src/indexing.metal b/candle-metal-kernels/src/indexing.metal index 007f9fed..312b27c7 100644 --- a/candle-metal-kernels/src/indexing.metal +++ b/candle-metal-kernels/src/indexing.metal @@ -75,6 +75,7 @@ kernel void FN_NAME( \ INDEX_OP(is_u32_f32, uint, float) +INDEX_OP(is_u32_f16, uint, half) #if __METAL_VERSION__ >= 310