more

mimiquate · Jan 18, 2024 · 6ba026e · 6ba026e
1 parent 95efec7
commit 6ba026e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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 */