We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 52c7176 commit 85bec83Copy full SHA for 85bec83
src/Overlay.jl
@@ -24,8 +24,9 @@ end
24
@reactant_overlay function EnzymeCore.ignore_derivatives(args...)
25
res = map(args) do arg
26
return Functors.fmap(arg) do argᵢ
27
- argᵢ isa AnyTracedRArray &&
28
- (argᵢ = call_with_reactant(materialize_traced_array, argᵢ))
+ if argᵢ isa AnyTracedRArray && !(argᵢ isa TracedType)
+ argᵢ = call_with_reactant(materialize_traced_array, argᵢ)
29
+ end
30
argᵢ isa TracedType && return @opcall ignore_derivatives(argᵢ)
31
return argᵢ
32
end
0 commit comments