feat: add trigonometric functions

crates/cubecl-core/src/frontend/element/float.rs

@@ -25,7 +25,19 @@ pub trait Float:
     + Log1p
     + Cos
     + Sin
+    + Tan
     + Tanh
+    + Sinh
+    + Cosh
+    + ArcCos
+    + ArcSin
+    + ArcTan
+    + ArcSinh
+    + ArcCosh
+    + ArcTanh
+    + Degrees
+    + Radians
+    + ArcTan2
     + Powf
     + Powi<i32>
     + Sqrt

crates/cubecl-core/src/frontend/element/float/typemap.rs

@@ -243,7 +243,19 @@ impl<const POS: u8> Log for ElemExpand<POS> {}
 impl<const POS: u8> Log1p for ElemExpand<POS> {}
 impl<const POS: u8> Cos for ElemExpand<POS> {}
 impl<const POS: u8> Sin for ElemExpand<POS> {}
+impl<const POS: u8> Tan for ElemExpand<POS> {}
 impl<const POS: u8> Tanh for ElemExpand<POS> {}
+impl<const POS: u8> Sinh for ElemExpand<POS> {}
+impl<const POS: u8> Cosh for ElemExpand<POS> {}
+impl<const POS: u8> ArcCos for ElemExpand<POS> {}
+impl<const POS: u8> ArcSin for ElemExpand<POS> {}
+impl<const POS: u8> ArcTan for ElemExpand<POS> {}
+impl<const POS: u8> ArcSinh for ElemExpand<POS> {}
+impl<const POS: u8> ArcCosh for ElemExpand<POS> {}
+impl<const POS: u8> ArcTanh for ElemExpand<POS> {}
+impl<const POS: u8> Degrees for ElemExpand<POS> {}
+impl<const POS: u8> Radians for ElemExpand<POS> {}
+impl<const POS: u8> ArcTan2 for ElemExpand<POS> {}
 impl<const POS: u8> Powf for ElemExpand<POS> {}
 impl<const POS: u8, I: CubePrimitive> Powi<I> for ElemExpand<POS> {}
 impl<const POS: u8> Sqrt for ElemExpand<POS> {}

crates/cubecl-core/src/frontend/operation/binary.rs

@@ -252,6 +252,17 @@ impl_binary_func!(
     f32,
     f64
 );
+impl_binary_func!(
+    ArcTan2,
+    atan2,
+    Arithmetic::ArcTan2,
+    f16,
+    bf16,
+    flex32,
+    tf32,
+    f32,
+    f64
+);
 impl_binary_func!(
     Max,
     max,

crates/cubecl-core/src/frontend/operation/unary.rs

@@ -175,6 +175,18 @@ impl_unary_func!(
     f32,
     f64
 );
+impl_unary_func!(
+    Tan,
+    tan,
+    __expand_tan,
+    Arithmetic::Tan,
+    f16,
+    bf16,
+    flex32,
+    tf32,
+    f32,
+    f64
+);
 impl_unary_func!(
     Tanh,
     tanh,
@@ -187,6 +199,126 @@ impl_unary_func!(
     f32,
     f64
 );
+impl_unary_func!(
+    Sinh,
+    sinh,
+    __expand_sinh,
+    Arithmetic::Sinh,
+    f16,
+    bf16,
+    flex32,
+    tf32,
+    f32,
+    f64
+);
+impl_unary_func!(
+    Cosh,
+    cosh,
+    __expand_cosh,
+    Arithmetic::Cosh,
+    f16,
+    bf16,
+    flex32,
+    tf32,
+    f32,
+    f64
+);
+impl_unary_func!(
+    ArcCos,
+    acos,
+    __expand_acos,
+    Arithmetic::ArcCos,
+    f16,
+    bf16,
+    flex32,
+    tf32,
+    f32,
+    f64
+);
+impl_unary_func!(
+    ArcSin,
+    asin,
+    __expand_asin,
+    Arithmetic::ArcSin,
+    f16,
+    bf16,
+    flex32,
+    tf32,
+    f32,
+    f64
+);
+impl_unary_func!(
+    ArcTan,
+    atan,
+    __expand_atan,
+    Arithmetic::ArcTan,
+    f16,
+    bf16,
+    flex32,
+    tf32,
+    f32,
+    f64
+);
+impl_unary_func!(
+    ArcSinh,
+    asinh,
+    __expand_asinh,
+    Arithmetic::ArcSinh,
+    f16,
+    bf16,
+    flex32,
+    tf32,
+    f32,
+    f64
+);
+impl_unary_func!(
+    ArcCosh,
+    acosh,
+    __expand_acosh,
+    Arithmetic::ArcCosh,
+    f16,
+    bf16,
+    flex32,
+    tf32,
+    f32,
+    f64
+);
+impl_unary_func!(
+    ArcTanh,
+    atanh,
+    __expand_atanh,
+    Arithmetic::ArcTanh,
+    f16,
+    bf16,
+    flex32,
+    tf32,
+    f32,
+    f64
+);
+impl_unary_func!(
+    Degrees,
+    to_degrees,
+    __expand_to_degrees,
+    Arithmetic::Degrees,
+    f16,
+    bf16,
+    flex32,
+    tf32,
+    f32,
+    f64
+);
+impl_unary_func!(
+    Radians,
+    to_radians,
+    __expand_to_radians,
+    Arithmetic::Radians,
+    f16,
+    bf16,
+    flex32,
+    tf32,
+    f32,
+    f64
+);
 impl_unary_func!(
     Sqrt,
     sqrt,

crates/cubecl-core/src/runtime_tests/binary.rs

@@ -150,6 +150,35 @@ test_binary_impl!(
     ]
 );
 
+test_binary_impl!(
+    test_atan2,
+    F,
+    F::atan2,
+    [
+        {
+            input_vectorization: 1,
+            out_vectorization: 1,
+            lhs: as_type![F: 0., 1., -1., 1., -1.],
+            rhs: as_type![F: 1., 0., 0., 1., -1.],
+            expected: as_type![F: 0., 1.57079632679, -1.57079632679, 0.78539816339, -2.35619449019]
+        },
+        {
+            input_vectorization: 2,
+            out_vectorization: 2,
+            lhs: as_type![F: 0., 1., -1., 1.],
+            rhs: as_type![F: 1., 0., 0., 1.],
+            expected: as_type![F: 0., 1.57079632679, -1.57079632679, 0.78539816339]
+        },
+        {
+            input_vectorization: 4,
+            out_vectorization: 4,
+            lhs: as_type![F: 0., 1., -1., 1.],
+            rhs: as_type![F: 1., 0., 0., 1.],
+            expected: as_type![F: 0., 1.57079632679, -1.57079632679, 0.78539816339]
+        }
+    ]
+);
+
 #[cube(launch_unchecked)]
 fn test_powi_kernel<F: Float>(
     lhs: &Array<Line<F>>,
@@ -321,6 +350,7 @@ macro_rules! testgen_binary {
             add_test!(test_dot);
             add_test!(test_powf);
             add_test!(test_powi);
+            add_test!(test_atan2);
         }
     };
 }