fix: exponents

Author: avik-pal

src/TracedRArray.jl

@@ -267,7 +267,7 @@ for (jlop, hloop) in (
         function $(jlop)(
             @nospecialize(lhs::TracedRArray{T1,0}), @nospecialize(rhs::TracedRArray{T2,0})
         ) where {T1,T2}
-            commonTy = promote_type(T1, T2)
+            commonTy = TracedRArray{Base.promote_type(T1, T2),0}
             lhs = promote_to(commonTy, lhs)
             rhs = promote_to(commonTy, rhs)
             return $(jlop)(lhs, rhs)

test/bcast.jl

@@ -101,3 +101,14 @@ end
     ln_comp = @compile custom_ln(x_ra)
     @test ln_comp(x_ra) ≈ custom_ln(x)
 end
+
+pow(x, n) = x .^ n
+
+@testset "Pow" begin
+    x = rand(2, 3)
+    x_ra = Reactant.to_rarray(x)
+
+    pow_compiled = @compile pow(x_ra, 2)
+
+    @test pow_compiled(x_ra) ≈ pow(x, 2)
+end