fix: clamp (#1248)

Author: avik-pal

src/TracedRNumber.jl

@@ -427,7 +427,14 @@ function Base.iseven(x::TracedRNumber{<:Real})
 end
 
 for (minT, maxT) in Iterators.product((Number, TracedRNumber), (Number, TracedRNumber))
-    @eval Base.clamp(x::TracedRNumber, min::$(minT), max::$(maxT)) = Ops.clamp(min, x, max)
+    @eval function Base.clamp(x::TracedRNumber, min::$(minT), max::$(maxT))
+        T = promote_type(unwrapped_eltype(x), unwrapped_eltype(min), unwrapped_eltype(max))
+        return Ops.clamp(
+            TracedUtils.promote_to(TracedRNumber{T}, min),
+            TracedUtils.promote_to(TracedRNumber{T}, x),
+            TracedUtils.promote_to(TracedRNumber{T}, max),
+        )
+    end
 end
 
 function Base.fill(x::TracedRNumber, dims::NTuple{N,Integer}) where {N}

test/basic.jl

@@ -509,6 +509,10 @@ end
         @test minimum(y) ≥ 0.0
         @test x_ra ≈ x
     end
+
+    x_ra = ConcreteRNumber(3.0)
+    y = @jit(clamp(x_ra, 0.0, 0.25))
+    @test y isa ConcreteRNumber{Float64}
 end
 
 @testset for op in [round, ceil, floor]