Fix 1.11 arg based gc rewrite (#2250)

* Update transforms.jl * Update transforms.jl * Update transforms.jl * Update transforms.jl * Update transforms.jl
EnzymeAD · Jan 5, 2025 · 2309abd · 2309abd · wsmoses · Jan 5, 2025
1 parent 1a6ff16
commit 2309abd
Showing 1 changed file with 12 additions and 11 deletions.
diff --git a/src/llvm/transforms.jl b/src/llvm/transforms.jl
@@ -563,27 +563,28 @@ function nodecayed_phis!(mod::LLVM.Module)
                         while length(addrtodo) != 0
                             v = pop!(addrtodo)
                             base, toffset = get_base_and_offset(v)
-                            if offset === nothing
+
+                            if in(base, seen)
+                                continue
+                            end
+                            push!(seen, base)		
+                            if isa(base, LLVM.PHIInst)
+                                for (v, _) in LLVM.incoming(base)
+                                    push!(addrtodo, v)
+                                end
+                                continue
+                            end
+			    if offset === nothing
                                 offset = toffset
                             else
                                 if offset != toffset
                                     all_args = false
                                     break
                                 end
                             end
-                            if in(base, seen)
-                                continue
-                            end
-                            push!(seen, base)
                             if isa(base, LLVM.Argument) && addrspace(value_type(base)) == 11
                                 continue
                             end
-                            if isa(base, LLVM.PHIInst)
-                                for (v, _) in LLVM.incoming(base)
-                                    push!(addrtodo, v)
-                                end
-                                continue
-                            end
                             all_args = false
                             break
                         end