Skip to content

Commit 5499312

Browse files
committed
if all partials AbstractZero don't call frule
1 parent 52a5583 commit 5499312

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/stage1/forward.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -160,7 +160,7 @@ function (::∂☆internal{1})(args::AbstractTangentBundle{1}...)
160160
end
161161
end
162162

163-
_frule(partials, primals...) = frule(DiffractorRuleConfig(), partials, primals...)
163+
_frule(partials, primals...) = frule(#== DiffractorRuleConfig(), ==# partials, primals...)
164164
function _frule(::NTuple{<:Any, AbstractZero}, f, primal_args...)
165165
# frules are linear in partials, so zero maps to zero, no need to evaluate the frule
166166
# If all partials are immutable AbstractZero subtyoes we know we don't have to worry about a mutating frule either

0 commit comments

Comments
 (0)