From 6ba026ed361ab58cf4b004179f956e8ea1024098 Mon Sep 17 00:00:00 2001 From: Gonzalo <456459+grzuy@users.noreply.github.com> Date: Thu, 18 Jan 2024 10:27:57 -0300 Subject: [PATCH] more --- native/candlex/src/metal_kernels/custom_unary.metal | 3 +++ 1 file changed, 3 insertions(+) diff --git a/native/candlex/src/metal_kernels/custom_unary.metal b/native/candlex/src/metal_kernels/custom_unary.metal index 36f07f5..7f7a7dd 100644 --- a/native/candlex/src/metal_kernels/custom_unary.metal +++ b/native/candlex/src/metal_kernels/custom_unary.metal @@ -52,8 +52,11 @@ CUSTOM_UNARY(float, float, atan_f32, atan) CUSTOM_UNARY(float, float, atanh_f32, atanh) CUSTOM_UNARY(float, float, cosh_f32, cosh) CUSTOM_UNARY(float, float, sign_f32, sign) +CUSTOM_UNARY(int64_t, int64_t, sign_i64, sign) CUSTOM_UNARY(float, float, sinh_f32, sinh) CUSTOM_UNARY(float, float, tan_f32, tan) +CUSTOM_UNARY(uint64_t, uint64_t, bit_not_i64, not) +CUSTOM_UNARY(uint8_t, uint8_t, bit_not_u8, not) /* bit_not */ /* cbrt */