Update src/gather.jl

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

Author: yuehhua

src/gather.jl

@@ -49,7 +49,7 @@ function gather_kernel!(dst, src, idx::CUDA.CuDeviceArray{<:CartesianIndex}, max
     return nothing
 end
 
-function checkbounds_src(src, dims::Int, ::Type{<:Any})
+function checkbounds_src(src, dims::Union{Int, Val}, ::Type{<:Any})
     return i -> checkbounds(Bool, src, ntuple(x -> Colon(), dims)..., i...)
 end