fix

EnzymeAD · Aug 9, 2024 · 86bc098 · 86bc098
1 parent 55f27cd
commit 86bc098
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/rules/jitrules.jl b/src/rules/jitrules.jl
@@ -987,7 +987,7 @@ end
         else
             mexprs = Vector{Expr}(undef, width)
             for w in 1:width
-                @inbounds mexprs[i] = :(tape.shadow_return[$w][][$i])
+                @inbounds mexprs[w] = :(tape.shadow_return[$w][][$i])
             end
             quote
                 ($(mexprs...),)

diff --git a/src/rules/typeunstablerules.jl b/src/rules/typeunstablerules.jl
@@ -628,7 +628,7 @@ end
     for i in 1:N
         @inbounds expr[i] = quote
             begin
-                dv = dptrs[$i]
+                dv = ptrs[$i]
                 getfield(dv isa Base.RefValue ? dv[] : dv, symname)
             end
         end