Second phi embarassing bugfix (#1245)

EnzymeAD · Jan 24, 2024 · 74935a3 · 74935a3
1 parent 0c9f0df
commit 74935a3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/compiler/optimize.jl b/src/compiler/optimize.jl
@@ -134,20 +134,20 @@ function nodecayed_phis!(mod::LLVM.Module)
                 if addr == 11
                     all_args = true
                     addrtodo = Value[inst]
+                    seen = Set{LLVM.Value}()
 
                     while length(addrtodo) != 0
                         v = pop!(addrtodo)
                         base = get_base_object(v)
-                        seen = Set{LLVM.Value}()
                         if in(base, seen)
                             continue
                         end
                         push!(seen, base)
                         if isa(base, LLVM.Argument) && addrspace(value_type(base)) == 11
                             continue
                         end
-                        if isa(v, LLVM.PHIInst)
-                            for (v, _) in LLVM.incoming(inst)
+                        if isa(base, LLVM.PHIInst)
+                            for (v, _) in LLVM.incoming(base)
                                 push!(addrtodo, v)
                             end
                             continue