fix array indexing backward rule (#43)

Co-authored-by: Simeon Schaub <[email protected]>
JuliaDiff · Sep 8, 2021 · b95eb68 · b95eb68
1 parent de287ab
commit b95eb68
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/stage1/generated.jl b/src/stage1/generated.jl
@@ -313,6 +313,7 @@ function (::∂⃖{N})(::typeof(Base.getindex), a::Array, inds...) where {N}
     getindex(a, inds...), let
         EvenOddOdd{1, c_order(N)}(
             (@Base.aggressive_constprop Δ->begin
+                Δ isa AbstractZero && return (NoTangent(), Δ, map(Returns(Δ), inds)...)
                 BB = zero(a)
                 BB[inds...] = Δ
                 (NoTangent(), BB, map(x->NoTangent(), inds)...)