From e9cc7d85d20f2a97901b1ee388ea729e6a7e1160 Mon Sep 17 00:00:00 2001
From: Katharine Hyatt <khyatt@flatironinstitute.org>
Date: Mon, 9 Mar 2026 06:32:44 -0400
Subject: [PATCH] Use Cartesian indices rather than linear

---
 ext/StridedGPUArraysExt.jl | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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