diff options
| author | Daniël de Kok <me@danieldk.eu> | 2024-03-22 11:51:41 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-03-22 11:51:41 +0100 |
| commit | fc1fe5e45b046771589126c355fdfb4d3bb49fe4 (patch) | |
| tree | a775f9b0f9df0d189deb2078d2f0c7dc8d8bf9ff /candle-kernels/src | |
| parent | 32f567bac491aa0f52dfbe1001ea4d6187bb4301 (diff) | |
| download | candle-fc1fe5e45b046771589126c355fdfb4d3bb49fe4.tar.gz candle-fc1fe5e45b046771589126c355fdfb4d3bb49fe4.tar.bz2 candle-fc1fe5e45b046771589126c355fdfb4d3bb49fe4.zip | |
Support scatter/index_add with i64 indices for f16 (#1915)
Diffstat (limited to 'candle-kernels/src')
| -rw-r--r-- | candle-kernels/src/indexing.cu | 2 |
1 files changed, 2 insertions, 0 deletions
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 |