diff --git a/src/host/construction.jl b/src/host/construction.jl index b3d6071f..1a4d8f7b 100644 --- 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