Support tracing of binary ops with only one operand being a `Concrete…

…RNumber`

src/TracedRNumber.jl

@@ -97,10 +97,20 @@ for (jlop, hloop) in (
     (:(Base.mod), :remainder),
     (:(Base.rem), :remainder),
 )
-    @eval function $(jlop)(
-        @nospecialize(lhs::TracedRNumber{T}), @nospecialize(rhs::TracedRNumber{T})
-    ) where {T}
-        return Ops.$(hloop)(lhs, rhs)
+    @eval begin
+        function $(jlop)(
+            @nospecialize(lhs::TracedRNumber{T}), @nospecialize(rhs::TracedRNumber{T})
+        ) where {T}
+            return Ops.$(hloop)(lhs, rhs)
+        end
+
+        function $(jlop)(@nospecialize(lhs::TracedRNumber{T}), @nospecialize(rhs)) where {T}
+            return Ops.$(hloop)(lhs, TracedUtils.promote_to(TracedRNumber{T}, rhs))
+        end
+
+        function $(jlop)(@nospecialize(lhs), @nospecialize(rhs::TracedRNumber{T})) where {T}
+            return Ops.$(hloop)(TracedUtils.promote_to(TracedRNumber{T}, lhs), rhs)
+        end
     end
 end
 

test/basic.jl

@@ -825,6 +825,15 @@ end
     @test !isfinite(Reactant.to_rarray(Inf; track_numbers=Number))
 end
 
+@testset "rem" begin
+    a = [-1.1,  7.7, -3.3, 9.9,  -5.5]
+    b = [ 6.6, -2.2, -8.8, 4.4, -10.1]
+    expected = rem.(a, b)
+    @test Reactant.@jit(rem.(Reactant.to_rarray(a), Reactant.to_rarray(b))) ≈ expected
+    @test Reactant.@jit(rem.(a, Reactant.to_rarray(b))) ≈ expected
+    @test Reactant.@jit(rem.(Reactant.to_rarray(a), b)) ≈ expected
+end
+
 @testset "reduce integers" begin
     x = rand(Bool, 100)
     x_ra = Reactant.to_rarray(x)