feat: implement atan(y, x) with atan2 (#1927)

* atan2

* test

Author: jumerckx

src/TracedRNumber.jl

@@ -533,6 +533,7 @@ Base.asind(x::TracedRNumber) = rad2deg(asin(x))
 Base.acosd(x::TracedRNumber) = rad2deg(acos(x))
 Base.atand(x::TracedRNumber) = rad2deg(atan(x))
 
+Base.atan(y::TracedRNumber, x::TracedRNumber) = @opcall atan2(y, x)
 Base.atand(y::TracedRNumber, x::TracedRNumber) = rad2deg(atan(y, x))
 
 Base.acscd(x::TracedRNumber) = rad2deg(asin(1 / x))

test/basic.jl

@@ -770,6 +770,14 @@ end
         @test @jit(fn.(xrad_ra)) isa ConcreteRArray{Float32,2}
     end
 
+    yrad = Reactant.TestUtils.construct_test_array(Float32, 4, 16)[:, 3:9]
+    yrad_ra = Reactant.to_rarray(yrad)
+
+    @testset for fn in (atan, atand)
+        @test @jit(fn.(yrad_ra, xrad_ra)) ≈ fn.(yrad, xrad)
+        @test @jit(fn.(yrad_ra, xrad_ra)) isa ConcreteRArray{Float32,2}
+    end
+
     x = 0.235f0
     x_ra = Reactant.to_rarray(x; track_numbers=Number)