From fc1fe5e45b046771589126c355fdfb4d3bb49fe4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dani=C3=ABl=20de=20Kok?= Date: Fri, 22 Mar 2024 11:51:41 +0100 Subject: [PATCH] Support scatter/index_add with i64 indices for f16 (#1915) --- candle-kernels/src/indexing.cu | 2 ++ 1 file changed, 2 insertions(+) diff --git a/candle-kernels/src/indexing.cu b/candle-kernels/src/indexing.cu index 8fc69363..8af2954d 100644 --- a/candle-kernels/src/indexing.cu +++ b/candle-kernels/src/indexing.cu @@ -168,8 +168,10 @@ IS_OP(__half, uint8_t, is_u8_f16) GATHER_OP(__half, int64_t, gather_i64_f16) GATHER_OP(__half, uint32_t, gather_u32_f16) GATHER_OP(__half, uint8_t, gather_u8_f16) +IA_OP(__half, int64_t, ia_i64_f16) IA_OP(__half, uint32_t, ia_u32_f16) IA_OP(__half, uint8_t, ia_u8_f16) +SA_OP(__half, int64_t, sa_i64_f16) SA_OP(__half, uint32_t, sa_u32_f16) SA_OP(__half, uint8_t, sa_u8_f16) #endif