Debug specfunc (#2052)

* Debug specfunc * Add test * Update optimize.jl
EnzymeAD · Nov 5, 2024 · df20f18 · df20f18
1 parent 2e18dfc
commit df20f18
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/src/compiler.jl b/src/compiler.jl
@@ -3453,7 +3453,7 @@ function annotate!(mod, mode)
                 LLVM.API.LLVMRemoveEnumAttributeAtIndex(
                     fn,
                     reinterpret(LLVM.API.LLVMAttributeIndex, LLVM.API.LLVMAttributeFunctionIndex),
-                    kind(EnumAttribute("allockind")),
+                    kind(EnumAttribute("allockind", AllocFnKind(AFKE_Alloc).data)),
                 )
                 push!(function_attributes(fn), no_escaping_alloc)
                 push!(function_attributes(fn), LLVM.EnumAttribute("allockind", (AllocFnKind(AFKE_Alloc) | AllocFnKind(AFKE_Uninitialized)).data))

diff --git a/test/optimize.jl b/test/optimize.jl
@@ -115,3 +115,18 @@ end
         Enzyme.Const(dist)
     )
 end
+
+@noinline function mc_g(i, _not_used)
+    k = (0.25)
+    return (i, k)
+end
+
+function mc_f(_not_used)
+    i = (0.0, 3.9555)
+    t = mc_g(i, _not_used)
+    return t[1][2]
+end
+
+@testset "Memcopy of constant" begin
+    @test Enzyme.autodiff(Enzyme.Forward, mc_f, Duplicated(2.7, 1.0))[1] ≈ 0.0
+end