Preserve source array when calling memset.

maleadt · maleadt · commit 2b22e89055d8 · 2025-01-25T09:24:16.000+01:00
diff --git a/src/array.jl b/src/array.jl
@@ -780,7 +780,7 @@ function Base.fill!(A::DenseCuArray{T}, x) where T <: MemsetCompatTypes
   U = memsettype(T)
   y = reinterpret(U, convert(T, x))
   context!(context(A)) do
-    memset(convert(CuPtr{U}, pointer(A)), y, length(A))
+    GC.@preserve A memset(convert(CuPtr{U}, pointer(A)), y, length(A))
   end
   A
 end

Original file line number	Diff line number	Diff line change
`@@ -780,7 +780,7 @@ function Base.fill!(A::DenseCuArray{T}, x) where T <: MemsetCompatTypes`
`780`	`780`	`U = memsettype(T)`
`781`	`781`	`y = reinterpret(U, convert(T, x))`
`782`	`782`	`context!(context(A)) do`
`783`		`- memset(convert(CuPtr{U}, pointer(A)), y, length(A))`
	`783`	`+ GC.@preserve A memset(convert(CuPtr{U}, pointer(A)), y, length(A))`
`784`	`784`	`end`
`785`	`785`	`A`
`786`	`786`	`end`