Handle load from global constant in gc fixup (#1179)

EnzymeAD · Dec 3, 2023 · 9d6963c · 9d6963c
1 parent e627736
commit 9d6963c
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/compiler/optimize.jl b/src/compiler/optimize.jl
@@ -239,6 +239,18 @@ function nodecayed_phis!(mod::LLVM.Module)
                         return getparent(operands(v)[1], offset, true)
                     end
 
+                    if addr == 13 && isa(v, LLVM.ConstantExpr)
+                        if opcode(v) == LLVM.API.LLVMAddrSpaceCast
+                            v2 = operands(v)[1]
+                            if addrspace(value_type(v2)) == 0
+                                if addr == 13 && isa(v, LLVM.ConstantExpr)
+                                    v2 = const_addrspacecast(operands(v)[1], LLVM.PointerType(eltype(value_type(v)), 10))
+                                    return v2, offset, hasload
+                                end
+                            end
+                        end
+                    end
+
                     if isa(v, LLVM.AddrSpaceCastInst)
                         if addrspace(value_type(operands(v)[1])) == 0
                             v2 = addrspacecast!(b, operands(v)[1], LLVM.PointerType(eltype(value_type(v)), 10))