diff --git a/src/compiler.jl b/src/compiler.jl index 5d2d29563f..8632469655 100644 --- 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 index df3b0ae181..31b2454a1b 100644 --- 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),