Update validation.jl

EnzymeAD · Dec 6, 2024 · d629c30 · d629c30
1 parent 39ac9eb
commit d629c30
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/src/compiler/validation.jl b/src/compiler/validation.jl
@@ -140,6 +140,21 @@ import GPUCompiler: IRError, InvalidIRError
 
 function restore_lookups(mod::LLVM.Module)::Nothing
     T_size_t = convert(LLVM.LLVMType, Int)
+    for (v, k) in FFI.ptr_map
+        if haskey(functions(mod), k)
+            f = functions(mod)[k]
+            replace_uses!(
+                f,
+                LLVM.Value(
+                    LLVM.API.LLVMConstIntToPtr(
+                        ConstantInt(T_size_t, convert(UInt, v)),
+                        value_type(f),
+                    ),
+                ),
+            )
+            eraseInst(mod, f)
+        end
+    end
     for f in functions(mod)
         for fattr in collect(function_attributes(f))        
             if isa(fattr, LLVM.StringAttribute)