More 1.11 debugging (#2007)

* More 1.11 debugging * fix shadow * fix
EnzymeAD · Oct 23, 2024 · 064e65c · 064e65c
1 parent 820c005
commit 064e65c
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/src/make_zero.jl b/src/make_zero.jl
@@ -352,7 +352,7 @@ else
 end
 
 @inline function EnzymeCore.make_zero!(
-    prev::Array{GenericMemory, Complex{T}},
+    prev::GenericMemory{kind, Complex{T}},
     seen::ST,
 )::Nothing where {T<:AbstractFloat,kind,ST}
     fill!(prev, zero(Complex{T}))

diff --git a/src/rules/llvmrules.jl b/src/rules/llvmrules.jl
@@ -656,6 +656,9 @@ function arraycopy_common(fwd, B, orig, shadowsrc, gutils, shadowdst; len=nothin
     algn = 0
     i8 = LLVM.IntType(8)
 
+    shadowsrcs = LLVM.Value[]
+    shadowdsts = LLVM.Value[]
+
 	for i = 1:width
 
 		evsrc = if width == 1
@@ -691,6 +694,14 @@ function arraycopy_common(fwd, B, orig, shadowsrc, gutils, shadowdst; len=nothin
 			LLVM.memset!(B, shadowdst0, LLVM.ConstantInt(i8, 0, false), length, algn)
 		end
 
+        push!(shadowsrcs, shadowsrc0)
+        push!(shadowdsts, shadowdst0)
+    end
+
+    for i in 1:width
+        shadowsrc0 = shadowsrcs[i]
+        shadowdst0 = shadowdsts[i]
+
 		API.sub_transfer(
 			gutils,
 			fwd ? API.DEM_ReverseModePrimal : API.DEM_ReverseModeGradient,