Mark gc preserve as readonly (#1724)

EnzymeAD · Aug 9, 2024 · 02c5855 · 02c5855
1 parent 957f082
commit 02c5855
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/compiler/utils.jl b/src/compiler/utils.jl
@@ -120,6 +120,9 @@ function is_readonly(f::LLVM.Function)
     if intr == LLVM.Intrinsic("llvm.assume").id
         return true
     end
+    if LLVM.name(f) == "llvm.julia.gc_preserve_begin" || LLVM.name(f) == "llvm.julia.gc_preserve_end"
+        return true
+    end
     for attr in collect(function_attributes(f))
         if kind(attr) == kind(EnumAttribute("readonly"))
             return true
@@ -149,6 +152,9 @@ function is_readnone(f::LLVM.Function)
     if intr == LLVM.Intrinsic("llvm.assume").id
         return true
     end
+    if LLVM.name(f) == "llvm.julia.gc_preserve_begin" || LLVM.name(f) == "llvm.julia.gc_preserve_end"
+        return true
+    end
     for attr in collect(function_attributes(cur))
         if kind(attr) == kind(EnumAttribute("readnone"))
             return true
@@ -175,6 +181,9 @@ function is_writeonly(f::LLVM.Function)
     if intr == LLVM.Intrinsic("llvm.assume").id
         return true
     end
+    if LLVM.name(f) == "llvm.julia.gc_preserve_begin" || LLVM.name(f) == "llvm.julia.gc_preserve_end"
+        return true
+    end
     for attr in collect(function_attributes(cur))
         if kind(attr) == kind(EnumAttribute("readnone"))
             return true