diff --git a/src/Ops.jl b/src/Ops.jl index e001ce9fc..ac1b1b61a 100644 --- a/src/Ops.jl +++ b/src/Ops.jl @@ -2118,7 +2118,7 @@ end TracedRNumber{Reactant.unwrapped_eltype(arg)}, arg ) end - return f(scalar_args...) + return Reactant.call_with_reactant(f, scalar_args...) end end