summaryrefslogtreecommitdiff
path: root/candle-metal-kernels
diff options
context:
space:
mode:
authorivarflakstad <69173633+ivarflakstad@users.noreply.github.com>2024-01-18 08:40:34 +0100
committerGitHub <noreply@github.com>2024-01-18 08:40:34 +0100
commit1cf34368b7d10600ca2fac197cd49c8bad2f6ad1 (patch)
tree77fe432b6cc3fd0ae4f5250090610b63c3e1cd41 /candle-metal-kernels
parent403680f17ddc086295fbaee316cbed22d97a519b (diff)
parent17e6e2d7ee111abac7a2cf806cd888ff780da452 (diff)
downloadcandle-1cf34368b7d10600ca2fac197cd49c8bad2f6ad1.tar.gz
candle-1cf34368b7d10600ca2fac197cd49c8bad2f6ad1.tar.bz2
candle-1cf34368b7d10600ca2fac197cd49c8bad2f6ad1.zip
Merge pull request #1602 from mimiquate/fix-metal-kernel-type
Metal: Use uint8_t as output type in int64_t binary op kernel
Diffstat (limited to 'candle-metal-kernels')
-rw-r--r--candle-metal-kernels/src/binary.metal2
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-metal-kernels/src/binary.metal b/candle-metal-kernels/src/binary.metal
index eb560f16..ae11286a 100644
--- a/candle-metal-kernels/src/binary.metal
+++ b/candle-metal-kernels/src/binary.metal
@@ -73,7 +73,7 @@ BINARY(FN, uint32_t, uint8_t, NAME##_u32, NAME##_u32_strided); \
BINARY(FN, uint8_t, uint8_t, NAME##_u8, NAME##_u8_strided);
#define INT64_BINARY_OP_OUT(NAME, FN) \
-BINARY(FN, int64_t, int8_t, NAME##_i64, NAME##_i64_strided);
+BINARY(FN, int64_t, uint8_t, NAME##_i64, NAME##_i64_strided);
BINARY_OP(x + y, add)
BINARY_OP(x - y, sub)