Fix make_zero on undef arrays (#1112)

EnzymeAD · Oct 27, 2023 · 644b9d9 · 644b9d9
1 parent 12b9c59
commit 644b9d9
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/compiler.jl b/src/compiler.jl
@@ -2108,8 +2108,10 @@ end
     newa = RT(undef, size(prev))
     seen[prev] = newa
     for I in eachindex(prev)
-        pv = prev[I]
-        @inbounds newa[I] = make_zero(Core.Typeof(pv), seen, pv, Val(copy_if_inactive))
+        if isassigned(prev, I)
+            pv = prev[I]
+            @inbounds newa[I] = make_zero(Core.Typeof(pv), seen, pv, Val(copy_if_inactive))
+        end
     end
     return newa
 end