Update src/gather.jl

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

Author: yuehhua

src/gather.jl

@@ -65,8 +65,7 @@ function NNlib.gather!(dst::AnyCuArray, src::AnyCuArray, idx::AnyCuArray)
     max_idx = max_dims_idx * length(idx)
 
     # check bounds
-    chk = checkbounds_src(src, dims, eltype(idx))
-    in_bnd = map(chk, collect(idx))
+    in_bnd = map(checkbounds_src(src, Val(dims), eltype(idx)), idx)
     if !all(in_bnd)
         j = findfirst(!, in_bnd)
         k = CUDA.@allowscalar idx[j]