Update src/gather.jl

Co-authored-by: Peter <[email protected]>

Author: yuehhua

src/gather.jl

@@ -68,7 +68,7 @@ function NNlib.gather!(dst::AnyCuArray, src::AnyCuArray, idx::AnyCuArray)
     chk = checkbounds_src(src, dims, eltype(idx))
     in_bnd = map(chk, collect(idx))
     if !all(in_bnd)
-        j = findfirst(i -> !i, in_bnd)
+        j = findfirst(!, in_bnd)
         k = CUDA.@allowscalar idx[j]
         throw(BoundsError(src, k))
     end