Speed up fill for high dimensional arrays (#591)

GiggleLiu · web-flow · commit 55a943ea5c87 · 2025-04-25T12:19:13.000+02:00
diff --git a/src/host/construction.jl b/src/host/construction.jl
@@ -19,7 +19,7 @@ function Base.fill!(A::AnyGPUArray{T}, x) where T
 
     # ndims check for 0D support
     kernel = fill_kernel!(get_backend(A))
-    kernel(A, x; ndrange = ndims(A) > 0 ? size(A) : (1,))
+    kernel(A, x; ndrange = length(A))
     A
 end
 

Original file line number	Diff line number	Diff line change
`@@ -19,7 +19,7 @@ function Base.fill!(A::AnyGPUArray{T}, x) where T`
`19`	`19`
`20`	`20`	`# ndims check for 0D support`
`21`	`21`	`kernel = fill_kernel!(get_backend(A))`
`22`		`- kernel(A, x; ndrange = ndims(A) > 0 ? size(A) : (1,))`
	`22`	`+ kernel(A, x; ndrange = length(A))`
`23`	`23`	`A`
`24`	`24`	`end`
`25`	`25`