Pass primal to getfield rev for debugging

EnzymeAD · Sep 6, 2024 · e59bc8d · e59bc8d
1 parent 754937b
commit e59bc8d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/compiler.jl b/src/compiler.jl
@@ -3419,7 +3419,7 @@ for (k, v) in (
     ("enz_runtime_jl_getfield_rev", Enzyme.Compiler.rt_jl_getfield_rev),
 
     ("enz_runtime_idx_jl_getfield_aug", Enzyme.Compiler.idx_jl_getfield_aug),
-    ("enz_runtime_idx_jl_getfield_rev", Enzyme.Compiler.idx_jl_getfield_aug),
+    ("enz_runtime_idx_jl_getfield_rev", Enzyme.Compiler.idx_jl_getfield_rev),
 
     ("enz_runtime_jl_setfield_aug", Enzyme.Compiler.rt_jl_setfield_aug),
     ("enz_runtime_jl_setfield_rev", Enzyme.Compiler.rt_jl_setfield_rev),

diff --git a/src/rules/typeunstablerules.jl b/src/rules/typeunstablerules.jl
@@ -796,7 +796,7 @@ end
     end
 end
 
-function idx_jl_getfield_rev(dptr::T, dret, ::Type{Val{symname}}, ::Val{isconst}, dptrs::Vararg{T2, Nargs}) where {T, T2, Nargs, symname, isconst}
+function idx_jl_getfield_rev(primal, dptr::T, dret, ::Type{Val{symname}}, ::Val{isconst}, dptrs::Vararg{T2, Nargs}) where {T, T2, Nargs, symname, isconst}
     cur = if dptr isa Base.RefValue
 	   Base.getfield(dptr[], symname+1)
     else
@@ -1135,6 +1135,8 @@ end
     end
 
     vals = LLVM.Value[]
+
+    push!(vals, lookup_value(gutils, new_from_original(gutils, ops[1]), B))
     push!(vals, inps[1])
 
     push!(vals, tape)