fix: call with reactant

avik-pal · avik-pal · commit ab15e8b9e114 · 2025-01-20T11:31:21.000-05:00
diff --git 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
 

Original file line number	Diff line number	Diff line change
`@@ -2118,7 +2118,7 @@ end`
`2118`	`2118`	`TracedRNumber{Reactant.unwrapped_eltype(arg)}, arg`
`2119`	`2119`	`)`
`2120`	`2120`	`end`
`2121`		`- return f(scalar_args...)`
	`2121`	`+ return Reactant.call_with_reactant(f, scalar_args...)`
`2122`	`2122`	`end`
`2123`	`2123`	`end`
`2124`	`2124`