diff --git a/ext/StridedGPUArraysExt.jl b/ext/StridedGPUArraysExt.jl index 4cbe7be..608d8b5 100644 --- a/ext/StridedGPUArraysExt.jl +++ b/ext/StridedGPUArraysExt.jl @@ -24,13 +24,13 @@ end function Base.fill!(A::StridedView{T, N, TA, F}, x) where {T, N, TA <: AbstractGPUArray{T}, F <: ALL_FS} isempty(A) && return A @kernel function fill_kernel!(a, val) - idx = @index(Global, Linear) + idx = @index(Global, Cartesian) @inbounds a[idx] = val end # ndims check for 0D support kernel = fill_kernel!(KernelAbstractions.get_backend(A)) f_x = F <: Union{typeof(conj), typeof(adjoint)} ? conj(x) : x - kernel(A, f_x; ndrange = length(A)) + kernel(A, f_x; ndrange = size(A)) return A end