diff options
author | ivarflakstad <69173633+ivarflakstad@users.noreply.github.com> | 2024-01-18 08:40:34 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2024-01-18 08:40:34 +0100 |
commit | 1cf34368b7d10600ca2fac197cd49c8bad2f6ad1 (patch) | |
tree | 77fe432b6cc3fd0ae4f5250090610b63c3e1cd41 /candle-metal-kernels | |
parent | 403680f17ddc086295fbaee316cbed22d97a519b (diff) | |
parent | 17e6e2d7ee111abac7a2cf806cd888ff780da452 (diff) | |
download | candle-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.metal | 2 |
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) |