diff options
| author | Gonzalo <456459+grzuy@users.noreply.github.com> | 2023-12-29 20:00:12 -0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-12-30 00:00:12 +0100 |
| commit | 0a245e6fa46c16f332555a58271dbd49a8058a9c (patch) | |
| tree | 0734edfe429befea01dbcac701207cc815269166 /candle-metal-kernels/src/unary.metal | |
| parent | 87d7f81b438db6a1696f2ac79606b40e61d448e8 (diff) | |
| download | candle-0a245e6fa46c16f332555a58271dbd49a8058a9c.tar.gz candle-0a245e6fa46c16f332555a58271dbd49a8058a9c.tar.bz2 candle-0a245e6fa46c16f332555a58271dbd49a8058a9c.zip | |
Metal: support unary abs (#1503)
* Metal: support unary abs
* cargo fmt
Diffstat (limited to 'candle-metal-kernels/src/unary.metal')
| -rw-r--r-- | candle-metal-kernels/src/unary.metal | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/candle-metal-kernels/src/unary.metal b/candle-metal-kernels/src/unary.metal index 15d1e400..7fbb613d 100644 --- a/candle-metal-kernels/src/unary.metal +++ b/candle-metal-kernels/src/unary.metal @@ -102,6 +102,7 @@ UNARY_OP(neg) UNARY_OP(exp) UNARY_OP(log) UNARY_OP(gelu) +UNARY_OP(abs) UNARY_OP(ceil) UNARY_OP(floor) UNARY_OP(round) |