refactor: don't use NaNMath.pow in codegen rewriters if integral exponent

Author: AayushSabharwal

src/code.jl

@@ -146,11 +146,13 @@ function function_to_expr(op::typeof(^), O, st)
             return toexpr(Term(inv, Any[ex]), st)
         else
             args = Any[Term(inv, Any[ex]), -args[2]]
-            op = get(st.rewrites, :nanmath, false) ? op : NaNMath.pow
+            op = get(st.rewrites, :nanmath, false) || args[2] isa Integer ? op : NaNMath.pow
             return toexpr(Term(op, args), st)
         end
     end
-    get(st.rewrites, :nanmath, false) === true || return nothing
+    if !get(st.rewrites, :nanmath, false) || args[2] isa Integer
+        return nothing
+    end
     return toexpr(Term(NaNMath.pow, args), st)
 end
 

test/code.jl

@@ -100,19 +100,19 @@ nanmath_st.rewrites[:nanmath] = true
     @test toexpr(NaNMath.pow(a, b), nanmath_st) == :($(NaNMath.pow)(a, b))
 
     @test toexpr(a^2) == :($(^)(a, 2))
-    @test toexpr(a^2, nanmath_st) == :($(NaNMath.pow)(a, 2))
+    @test toexpr(a^2, nanmath_st) == :($(^)(a, 2))
     @test toexpr(NaNMath.pow(a, 2)) == :($(^)(a, 2))
-    @test toexpr(NaNMath.pow(a, 2), nanmath_st) == :($(NaNMath.pow)(a, 2))
+    @test toexpr(NaNMath.pow(a, 2), nanmath_st) == :($(^)(a, 2))
 
     @test toexpr(a^-1) == :($(/)(1, a))
     @test toexpr(a^-1, nanmath_st) == :($(/)(1, a))
     @test toexpr(NaNMath.pow(a, -1)) == :($(inv)(a))
     @test toexpr(NaNMath.pow(a, -1), nanmath_st) == :($(inv)(a))
 
     @test toexpr(a^-2) == :($(/)(1, $(^)(a, 2)))
-    @test toexpr(a^-2, nanmath_st) == :($(/)(1, $(NaNMath.pow)(a, 2)))
-    @test toexpr(NaNMath.pow(a, -2)) == :($(NaNMath.pow)($(inv)(a), 2))
-    @test toexpr(NaNMath.pow(a, -2), nanmath_st) == :($(NaNMath.pow)($(inv)(a), 2))
+    @test toexpr(a^-2, nanmath_st) == :($(/)(1, $(^)(a, 2)))
+    @test toexpr(NaNMath.pow(a, -2)) == :($(^)($(inv)(a), 2))
+    @test toexpr(NaNMath.pow(a, -2), nanmath_st) == :($(^)($(inv)(a), 2))
 
     f = GlobalRef(NaNMath, :sin)
     test_repr(toexpr(LiteralExpr(:(let x=1, y=2