Fix gc loaded usage

EnzymeAD · Sep 2, 2024 · 4aa1d5b · 4aa1d5b
1 parent 4ddaccd
commit 4aa1d5b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/src/compiler.jl b/src/compiler.jl
@@ -3777,6 +3777,13 @@ function annotate!(mod, mode)
         end
     end
 
+    for fname in ("julia.gc_loaded",)
+        if haskey(fns, fname)
+            fn = fns[fname]
+            push!(function_attributes(fn), LLVM.StringAttribute("enzyme_shouldrecompute"))
+        end
+    end
+
     for fname in ("julia.get_pgcstack", "julia.ptls_states", "jl_get_ptls_states", "julia.safepoint", "ijl_throw", "julia.pointer_from_objref",
                   "ijl_array_grow_end", "jl_array_grow_end", "ijl_array_del_end", "jl_array_del_end",
                   "ijl_array_grow_beg", "jl_array_grow_beg", "ijl_array_del_beg", "jl_array_del_beg",

diff --git a/src/rules/llvmrules.jl b/src/rules/llvmrules.jl
@@ -1368,12 +1368,6 @@ end
         @revfunc(jlcall2_rev),
         @fwdfunc(jlcall2_fwd),
     )
-    register_handler!(
-        ("julia.gc_loaded",),
-        @augfunc(gcloaded_augfwd),
-        @revfunc(gcloaded_rev),
-        @fwdfunc(gcloaded_fwd),
-    )
     register_handler!(
         ("jl_apply_generic", "ijl_apply_generic"),
         @augfunc(generic_augfwd),