fix: call with reactant

EnzymeAD · Jan 20, 2025 · ab15e8b · ab15e8b
1 parent 02bdfb1
commit ab15e8b
Showing 1 changed file with 1 addition and 1 deletion.
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